def test_push(self):
pusher = CognitePusher(self.client, "pre_", push_interval=1)
TestCognitePusher.gauge.set(5)
pusher._push_to_server()
self.client.datapoints.insert_multiple.assert_called_once()
args = self.client.datapoints.insert_multiple.call_args_list[0][0][0][
-1]
timestamp = int(arrow.get().float_timestamp * 1000)
self.assertEqual(args["externalId"], "pre_gauge")
self.assertLess(abs(timestamp - args["datapoints"][0][0]),
100) # less than 100 ms
self.assertAlmostEqual(args["datapoints"][0][1], 5)
def test_init_empty_cdf(self):
self.client.time_series.retrieve_multiple = Mock(
side_effect=CogniteNotFoundError([{
"externalId": "pre_gauge"
}]))
return_asset = Asset(id=123, external_id="asset", name="asset")
new_asset = Asset(external_id="asset", name="asset")
self.client.assets.create = Mock(return_value=return_asset)
pusher = CognitePusher(self.client,
external_id_prefix="pre_",
asset=new_asset,
push_interval=1)
# Assert time series created
# Hacky assert_called_once_with as the TimeSeries object is not the same obj, just equal content
self.client.time_series.create.assert_called_once()
print(self.client.time_series.create.call_args_list)
self.assertDictEqual(
self.client.time_series.create.call_args_list[0][0][0][0].dump(),
TimeSeries(external_id="pre_gauge",
name="gauge",
legacy_name="pre_gauge",
description="Test gauge",
asset_id=123).dump(),
)
# Assert asset created
self.client.assets.create.assert_called_once_with(new_asset)
def test_init_existing_all(self):
return_asset = Asset(id=123, external_id="assetid", name="asset")
new_asset = Asset(external_id="assetid", name="asset")
self.client.assets.create = Mock(
side_effect=CogniteDuplicatedError(["assetid"]))
self.client.assets.retrieve = Mock(return_value=return_asset)
pusher = CognitePusher(self.client,
external_id_prefix="pre_",
asset=new_asset,
push_interval=1)
# Assert time series created
self.client.time_series.create.assert_not_called()
# Assert asset created
self.client.assets.create.assert_called_once_with(new_asset)
self.client.assets.retrieve.assert_called_once_with(
external_id="assetid")