from validate import check_query_f
import pytest

usefixtures = pytest.mark.usefixtures

# use_mutation_fixtures = usefixtures(
#     'per_class_db_schema_for_mutation_tests',
#     'per_method_db_data_for_mutation_tests'
# )

@usefixtures('per_class_tests_db_state')
class TestV2SelectBasic: # Basic RQL Tests on v2/query
    @classmethod
    def dir(cls):
        return 'queries/v2/basic'

    def test_select_query_author(self, hge_ctx):
        check_query_f(hge_ctx, self.dir() + '/select_article.yaml')

    def test_select_query_author_with_user_role_success(self, hge_ctx):
        check_query_f(hge_ctx, self.dir() + '/select_article_role_success.yaml')

    # TODO: Fix this test for JWT
    # def test_select_query_author_with_user_role_failure(self, hge_ctx):
    #     check_query_f(hge_ctx, self.dir() + '/select_article_role_error.yaml')

@pytest.mark.backend('mssql')
@usefixtures('per_class_tests_db_state')
class TestRunSQLMSSQL:

    def test_drop_article_table_without_cascade(self, hge_ctx):
        check_query_f(hge_ctx, self.dir() + '/drop_article_table_without_cascade.yaml')

    def test_drop_article_table_with_cascade(self, hge_ctx):
        check_query_f(hge_ctx, self.dir() + '/drop_article_table_with_cascade.yaml')

    def test_create_author_table_fail(self, hge_ctx):
        check_query_f(hge_ctx, self.dir() + '/create_author_table_fail.yaml')

    def test_invalid_sql_query(self, hge_ctx):
        check_query_f(hge_ctx, self.dir() + '/invalid_sql_query.yaml')

    def test_select_query(self, hge_ctx):
        check_query_f(hge_ctx, self.dir() + '/sql_select_query_mssql.yaml')

    def test_drop_table(self, hge_ctx):
        check_query_f(hge_ctx, self.dir() + '/sql_drop_table_mssql.yaml')

    def test_rename_table(self, hge_ctx):
        check_query_f(hge_ctx, self.dir() + '/sql_rename_table_mssql.yaml')

    def test_drop_column(self, hge_ctx):
        check_query_f(hge_ctx, self.dir() + '/sql_drop_column_mssql.yaml')

    def test_add_column(self, hge_ctx):
        check_query_f(hge_ctx, self.dir() + '/sql_add_column_mssql.yaml')

    def test_rename_column(self, hge_ctx):
        check_query_f(hge_ctx, self.dir() + '/sql_rename_column_mssql.yaml')

    def test_select_query_fail(self, hge_ctx):
        check_query_f(hge_ctx, self.dir() + '/sql_select_query_fail.yaml')

    def test_add_column_fail(self, hge_ctx):
        check_query_f(hge_ctx, self.dir() + '/sql_add_column_fail.yaml')

    def test_drop_column_fail(self, hge_ctx):
        check_query_f(hge_ctx, self.dir() + '/sql_drop_column_fail.yaml')

    def test_create_index_fail(self, hge_ctx):
        check_query_f(hge_ctx, self.dir() + '/create_index_fail.yaml')

    @classmethod
    def dir(cls):
        return 'queries/v2/mssql/run_sql'