chore: add dbt test and fixtures

.gitignore

@@ -34,3 +34,8 @@ bench_data/
 
 # Logs
 *.log
+
+# dbt compiled models
+tests/fixtures/dbt_project/target/
+tests/fixtures/dbt_project/.user.yml
+tests/fixtures/dbt_project/logs/

tests/fixtures/dbt_project/dbt_project.yml

@@ -0,0 +1,36 @@
+
+# Name your project! Project names should contain only lowercase characters
+# and underscores. A good package name should reflect your organization's
+# name or the intended use of these models
+name: 'glaredb_dbt_test'
+version: '1.0.0'
+config-version: 2
+
+# This setting configures which "profile" dbt uses for this project.
+profile: 'glaredb_dbt_test'
+
+# These configurations specify where dbt should look for different types of files.
+# The `model-paths` config, for example, states that models in this project can be
+# found in the "models/" directory. You probably won't need to change these!
+model-paths: ["models"]
+test-paths: ["tests"]
+macro-paths: ["macros"]
+snapshot-paths: ["snapshots"]
+
+clean-targets:         # directories to be removed by `dbt clean`
+  - "target"
+  - "dbt_packages"
+
+
+# Configuring models
+# Full documentation: https://docs.getdbt.com/docs/configuring-models
+
+# In this example config, we tell dbt to build all models in the example/
+# directory as views. These settings can be overridden in the individual model
+# files using the `{{ config(...) }}` macro.
+
+models:
+  dbt_test:
+    # Config indicated by + and applies to all files under models/example/
+    glaredb_data:
+      +materialized: view

tests/fixtures/dbt_project/models/glaredb_data/glaredb_model.sql

@@ -0,0 +1,2 @@
+select * 
+from {{ source('public', 'dbt_test')}}

tests/fixtures/dbt_project/models/glaredb_data/sources.yml

@@ -0,0 +1,7 @@
+version: 2
+
+sources:
+  - name: public
+    database: glaredb
+    tables: 
+      - name: dbt_test

tests/fixtures/dbt_project/profiles.yml

@@ -0,0 +1,12 @@
+glaredb_dbt_test:
+  outputs:
+    test_target:
+      dbname: glaredb
+      host: 127.0.0.1
+      pass: ""
+      port: 5432
+      schema: public
+      threads: 1
+      type: postgres
+      user: "{{ env_var('DBT_USER') }}"
+  target: test_target

tests/pyproject.toml

@@ -13,6 +13,7 @@ psycopg2 = "^2.9.9"
 pyarrow = "^14.0.2"
 pylance = "^0.9.6"
 ruff = "0.1.14"
+dbt-core = "^1.7.7"
 
 [build-system]
 requires = ["poetry-core"]

tests/tests/test_dbt.py

@@ -0,0 +1,46 @@
+import psycopg2.extensions
+import pytest
+import os
+
+from tests.fixtures.glaredb import glaredb_connection, debug_path
+
+from dbt.cli.main import dbtRunner, dbtRunnerResult
+
+
+def test_dbt_glaredb(
+    glaredb_connection: psycopg2.extensions.connection,
+):
+    dbt: dbtRunner = dbtRunner()
+
+    os.environ["DBT_USER"] = glaredb_connection.info.user
+
+    model_name: str = "glaredb_model" # TODO
+    project_directory: str = "../fixtures/dbt_project/" # TODO
+    dbt_profiles_directory: str = "../fixtures/dbt_project/" # TODO
+
+    with glaredb_connection.cursor() as curr:
+        curr.execute("create table dbt_test (amount int)")
+        curr.execute(
+            "INSERT INTO dbt_test (amount) VALUES (0), (1), (2), (3), (4), (5), (6), (7), (8), (9)"
+        )
+
+    cli_args: list = [
+        "run",
+        "--project-dir",
+        project_directory,
+        "--profiles-dir",
+        dbt_profiles_directory,
+        "-m",
+        model_name
+    ]
+    #
+    res: dbtRunnerResult = dbt.invoke(cli_args)
+
+    # The below will currently fail. res contains the error message, but that message can be in different places based
+    # on where the failure is. Currently, it is under the top level `res.exception` key.
+    # assert res.success is True
+    #
+    # with glaredb_connection.cursor() as curr:
+    #     query_result: list = curr.execute(f"select * from {model_name}").fetchall()
+    #
+    # assert len(query_result) == 10