def test_timeout(self):
expected = self.create('BUILDING', datetime(2000, 01, 01, 1, 1, 1))
self.called = False
def urlopen_timeout(request, **kwargs):
self.called = True
raise socket.timeout('TIMEOUT')
restore = RestoreSuspects(self.conf, self.db)
with patch(suspects, 'urlopen', urlopen_timeout):
restore.run(datetime(2000, 01, 01, 1, 1, 30))
self.assert_(self.called)
def test_run(self):
expected = self.create('BUILDING', datetime(2000, 01, 01, 1, 1, 1))
available = self.create('AVAILABLE', datetime(2000, 01, 01, 1, 1, 1))
not_expected = self.create('BUILDING',
datetime(2000, 01, 01, 1, 1, 25))
self.count = 0
def urlopen(request, **kwargs):
self.assert_(kwargs.get('timeout'))
self.assertEquals(kwargs['timeout'], self.timeout)
self.count += 1
data = parse_qs(request.get_data())
self.assertIn('backup_source_volume_id', data)
self.assertEquals(data['backup_id'][0], expected.restore_of)
self.assertEquals(data['size'][0], '0')
return MockResponse('{}', 200)
restore = RestoreSuspects(self.conf, self.db)
with patch(suspects, 'urlopen', urlopen):
restore.run(datetime(2000, 01, 01, 1, 1, 30))
self.assertEquals(self.count, 1)