def setUp(self, *args, **kwargs): warnings.simplefilter('ignore', ResourceWarning) self.maxDiff = None ADDRESSES_TABLE = 'addresses' self.user_adapter = syngenta_digital_dta.adapter( engine='redshift', table='users', endpoint='localhost', database='dta-redshift', port=5439, user='******', password='******', model_schema='test-redshift-user-model', model_schema_file='tests/openapi.yml', model_identifier='user_id', model_version_key='modified', relationships={f'{ADDRESSES_TABLE}': 'user_id'}) self.address_adapter = syngenta_digital_dta.adapter( engine='redshift', endpoint='localhost', database='dta-redshift', table=ADDRESSES_TABLE, port=5439, user='******', password='******', model_schema='test-redshift-address-model', model_schema_file='tests/openapi.yml', model_identifier='address_id', model_version_key='modified') self.user_adapter.connect() self.address_adapter.connect() self.__create_tables()
def test_adapter_exception(self): try: syngenta_digital_dta.adapter(engine='not-supported') self.assertEqual(True, False) except Exception as e: self.assertEqual( str(e), 'engine not-supported not supported; contribute to get it supported :)' )
def test_elasticseach_adapter(self): es = syngenta_digital_dta.adapter( engine='elasticsearch', index='users', endpoint='localhost', model_schema='test-elasticsearch-user-model', model_schema_file='tests/openapi.yml', model_identifier='user_id', model_version_key='modified') self.assertIsInstance(es, syngenta_digital_dta.ElasticsearchAdapter)
def test_dynamodb_adapter(self): ddb = syngenta_digital_dta.adapter( engine='dynamodb', table='TABLE_NAME', endpoint='http://localhost:4000', model_schema='test-dynamo-model', model_schema_file='tests/openapi.yml', model_identifier='test_id', model_version_key='modified') self.assertIsInstance(ddb, syngenta_digital_dta.DynamodbAdapter)
def setUp(self, *args, **kwargs): warnings.simplefilter('ignore', ResourceWarning) self.maxDiff = None self.es_mapper_adapter = syngenta_digital_dta.adapter( engine='elasticsearch', index='company', endpoint='localhost', model_schema='test-es-mapper-model', model_schema_file='tests/openapi.yml', model_identifier='company_id', model_version_key='modified')
def setUp(self, *args, **kwargs): warnings.simplefilter('ignore', ResourceWarning) self.maxDiff = None self.bucket = 'unit-test' self.endpoint = 'http://localhost:4566' self.file = open('./tests/mock/example.json') self.__create_unit_test_bucket() self.adapter = syngenta_digital_dta.adapter( engine='s3', endpoint=self.endpoint, bucket=self.bucket )
def setUp(self, *args, **keywargs): TABLE_NAME = 'unittestsort' warnings.simplefilter('ignore', ResourceWarning) self.maxDiff = None self.mock_table = MockTable(table_name=TABLE_NAME) self.mock_table.setup_test_table() self.adapter = syngenta_digital_dta.adapter( engine='dynamodb', table=TABLE_NAME, endpoint='http://localhost:4000', model_schema='test-dynamo-model', model_schema_file='tests/openapi.yml', model_identifier='test_id', model_version_key='modified')
def test_redshift_adapter(self): ps = syngenta_digital_dta.adapter( engine='redshift', table='users', endpoint='localhost', database='dta-redshift', port=5432, user='******', password='******', model_schema='test-redshift-user-model', model_schema_file='tests/openapi.yml', model_identifier='user_id', model_version_key='modified', relationships={'ADDRESSES_TABLE': 'user_id'}) self.assertIsInstance(ps, syngenta_digital_dta.PostgresAdapter)
def test_create_index_no_template(self): try: adapter = syngenta_digital_dta.adapter( engine='elasticsearch', index=uuid.uuid4().hex, endpoint='localhost', model_schema='test-elasticsearch-user-model', model_schema_file='tests/openapi.yml', model_identifier='user_id', model_version_key='modified') adapter.create_index(template=False) self.assertEqual(True, True) except Exception as e: print(e) self.assertEqual(False, True)