add test to check source uses staging api

glassflow · Sep 18, 2024 · 77fcdd4 · 77fcdd4
1 parent 257d15f
commit 77fcdd4
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/tests/glassflow/integration_tests/pipeline_test.py b/tests/glassflow/integration_tests/pipeline_test.py
@@ -3,6 +3,11 @@
 from glassflow import errors
 
 
+def test_using_staging_server(source, sink):
+    assert source.glassflow_config.server_url == "https://staging.api.glassflow.dev/v1"
+    assert sink.glassflow_config.server_url == "https://staging.api.glassflow.dev/v1"
+
+
 def test_pipeline_data_source_validate_credentials_ok(source):
     try:
         source.validate_credentials()