Merge "Check if samples returned by get_sample_data are not None"
This commit is contained in:
commit
aa9e8cbb99
@ -71,7 +71,8 @@ class _Base(plugin.CentralPollster):
|
||||
parse_url,
|
||||
params,
|
||||
cache)
|
||||
for data in sample_data:
|
||||
|
||||
for data in sample_data or []:
|
||||
if data is None:
|
||||
continue
|
||||
if not isinstance(data, list):
|
||||
|
@ -175,3 +175,13 @@ class TestBaseGetSamples(test.BaseTestCase):
|
||||
samples = self._get_samples('http://foo')
|
||||
|
||||
self.assertEqual(len(samples), 0)
|
||||
|
||||
def test_get_samples_return_no_generator(self):
|
||||
class NoneFakeDriver(driver.Driver):
|
||||
|
||||
def get_sample_data(self, meter_name, parse_url, params, cache):
|
||||
return None
|
||||
|
||||
self._setup_ext_mgr(http=NoneFakeDriver())
|
||||
samples = self._get_samples('http://foo')
|
||||
self.assertFalse(samples)
|
||||
|
Loading…
x
Reference in New Issue
Block a user