def test_compressed_text_from_url(self, mock_url_call, mock_gzip_read): mock_gzip_read.return_value = "Hello World" mock_url_call.return_value = mock.MagicMock( status_code=200, headers={'content-type': "application/x-gzip"}) resource = Resource("http://simple.com") data = resource.read() self.assertEqual("Hello World", data)
def test_open_json_from_file(self): file_resource = os.path.join(self.test_data_path, 'simple.json') resource = Resource(file_resource) data = resource.read() self.assertEqual( json.loads( "{ \"name\":\"John\", \"age\":31, \"city\":\"New York\" }"), data)
def test_text_from_url(self, mock_url_call): mock_url_call.return_value = mock.MagicMock( status_code=200, headers={'content-type': "text/plain"}, content="Hello World") resource = Resource("http://simple.com") data = resource.read() self.assertEqual("Hello World", data)
def test_compressed_json_from_url(self, mock_url_call, mock_gzip_read): mock_gzip_read.return_value = "{ \"name\":\"John\", \"age\":31, \"city\":\"New York\" }" mock_url_call.return_value = mock.MagicMock( status_code=200, headers={'content-type': "application/x-gzip"}) resource = Resource("http://simple.com") data = resource.read() self.assertEqual( json.loads( "{ \"name\":\"John\", \"age\":31, \"city\":\"New York\" }"), data)
def test_open_from_directory(self): test_data = [ 'files.csv', 'platforms/linux/local/one_cve.py', 'platforms/linux/remote/invalid_cve.py', 'platforms/linux/remote/no_cve.php', 'platforms/linux/remote/two_different_cves.txt', 'platforms/linux/remote/two_different_one_same_cve.txt', 'platforms/linux/remote/two_same_cves.c' ] resource = Resource(os.path.join(self.test_data_path, 'exploit-source')) data = resource.read() common_prefix = os.path.commonprefix(data) self.assertTrue(isinstance(data, list)) for filename in data: self.assertIn(filename.replace(common_prefix, ''), test_data)
def test_delete(self): test_cache_file = Resource(self.location) self.assertTrue(test_cache_file.exists) test_cache_file.delete() self.assertFalse(test_cache_file.exists)
def test_exists(self): test_cache_file = Resource(self.location) self.assertTrue(test_cache_file.exists)
def test_init_two(self): test_cache_file = Resource(self.location) self.assertEqual(self.location, test_cache_file.location) self.assertEqual(self.cache_file, test_cache_file.filename) self.assertEqual(self.cache_path, test_cache_file.filepath)
def test_open_text_from_file(self): file_resource = os.path.join(self.test_data_path, 'simple.txt') resource = Resource(file_resource) data = resource.read() self.assertEqual("This is a simple test.\n", data)