class FileManagerTestCase(BaseTestCase):
def setUp(self):
self.file_manager = FileManager()
super().setUp()
def tearDown(self):
super().tearDown()
if os.path.isdir('../data_scraper/cache/'):
shutil.rmtree('../data_scraper/cache/')
@patch.object(CacheManager, 'checkCache')
@patch.object(CacheManager, 'saveCache')
def test_file_manager_response(self, save_cache_mock, check_cache_mock):
file_content = 'some sort of string'
check_cache_mock.return_value = None
with patch.object(requests, 'get') as get_mock:
get_mock.return_value = mock_response = Mock()
mock_response.content = file_content
mock_response.status_code = 200
self.assertEquals(
file_content, self.file_manager.getPage('http://httpbin.org/'))
@patch.object(CacheManager, 'checkCache')
@patch.object(CacheManager, 'saveCache')
def test_file_manager_exception(self, save_cache_mock, check_cache_mock):
file_content = 'some sort fo string'
check_cache_mock.return_value = None
with self.assertRaises(Exception):
with patch.object(requests, 'get') as get_mock:
get_mock.return_value = mock_response = Mock()
mock_response.content = file_content
mock_response.status_code = 500
self.file_manager.getPage('http://httpbin/')
@patch.object(CacheManager, 'checkCache')
@patch.object(CacheManager, 'saveCache')
def test_dict_reader(self, save_cache_mock, check_cache_mock):
check_cache_mock.return_value = None
response_dict = """'first_name','second_name'
'Hammy','Goonan'"""
response = self.file_manager.dictReader(response_dict)
self.assertTrue(iter(response))
self.assertIsInstance(next(response), dict)
def setUp(self):
self.file_manager = FileManager()
super().setUp()
