"""Unit tests for art_relations_mysql_hook.""" from dags.lib.config import AR_MYSQL_CONN_NAME from hooks.art_relations_mysql_hook import ArtRelationsMySqlHook def test_initializing_art_relations_mysql_hook(): """Test initializing hook without arguments sets a mysql_conn_id.""" ar_conn = ArtRelationsMySqlHook() assert ar_conn.mysql_conn_id == AR_MYSQL_CONN_NAME other_ar_conn = ArtRelationsMySqlHook(mysql_conn_id='other') assert other_ar_conn.mysql_conn_id == 'other'