pk/atst
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add sample reporting data to mock cloud provider

Browse Source
This commit is contained in:
graham-dds 2020-02-05 11:10:51 -05:00
parent 5716e20b9e
commit 411d8a877c
1 changed files with 13 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
atst/domain/csp/cloud/mock_cloud_provider.py
View File

@ -1,4 +1,5 @@
from uuid import uuid4 from uuid import uuid4
import pendulum
from .cloud_provider_interface import CloudProviderInterface from .cloud_provider_interface import CloudProviderInterface
from .exceptions import ( from .exceptions import (
@ -459,6 +460,11 @@ class MockCloudProvider(CloudProviderInterface):
self._maybe_raise(self.UNAUTHORIZED_RATE, self.AUTHORIZATION_EXCEPTION) self._maybe_raise(self.UNAUTHORIZED_RATE, self.AUTHORIZATION_EXCEPTION)
object_id = str(uuid4()) object_id = str(uuid4())
start_of_month = pendulum.today(tz="utc").start_of("month").replace(tzinfo=None)
this_month = start_of_month.to_atom_string()
last_month = start_of_month.subtract(months=1).to_atom_string()
two_months_ago = start_of_month.subtract(months=2).to_atom_string()
properties = CostManagementQueryProperties( properties = CostManagementQueryProperties(
**dict( **dict(
columns=[ columns=[
@ -467,7 +473,13 @@ class MockCloudProvider(CloudProviderInterface):
{"name": "InvoiceId", "type": "String"}, {"name": "InvoiceId", "type": "String"},
{"name": "Currency", "type": "String"}, {"name": "Currency", "type": "String"},
], ],
rows=[], rows=[
[1.0, two_months_ago, "", "USD"],
[500.0, two_months_ago, "e05009w9sf", "USD"],
[50.0, last_month, "", "USD"],
[1000.0, last_month, "e0500a4qhw", "USD"],
[500.0, this_month, "", "USD"],
],
) )
) )