def test_endpoint_not_exist(self): with mockpatch.PatchObject(kwapi._Base, 'get_kwapi_client', side_effect=self.fake_get_kwapi_client): pollster = kwapi.EnergyPollster() samples = list(pollster.get_samples(self.manager, {})) self.assertEqual(0, len(samples))
def test_get_samples_cached(self): probe = {'id': 'A'} probe.update(PROBE_DICT['probes']['A']) cache = { kwapi.EnergyPollster.CACHE_KEY_PROBE: [probe], } self.manager.keystone = mock.Mock() pollster = kwapi.EnergyPollster() with mock.patch.object(pollster, '_get_probes') as do_not_call: do_not_call.side_effect = AssertionError('should not be called') samples = list(pollster.get_samples(self.manager, cache)) self.assertEqual(1, len(samples))
def test_sample(self): cache = {} samples = list(kwapi.EnergyPollster().get_samples( self.manager, cache, )) self.assertEqual(3, len(samples)) samples_by_name = dict((s.resource_id, s) for s in samples) for name, probe in PROBE_DICT['probes'].items(): sample = samples_by_name[name] expected = datetime.datetime.fromtimestamp( probe['timestamp']).isoformat() self.assertEqual(expected, sample.timestamp) self.assertEqual(probe['kwh'], sample.volume)
def test_default_discovery(self): pollster = kwapi.EnergyPollster() self.assertEqual('endpoint:energy', pollster.default_discovery)
def test_endpoint_not_exist(self): self.stubs.Set(kwapi._Base, 'get_kwapi_client', self.fake_get_kwapi_client) samples = list(kwapi.EnergyPollster().get_samples(self.manager, {})) self.assertEqual(len(samples), 0)