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 + "/")