chore(tests): Add publish/unpublish to preview test.

Author: b4handjr

experimenter/experimenter/kinto/client.py

@@ -16,6 +16,9 @@ def __init__(self, collection, review=True):
         )
         self.review = review
         self.collection_data = None
+        self.read_bucket = (
+            settings.KINTO_BUCKET_WORKSPACE if not review else settings.KINTO_BUCKET_MAIN
+        )
 
     def _fetch_collection_data(self):
         self.collection_data = (
@@ -94,6 +97,6 @@ def get_main_records(self):
         return {
             r["id"]: r
             for r in self.kinto_http_client.get_records(
-                bucket=settings.KINTO_BUCKET_MAIN, collection=self.collection
+                bucket=self.read_bucket, collection=self.collection
             )
         }

experimenter/tests/integration/nimbus/conftest.py

@@ -147,10 +147,11 @@ def _verify_url(request, base_url):
 
 @pytest.fixture
 def kinto_client(default_data):
-    kinto_url = os.getenv("INTEGRATION_TEST_KINTO_URL", "http://kinto:8888/v1")
-    return KintoClient(
-        APPLICATION_KINTO_COLLECTION[default_data.application], server_url=kinto_url
-    )
+    def _kinto_client(collection=APPLICATION_KINTO_COLLECTION[default_data.application]):
+        kinto_url = os.getenv("INTEGRATION_TEST_KINTO_URL", "http://kinto:8888/v1")
+        return KintoClient(collection=collection, server_url=kinto_url)
+
+    return _kinto_client
 
 
 @pytest.fixture

experimenter/tests/integration/nimbus/kinto/client.py

@@ -8,6 +8,7 @@
 KINTO_COLLECTION_DESKTOP = "nimbus-desktop-experiments"
 KINTO_COLLECTION_MOBILE = "nimbus-mobile-experiments"
 KINTO_COLLECTION_WEB = "nimbus-web-experiments"
+KINTO_COLLECTION_PREVIEW = "nimbus-preview"
 KINTO_BUCKET_WORKSPACE = "main-workspace"
 KINTO_REVIEW_STATUS = "to-review"
 KINTO_REJECTED_STATUS = "work-in-progress"
@@ -64,3 +65,8 @@ def reject(self):
                 return
             time.sleep(2)
         raise Exception("Unable to reject kinto review")
+
+    def get_record_data(self):
+        return self.kinto_http_client.get_records(
+            collection=self.collection, bucket=KINTO_BUCKET_WORKSPACE
+        )

experimenter/tests/integration/nimbus/test_remote_settings.py

@@ -1,3 +1,5 @@
+import time
+
 import pytest
 
 from nimbus.pages.experimenter.home import HomePage
@@ -320,3 +322,41 @@ def test_create_new_experiment_timeout_remote_settings(
     summary = SummaryPage(selenium, experiment_url).open()
     summary.launch_and_approve()
     summary.wait_for_timeout_alert()
+
+
+@pytest.mark.remote_settings_launch
+def test_create_new_experiment_publish_to_preview_and_unpublish(
+    selenium,
+    experiment_url,
+    kinto_client,
+    base_url,
+    application,
+    default_data_api,
+    experiment_slug,
+):
+    timeout = time.time() + 120
+    records_before = len(kinto_client(collection="nimbus-preview").get_record_data())
+
+    helpers.create_experiment(experiment_slug, application, default_data_api)
+
+    summary = SummaryPage(selenium, experiment_url).open()
+    summary.launch_to_preview()
+
+    while time.time() < timeout:
+        preview_data = kinto_client(collection="nimbus-preview").get_record_data()
+        if any(record.get("slug") == experiment_slug for record in preview_data):
+            break
+        time.sleep(15)
+    else:
+        pytest.fail("Experiment not published to preview in time")
+
+    summary = SummaryPage(selenium, experiment_url).open()
+    summary.back_to_draft()
+
+    while time.time() < timeout:
+        records_after = kinto_client(collection="nimbus-preview").get_record_data()
+        if len(records_after) == records_before:
+            break
+        time.sleep(15)
+    else:
+        pytest.fail("Experiment not unpublished from preview in time")