Merge "Check if samples returned by get_sample_data are not None"

This commit is contained in:
Jenkins 2014-05-22 20:43:06 +00:00 committed by Gerrit Code Review
commit aa9e8cbb99
2 changed files with 12 additions and 1 deletions

View File

@ -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):

View File

@ -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)