def test_scrape_schemas_metadata_with_csv_should_return_objects( self, normalize, read_csv): # noqa metadata = \ utils.Utils.convert_json_to_object( self.__MODULE_PATH, 'metadata.json') read_csv.return_value = metadata normalize.return_value = metadata scraper = metadata_scraper.MetadataScraper() schemas_metadata = scraper.scrape({}, csv_path='csv') self.assertEqual(1, len(schemas_metadata))
def test_scrape_schemas_metadata_on_exception_should_re_raise( self, normalize, connect): # noqa connect.side_effect = Exception('Error when connecting to Server') scraper = metadata_scraper.MetadataScraper() self.assertRaises(Exception, scraper.scrape, {}, connection_args={ 'database': 'db', 'host': 'mysql_host', 'user': '******', 'pass': '******' }) self.assertEqual(connect.call_count, 1) self.assertEqual(normalize.call_count, 0)
def test_scrape_schemas_metadata_with_credentials_should_return_objects( self, normalize, connect): # noqa metadata = \ utils.Utils.convert_json_to_object( self.__MODULE_PATH, 'metadata.json') con = Mock() connect.return_value = con cursor = Mock() con.cursor.return_value = cursor cursor.fetchall.return_value = \ utils.Utils.convert_json_to_object( self.__MODULE_PATH, 'rows.json') cursor.description =\ utils.Utils.convert_json_to_object( self.__MODULE_PATH, 'description.json') normalize.return_value = metadata scraper = metadata_scraper.MetadataScraper() schemas_metadata = scraper.scrape({}, connection_args={ 'database': 'db', 'host': 'mysql_host', 'user': '******', 'pass': '******' }) self.assertEqual(1, len(schemas_metadata)) self.assertEqual(connect.call_count, 1)