def test_identify_calls_identify_rfu(self):
tempdir = '/tmp/dscan18293u1/'
with patch(ASYNC_SCAN + 'download_rfu', return_value=tempdir, autospec=True) as dr:
with patch(ASYNC_SCAN + 'identify_rfu') as ir:
identify_url(self.base_url, None)
args, kwargs = ir.call_args
self.assertEqual(ir.call_count, 1)
self.assertEqual(args[0], tempdir)
def test_identify_calls_all_rfu(self):
rfu = pu.get_rfu()
with patch(ASYNC + 'download_url', autospec=True) as du:
identify_url(self.base_url, None)
self.assertEquals(du.call_count, len(rfu))
for i, call in enumerate(du.call_args_list):
args, kwargs = call
self.assertEquals(args[0], self.base_url + rfu[i])
self.assertTrue(args[2].endswith(async.filename_encode(rfu[i])))
def test_identify_calls_all_rfu(self):
rfu = pu.get_rfu()
with patch(ASYNC + 'download_url', autospec=True) as du:
identify_url(self.base_url, None)
self.assertEquals(du.call_count, len(rfu))
for i, call in enumerate(du.call_args_list):
args, kwargs = call
self.assertEquals(args[0], self.base_url + rfu[i])
self.assertTrue(args[2].endswith(async .filename_encode(
rfu[i])))
def test_identify_calls_identify_rfu(self):
tempdir = '/tmp/dscan18293u1/'
with patch(ASYNC_SCAN + 'download_rfu',
return_value=tempdir,
autospec=True) as dr:
with patch(ASYNC_SCAN + 'identify_rfu') as ir:
identify_url(self.base_url, None)
args, kwargs = ir.call_args
self.assertEqual(ir.call_count, 1)
self.assertEqual(args[0], tempdir)
def test_identify_url_cleans_on_failure(self, rt):
tempdir = '/tmp/dscan18293u1/'
with patch('os.path.isdir', return_value=True, autospec=True) as isdir:
with patch(ASYNC_SCAN + 'download_rfu', return_value=tempdir, autospec=True) as dr:
with patch(ASYNC_SCAN + 'identify_rfu', side_effect=RuntimeError()) as ir:
self.assertFailure(identify_url('http://google.com/', None),
RuntimeError)
args, kwargs = rt.call_args
self.assertEquals(rt.call_count, 1)
self.assertEquals(isdir.call_count, 1)
self.assertEquals(args[0], tempdir)
def test_identify_url_cleans_on_failure(self, rt):
tempdir = '/tmp/dscan18293u1/'
with patch('os.path.isdir', return_value=True, autospec=True) as isdir:
with patch(ASYNC_SCAN + 'download_rfu',
return_value=tempdir,
autospec=True) as dr:
with patch(ASYNC_SCAN + 'identify_rfu',
side_effect=RuntimeError()) as ir:
self.assertFailure(
identify_url('http://google.com/', None), RuntimeError)
args, kwargs = rt.call_args
self.assertEquals(rt.call_count, 1)
self.assertEquals(isdir.call_count, 1)
self.assertEquals(args[0], tempdir)
def test_identify_raises_when_none_found(self, rt, mt, isdir):
ret = '/tmp/lelelellee'
mt.return_value = ret
def fail(*args, **kwargs):
return f()
rfu = pu.get_rfu()
with patch(ASYNC + 'download_url', side_effect=fail, autospec=True) as du:
with patch(ASYNC_SCAN + 'identify_rfu') as ir:
self.assertFailure(identify_url(self.base_url, None),
UnknownCMSException)
self.assertEquals(ir.call_count, 0)
args, kwargs = rt.call_args
self.assertEquals(rt.call_count, 1)
self.assertEquals(mt.call_count, 1)
self.assertEquals(args[0], ret + "/")
def test_identify_raises_when_none_found(self, rt, mt, isdir):
ret = '/tmp/lelelellee'
mt.return_value = ret
def fail(*args, **kwargs):
return f()
rfu = pu.get_rfu()
with patch(ASYNC + 'download_url', side_effect=fail,
autospec=True) as du:
with patch(ASYNC_SCAN + 'identify_rfu') as ir:
self.assertFailure(identify_url(self.base_url, None),
UnknownCMSException)
self.assertEquals(ir.call_count, 0)
args, kwargs = rt.call_args
self.assertEquals(rt.call_count, 1)
self.assertEquals(mt.call_count, 1)
self.assertEquals(args[0], ret + "/")
