def setup() -> None:
# TODO: implement to create custom resource
engine: str = os.environ.get('ENGINE', 'MySQLJDBC')
if engine.upper().startswith('MYSQL'):
db_setting: DBSetting = DBSetting(
HOST=os.environ.get('MYSQL_HOST', '127.0.0.1'),
PORT=os.environ.get('MYSQL_PORT', '3306'),
USER=os.environ.get('MYSQL_USER', 'root'),
PASSWORD=os.environ.get('MYSQL_PASSWORD', 'example'),
JAR_PATH=os.environ.get(
'MYSQL_JDBC_JAR_PATH', '/usr/lib/jvm/mariadb-java-client.jar'
),
)
else:
db_setting = DBSetting(
HOST=os.environ.get('POSTGRES_HOST', '127.0.0.1'),
PORT=os.environ.get('POSTGRES_PORT', '5432'),
USER=os.environ.get('POSTGRES_USER', 'postgres'),
PASSWORD=os.environ.get('POSTGRES_PASSWORD', 'example'),
JAR_PATH=os.environ.get(
'POSTGRES_JDBC_JAR_PATH', '/usr/lib/jvm/postgresql-java-client.jar'
),
)
register_secret(db_setting.USER, db_setting.PASSWORD, SECRET_ARN)
register_resource(
RESOURCE_ARN,
engine,
db_setting.HOST,
db_setting.PORT,
db_setting.USER,
db_setting.PASSWORD,
{'JAR_PATH': db_setting.JAR_PATH},
)
def test_register_resource(clear) -> None:
resource_arn: str = 'dummy_resource_arn'
register_resource(
resource_arn,
'MySQL',
host='localhost',
port=3306,
user_name='test',
password='******',
)
resource_meta: ResourceMeta = RESOURCE_METAS[resource_arn]
assert resource_meta.host == 'localhost'
assert resource_meta.port == 3306
assert resource_meta.user_name == 'test'
assert resource_meta.password == 'pw'