ci: Add example test for sql registry using in-memory sqlite (feast-dev#3012)

achals · web-flow · commit d4af068953e3 · 2022-08-04T12:33:33.000-07:00
Signed-off-by: Achal Shah &lt;achals@gmail.com&gt;
diff --git 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
