feast-project option (#1125)

pyalex · web-flow · commit bd4f62595889 · 2020-11-03T14:51:00.000+08:00
Signed-off-by: Oleksii Moskalenko &lt;moskalenko.alexey@gmail.com&gt;
diff --git a/tests/e2e/conftest.py b/tests/e2e/conftest.py
@@ -23,6 +23,7 @@ def pytest_addoption(parser):
     parser.addoption("--redis-cluster", action="store_true")
     parser.addoption("--feast-version", action="store")
     parser.addoption("--bq-project", action="store")
+    parser.addoption("--feast-project", action="store", default="default")
 
 
 def pytest_runtest_setup(item):
diff --git a/tests/e2e/fixtures/client.py b/tests/e2e/fixtures/client.py
@@ -21,7 +21,7 @@ def feast_client(
     enable_auth,
 ):
     if pytestconfig.getoption("env") == "local":
-        return Client(
+        c = Client(
             core_url=f"{feast_core[0]}:{feast_core[1]}",
             serving_url=f"{feast_serving[0]}:{feast_serving[1]}",
             spark_launcher="standalone",
@@ -36,8 +36,8 @@ def feast_client(
             ),
         )
 
-    if pytestconfig.getoption("env") == "gcloud":
-        return Client(
+    elif pytestconfig.getoption("env") == "gcloud":
+        c = Client(
             core_url=f"{feast_core[0]}:{feast_core[1]}",
             serving_url=f"{feast_serving[0]}:{feast_serving[1]}",
             spark_launcher="dataproc",
@@ -52,6 +52,11 @@ def feast_client(
                 local_staging_path, "historical_output"
             ),
         )
+    else:
+        raise KeyError(f"Unknown environment {pytestconfig.getoption('env')}")
+
+    c.set_project(pytestconfig.getoption("feast_project"))
+    return c
 
 
 @pytest.fixture(scope="session")
