def test_should_return_timeseries_json(self):
# expect
self.assertEqual(
monitoring.create_timeseries_data('metric type', 'job id', 'project id', 'region', 'env name', 'start',
'end'), {
"metric": {
"type": 'metric type',
"labels": {
"job_id": 'job id'
}
},
"resource": {
"type": 'cloud_composer_environment',
"labels": {
'project_id': 'project id',
'location': 'region',
'environment_name': 'env name'
}
},
"points": [
{
"interval": {
"startTime": 'start',
"endTime": 'end'
},
"value": {
"int64Value": 1
}
}
]
})
def test_should_call_api_create_timeseries(
self, api_create_timeseries_mock: mock.Mock):
# given
monitoring_config = monitoring.MonitoringConfig(
'test project', 'eu-west', 'env')
# when
monitoring.increment_counter('client', monitoring_config,
'metric type', 'job id')
# then
api_create_timeseries_mock.assert_called_once_with(
'client', monitoring_config,
monitoring.create_timeseries_data('metric type', 'job id',
'test project', 'eu-west', 'env',
TEST_DATETIME_RFC3339,
TEST_DATETIME_RFC3339))