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)
