From c3c1ddb03c450fce6df1e8a840a0750ffefca693 Mon Sep 17 00:00:00 2001
From: khoaguin <dkn.work@protonmail.com>
Date: Thu, 19 Sep 2024 11:13:33 +0700
Subject: [PATCH] working on low-side ds workflow

---
 .../bigquery/helpers/fixtures_sync.py         |  2 +-
 .../scenarios/bigquery/level_0_basic_test.py  | 28 -------------------
 tests/scenarios/bigquery/level_0_test.py      | 25 ++++++++++++++---
 3 files changed, 22 insertions(+), 33 deletions(-)
 delete mode 100644 tests/scenarios/bigquery/level_0_basic_test.py

diff --git a/tests/scenarios/bigquery/helpers/fixtures_sync.py b/tests/scenarios/bigquery/helpers/fixtures_sync.py
index 099b7c02433..0ce1dac7289 100644
--- a/tests/scenarios/bigquery/helpers/fixtures_sync.py
+++ b/tests/scenarios/bigquery/helpers/fixtures_sync.py
@@ -18,7 +18,7 @@ def make_user(
     email: str | None = None,
     password: str | None = None,
     role: ServiceRole = ServiceRole.DATA_SCIENTIST,
-):
+) -> TestUser:
     fake = Faker()
     if name is None:
         name = fake.name()
diff --git a/tests/scenarios/bigquery/level_0_basic_test.py b/tests/scenarios/bigquery/level_0_basic_test.py
deleted file mode 100644
index e7cbcfd71e8..00000000000
--- a/tests/scenarios/bigquery/level_0_basic_test.py
+++ /dev/null
@@ -1,28 +0,0 @@
-# @pytest.mark.asyncio
-# async def test_level_2_basic_scenario(request):
-#     ensure_package_installed("google-cloud-bigquery", "google.cloud.bigquery")
-#     ensure_package_installed("db-dtypes", "db_dtypes")
-
-#     scenario = Scenario(
-#         name="test_create_apis_and_triage_requests",
-#         events=[
-#             EVENT_USER_ADMIN_CREATED,
-#             EVENT_PREBUILT_WORKER_IMAGE_BIGQUERY_CREATED,
-#             EVENT_EXTERNAL_REGISTRY_BIGQUERY_CREATED,
-#             EVENT_WORKER_POOL_CREATED,
-#             EVENT_ALLOW_GUEST_SIGNUP_DISABLED,
-#             EVENT_USERS_CREATED,
-#             EVENT_USERS_CREATED_CHECKED,
-#             EVENT_QUERY_ENDPOINT_CREATED,
-#             EVENT_QUERY_ENDPOINT_CONFIGURED,
-#             EVENT_SCHEMA_ENDPOINT_CREATED,
-#             EVENT_SUBMIT_QUERY_ENDPOINT_CREATED,
-#             EVENT_SUBMIT_QUERY_ENDPOINT_CONFIGURED,
-#             EVENT_USERS_CAN_QUERY_MOCK,
-#             EVENT_USERS_CAN_SUBMIT_QUERY,
-#             EVENT_USERS_QUERY_NOT_READY,
-#             EVENT_ADMIN_APPROVED_FIRST_REQUEST,
-#             EVENT_USERS_CAN_GET_APPROVED_RESULT,
-#         ],
-#     )
-
diff --git a/tests/scenarios/bigquery/level_0_test.py b/tests/scenarios/bigquery/level_0_test.py
index a98c7e51572..ee51ab5131a 100644
--- a/tests/scenarios/bigquery/level_0_test.py
+++ b/tests/scenarios/bigquery/level_0_test.py
@@ -7,7 +7,9 @@
 from helpers.api import create_endpoints_query
 from helpers.api import create_endpoints_schema
 from helpers.api import create_endpoints_submit_query
+from helpers.api import run_api_path
 from helpers.api import set_endpoint_settings
+from helpers.asserts import result_is
 from helpers.events import Event
 from helpers.events import EventManager
 from helpers.events import Scenario
@@ -21,6 +23,7 @@
 from helpers.workers import create_prebuilt_worker_image
 from helpers.workers import create_worker_pool
 from helpers.workers import get_prebuilt_worker_image
+from level_2_basic_test import query_sql
 import pytest
 from unsync import unsync
 
@@ -72,9 +75,23 @@ async def user_low_side_activity(_, events, user, after=None):
     guest_user_setup_flow(user.email)
 
     # login_user
+    user_client = user.client()
 
     # submit_code
     # request_approval
+    test_query_path = "bigquery.test_query"
+    await result_is(
+        events,
+        lambda: len(run_api_path(user_client, test_query_path, sql_query=query_sql()))
+        == 10000,
+        matches=True,
+        after=[
+            Event.QUERY_ENDPOINT_CONFIGURED,
+            Event.USERS_CREATED_CHECKED,
+            Event.ADMIN_SYNC_HIGH_TO_LOW,
+        ],
+        register=Event.USERS_CAN_QUERY_MOCK,
+    )
 
     # loop: wait for approval
 
@@ -236,7 +253,7 @@ async def admin_low_side_activity(_, events):
 
 
 @unsync_
-async def admin_create_api_endpoint(
+async def admin_create_sync_api_endpoints(
     _,
     events,
     admin_client_high,
@@ -323,7 +340,7 @@ async def admin_high_side_activity(_, events):
         events,
     )
 
-    admin_create_api_endpoint(
+    admin_create_sync_api_endpoints(
         _,
         events,
         admin_client_high,
@@ -356,11 +373,11 @@ async def test_level_0_k8s(request):
     events.add_scenario(scenario)
     events.monitor()
 
-    # start admin activity on high side
+    # start admin activity on low side
     admin_low_side_activity(request, events)
 
     # todo
-    # admin_high_side_activity(request, events)
+    admin_high_side_activity(request, events)
 
     # todo - only start syncing after the root user created other admin users
     # root_sync_activity(request, events, after=Event.USER_ADMIN_CREATED)