fix unit tests

astronomer · Feb 7, 2024 · f7cfc1e · f7cfc1e
1 parent fafdb87
commit f7cfc1e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/profiles/bigquery/test_bq_service_account_keyfile_dict.py b/tests/profiles/bigquery/test_bq_service_account_keyfile_dict.py
@@ -76,6 +76,7 @@ def test_profile_without_dataset(mock_bigquery_conn_with_dict: Connection):
     mock_bigquery_conn_with_dict.extra = json.dumps(
         {
             "project": "my_project",
+            "dataset": None,
             "keyfile_dict": json.dumps(sample_keyfile_dict),
         }
     )
@@ -107,7 +108,6 @@ def test_mock_profile(mock_bigquery_conn_with_dict: Connection):
         "type": "bigquery",
         "method": "service-account-json",
         "project": "mock_value",
-        "dataset": "mock_value",
         "threads": 1,
         "keyfile_json": None,
     }