feat: add sqlalchemy tests (#1756)

* Create requirements.py

* Create setup.cfg

* Create conftest.py

* Update conftest.py

* Create sqlalchemy_standardtest_example.py

* Update setup.py

* Update sqlalchemy_standardtest_example.py

* Update setup.cfg

* Rename setup.cfg to sqlalchemy_test_config

* Update conftest.py

Co-authored-by: HuilinWu2 <[email protected]>

Author: tobegit3hub

python/openmldb/sqlalchemy_openmldb/requirements.py

@@ -0,0 +1,88 @@
+from sqlalchemy.testing.requirements import SuiteRequirements
+
+from sqlalchemy.testing import exclusions
+
+
+class Requirements(SuiteRequirements):
+    @property
+    def bound_limit_offset(self):
+        return exclusions.closed()
+
+    @property
+    def date(self):
+        return exclusions.closed()
+
+    @property
+    def datetime_microseconds(self):
+        return exclusions.closed()
+
+    @property
+    def floats_to_four_decimals(self):
+        return exclusions.closed()
+
+    # TODO: remove this when SQLA released with
+    #       https://gerrit.sqlalchemy.org/c/sqlalchemy/sqlalchemy/+/2990
+    @property
+    def implicitly_named_constraints(self):
+        return exclusions.open()
+
+    @property
+    def nullable_booleans(self):
+        """Target database allows boolean columns to store NULL."""
+        # Access Yes/No doesn't allow null
+        return exclusions.closed()
+
+    @property
+    def offset(self):
+        # Access does LIMIT (via TOP) but not OFFSET
+        return exclusions.closed()
+
+    @property
+    def parens_in_union_contained_select_w_limit_offset(self):
+        return exclusions.closed()
+
+    @property
+    def precision_generic_float_type(self):
+        return exclusions.closed()
+
+    @property
+    def reflects_pk_names(self):
+        return exclusions.open()
+
+    @property
+    def sql_expression_limit_offset(self):
+        return exclusions.closed()
+
+    @property
+    def temp_table_reflection(self):
+        return exclusions.closed()
+
+    @property
+    def temporary_tables(self):
+        return exclusions.closed()
+
+    @property
+    def temporary_views(self):
+        return exclusions.closed()
+
+    @property
+    def time(self):
+        return exclusions.closed()
+
+    @property
+    def time_microseconds(self):
+        return exclusions.closed()
+
+    @property
+    def timestamp_microseconds(self):
+        return exclusions.closed()
+
+    @property
+    def unicode_ddl(self):
+        # Access won't let you drop a child table unless
+        # you drop the FK constraint first. Not worth the grief.
+        return exclusions.closed()
+
+    @property
+    def view_column_reflection(self):
+        return exclusions.open()

python/openmldb/sqlalchemy_test_config

@@ -0,0 +1,26 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+# Copyright 2021 4Paradigm
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# [tool:pytest]
+# addopts= --tb native -v -r fxX --maxfail=25 -p no:warnings
+# python_files=/openmldb/test/*test_*.py
+
+# [sqla_testing]
+# requirement_cls=openmldb.sqlalchemy_openmldb.requirements:Requirements
+# profile_file=test/profiles.txt
+
+# [db]
+# default = openmldb:///db_test?zk=127.0.0.1:6181&zkPath=/onebox

python/openmldb/test/conftest.py

@@ -0,0 +1,24 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+# Copyright 2021 4Paradigm
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from sqlalchemy.dialects import registry
+import pytest
+
+# registry.register("openmldb", "openmldb.sqlalchemy_openmldb.openmldb_dialect", "OpenmldbDialect")
+
+# pytest.register_assert_rewrite("sqlalchemy.testing.assertions")
+
+# from sqlalchemy.testing.plugin.pytestplugin import *

python/openmldb/test/sqlalchemy_standardtest_example.py

@@ -0,0 +1,27 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+# Copyright 2021 4Paradigm
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from sqlalchemy.testing.suite import *
+
+from sqlalchemy.testing.suite import IntegerTest as _IntegerTest
+
+# class IntegerTest(_IntegerTest):
+
+#     @testing.skip("access")
+#     def test_huge_int(self):
+#         # bypass this test because Access ODBC fails with
+#         # [ODBC Microsoft Access Driver] Optional feature not implemented.
+#         return

python/setup.py

@@ -31,6 +31,7 @@
         "sqlalchemy < 1.4.0",
         "IPython",
         "prettytable",
+        "tox",
         "pytest"
     ],
     include_package_data=True,