from tests.end_to_end.target_snowflake import TargetSnowflake


class TapPostgres(TargetSnowflake):
    """
    Base class for E2E tests for tap postgres -> target snowflake
    """

    # pylint: disable=arguments-differ
    def setUp(self, tap_id: str, target_id: str):
        super().setUp(tap_id=tap_id, target_id=target_id, tap_type='TAP_POSTGRES')
        self.e2e_env.clean_up_temp_dir()
        self.e2e_env.setup_tap_postgres()
