def test_clearSeries(self):
client = self._makeTempClient()
series = "test"
dp = iobeam.DataPoint(45)
client.addDataPoint(series, dp)
self.assertEqual(1, len(client._dataset))
self.assertEqual(1, len(client._dataset[series]))
client.clearSeries(series)
self.assertEqual(0, len(client._dataset))
def test_addDataPoint(self):
client = self._makeTempClient()
series = "test"
vals = [0, 11, 28]
dp = iobeam.DataPoint(vals[0])
client.addDataPoint(series, dp)
self.assertEqual(1, len(client._dataset))
self.assertEqual(1, len(client._dataset[series]))
self.assertTrue(dp in client._dataset[series])
dp = iobeam.DataPoint(vals[1])
client.addDataPoint(series, dp)
self.assertEqual(1, len(client._dataset))
self.assertEqual(2, len(client._dataset[series]))
self.assertTrue(dp in client._dataset[series])
dp = iobeam.DataPoint(vals[2])
client.addDataPoint(series, dp)
self.assertEqual(1, len(client._dataset))
self.assertEqual(3, len(client._dataset[series]))
self.assertTrue(dp in client._dataset[series])
def test_convertDataSetToBatches(self):
dummy = DummyBackend()
backend = request.DummyRequester(dummy)
client = self._makeTempClient(backend=backend, deviceId="fake")
series = "test"
vals = [0, 11, 28]
ts = data.Timestamp(1)
dp = iobeam.DataPoint(vals[0], timestamp=ts)
client.addDataPoint(series, dp)
ts = data.Timestamp(2)
dp = iobeam.DataPoint(vals[1], timestamp=ts)
client.addDataPoint(series, dp)
self.assertEqual(1, len(client._dataset))
self.assertEqual(0, len(client._batches))
batches = client._convertDataSetToBatches()
self.assertEqual(0, len(client._dataset))
self.assertEqual(1, len(batches))
for b in batches:
self.assertEqual(1, len(b.columns()))
self.assertEqual(series, b.columns()[0])
self.assertEqual(2, len(b))
def test_addDataPointInvalidNames(self):
client = self._makeTempClient()
dp = iobeam.DataPoint(0)
def verify(series, point):
try:
client.addDataPoint(series, point)
self.assertTrue(False)
except ValueError:
pass
vals = [
(None, dp),
(1, dp),
("", dp)
]
for (series, point) in vals:
verify(series, point)
def test_send(self):
dummy = DummyBackend()
backend = request.DummyRequester(dummy)
client = self._makeTempClient(backend=backend, deviceId="fake")
client._checkToken = checkTokenNone
series = "test"
vals = [0, 11, 28]
dp = iobeam.DataPoint(vals[0])
client.addDataPoint(series, dp)
client.send()
self.assertEqual(0, len(client._batches))
self.assertEqual(0, len(client._dataset))
self.assertEqual(1, dummy.calls)
self.assertTrue(dummy.lastUrl.endswith("/imports"))
self.assertEqual(1, dummy.lastJson["project_id"])
self.assertEqual("fake", dummy.lastJson["device_id"])
# 2: 'fields' and 'data', batch format
self.assertEqual(2, len(dummy.lastJson["sources"]))