Merge pull request #4 from everpcpc/fix-driver

feat: switch driver backend

Author: hantmac

.github/workflows/test.yml

@@ -11,7 +11,6 @@ jobs:
     runs-on: ubuntu-latest
     services:
       databend:
-        # image: datafuselabs/databend-query
         image: datafuselabs/databend
         env:
           QUERY_DEFAULT_USER: databend
@@ -22,26 +21,32 @@ jobs:
           - 9000:9000
     steps:
       - name: Checkout
-        uses: actions/checkout@v2
+        uses: actions/checkout@v4
 
-      - name: Setup Python-3.10
+      - name: Setup Python
         uses: actions/setup-python@v4
         with:
-          python-version: '3.10'
+          python-version: '3.11'
 
       - name: Pip Install
         run: |
-          make install
+          pip install pipenv
+          pipenv install --dev --skip-lock
 
       - name: Verify Service Running
         run: |
           cid=$(docker ps -a | grep databend | cut -d' ' -f1)
           docker logs ${cid}
           curl -v http://localhost:8000/v1/health
 
-      - name: Test
+      - name: Unit Test
         env:
-          TEST_DATABEND_DSN: "http://databend:databend@localhost:8000/default?secure=false"
+          TEST_DATABEND_DSN: "databend://databend:databend@localhost:8000/default?sslmode=disable"
         run: |
-          make unittest
-          make integration
+          pipenv run pytest -s tests/unit
+
+      - name: Integration Test
+        env:
+          TEST_DATABEND_DSN: "databend://databend:databend@localhost:8000/default?sslmode=disable"
+        run: |
+          pipenv run pytest -s tests/integration

Pipfile

@@ -0,0 +1,15 @@
+[[source]]
+url = "https://pypi.org/simple"
+verify_ssl = true
+name = "pypi"
+
+[packages]
+databend-sqlalchemy = {file = ".", editable = true}
+
+[dev-packages]
+black = "*"
+flake8 = "*"
+pytest = "*"
+
+[requires]
+python_version = "3.11"

Pipfile.lock

@@ -18,14 +18,14 @@ The DSN format is similar to that of regular Postgres::
         from sqlalchemy import create_engine, text
         from sqlalchemy.engine.base import Connection, Engine
         engine = create_engine(
-            f"databend://{username}:{password}@{host_port_name}/{database_name}?secure=false"
+            f"databend://{username}:{password}@{host_port_name}/{database_name}?sslmode=disable"
         )
         connection = engine.connect()
         result = connection.execute(text("SELECT 1"))
         assert len(result.fetchall()) == 1
 
         import connector
-        cursor = connector.connect('http://root:@localhost:8081').cursor()
+        cursor = connector.connect('databend://root:@localhost:8000?sslmode=disable').cursor()
         cursor.execute('SELECT * FROM test')
         # print(cursor.fetchone())
         # print(cursor.fetchall())

README.rst

@@ -1,7 +1,5 @@
 #!/usr/bin/env python
 
 
-from databend_sqlalchemy.entry_points import validate_entrypoints  # noqa
-
 VERSION = (0, 3, 2)
 __version__ = ".".join(str(x) for x in VERSION)