From d4af068953e3e109c5319fcc9fcc5ef7c0c612b5 Mon Sep 17 00:00:00 2001
From: Achal Shah <achals@gmail.com>
Date: Thu, 4 Aug 2022 12:33:33 -0700
Subject: [PATCH] ci: Add example test for sql registry using in-memory sqlite
 (#3012)

Signed-off-by: Achal Shah <achals@gmail.com>
---
 .../registration/test_sql_registry.py         | 41 ++++++++++++++++---
 1 file changed, 36 insertions(+), 5 deletions(-)

diff --git a/sdk/python/tests/integration/registration/test_sql_registry.py b/sdk/python/tests/integration/registration/test_sql_registry.py
index 286b1abd21..5b44aaa5b3 100644
--- a/sdk/python/tests/integration/registration/test_sql_registry.py
+++ b/sdk/python/tests/integration/registration/test_sql_registry.py
@@ -108,13 +108,28 @@ def mysql_registry():
     container.stop()
 
 
+@pytest.fixture(scope="session")
+def sqlite_registry():
+
+    registry_config = RegistryConfig(
+        registry_type="sql",
+        path="sqlite://",
+    )
+
+    yield SqlRegistry(registry_config, None)
+
+
 @pytest.mark.skipif(
     sys.platform == "darwin" and "GITHUB_REF" in os.environ,
     reason="does not run on mac github actions",
 )
 @pytest.mark.parametrize(
     "sql_registry",
-    [lazy_fixture("mysql_registry"), lazy_fixture("pg_registry")],
+    [
+        lazy_fixture("mysql_registry"),
+        lazy_fixture("pg_registry"),
+        lazy_fixture("sqlite_registry"),
+    ],
 )
 def test_apply_entity_success(sql_registry):
     entity = Entity(
@@ -174,7 +189,11 @@ def assert_project_uuid(project, project_uuid, sql_registry):
 )
 @pytest.mark.parametrize(
     "sql_registry",
-    [lazy_fixture("mysql_registry"), lazy_fixture("pg_registry")],
+    [
+        lazy_fixture("mysql_registry"),
+        lazy_fixture("pg_registry"),
+        lazy_fixture("sqlite_registry"),
+    ],
 )
 def test_apply_feature_view_success(sql_registry):
     # Create Feature Views
@@ -250,7 +269,11 @@ def test_apply_feature_view_success(sql_registry):
 )
 @pytest.mark.parametrize(
     "sql_registry",
-    [lazy_fixture("mysql_registry"), lazy_fixture("pg_registry")],
+    [
+        lazy_fixture("mysql_registry"),
+        lazy_fixture("pg_registry"),
+        lazy_fixture("sqlite_registry"),
+    ],
 )
 def test_apply_on_demand_feature_view_success(sql_registry):
     # Create Feature Views
@@ -334,7 +357,11 @@ def location_features_from_push(inputs: pd.DataFrame) -> pd.DataFrame:
 )
 @pytest.mark.parametrize(
     "sql_registry",
-    [lazy_fixture("mysql_registry"), lazy_fixture("pg_registry")],
+    [
+        lazy_fixture("mysql_registry"),
+        lazy_fixture("pg_registry"),
+        lazy_fixture("sqlite_registry"),
+    ],
 )
 @pytest.mark.parametrize(
     "request_source_schema",
@@ -462,7 +489,11 @@ def odfv1(feature_df: pd.DataFrame) -> pd.DataFrame:
 @pytest.mark.integration
 @pytest.mark.parametrize(
     "sql_registry",
-    [lazy_fixture("mysql_registry"), lazy_fixture("pg_registry")],
+    [
+        lazy_fixture("mysql_registry"),
+        lazy_fixture("pg_registry"),
+        lazy_fixture("sqlite_registry"),
+    ],
 )
 def test_apply_data_source(sql_registry):
     # Create Feature Views