def test_get_clobber_times_null(self, mock_urlopen):
# make sure it can handle no data
fake_data_null = {'result': []}
fake_data_json_str = json.dumps(fake_data_null)
urllib2_mock = mock.Mock()
urllib2_mock.read.side_effect = [fake_data_json_str]
mock_urlopen.return_value = urllib2_mock
clobber_dates_return = get_clobber_times('clobberer/lastclobber/all')
self.assertDictEqual(clobber_dates_return, fake_data_null)
def test_get_clobber_times(self, mock_urlopen):
builddir = 'the-roadhouse'
timestamp = 9999
who = '*****@*****.**'
fake_data = {'result': [
{
'builddir': builddir,
'lastclobber': timestamp,
'who': who,
'slave': None
}
]
}
fake_data_json_str = json.dumps(fake_data)
urllib2_mock = mock.Mock()
urllib2_mock.read.side_effect = [fake_data_json_str]
mock_urlopen.return_value = urllib2_mock
clobber_dates_return = get_clobber_times('clobberer/lastclobber/all')
self.assertDictEqual(clobber_dates_return, fake_data)
