Synchronize COMMON_DOC_KEY with self.key across tests

Change-Id: I37b2c4d456d11cf6087a5ddd694b8c93b9a6fb61
Reviewed-on: https://review.couchbase.org/c/TAF/+/244087
Reviewed-by: Ashwin <ashwin.govindarajulu@couchbase.com>
Tested-by: <shiva.adky@couchbase.com>
Tested-by: Build Bot <build@couchbase.com>

Author: shivanand-adky

pytests/bucket_collections/collections_base.py

@@ -976,6 +976,9 @@ def over_ride_bucket_template_params(test_obj, bucket_storage,
 
     @staticmethod
     def over_ride_doc_loading_template_params(test_obj, target_spec):
+        if target_spec and "doc_crud" in target_spec:
+            target_spec["doc_crud"][MetaCrudParams.DocCrud.COMMON_DOC_KEY] = test_obj.key
+
         for key, value in test_obj.input.test_params.items():
             if key == "durability":
                 target_spec[MetaCrudParams.DURABILITY_LEVEL] = \

pytests/bucket_collections/flush_test.py

@@ -63,7 +63,7 @@ def __get_mutate_spec(doc_ttl, durability):
             # Doc loading params
             "doc_crud": {
                 MetaCrudParams.DocCrud.NUM_ITEMS_FOR_NEW_COLLECTIONS: 5000,
-                MetaCrudParams.DocCrud.COMMON_DOC_KEY: "test_collections",
+                MetaCrudParams.DocCrud.COMMON_DOC_KEY: self.key,
                 MetaCrudParams.DocCrud.CREATE_PERCENTAGE_PER_COLLECTION: 100,
             },
 
@@ -93,6 +93,7 @@ def run_collection_mutatation(self, doc_ttl, durability):
                       "doc_ttl: %s, durability: %s"
                       % (doc_ttl, durability))
         mutate_spec = self.__get_mutate_spec(doc_ttl, durability)
+        CollectionBase.over_ride_doc_loading_template_params(self, mutate_spec)
         doc_loading_task = \
             self.bucket_util.run_scenario_from_spec(
                 self.task,
@@ -133,6 +134,8 @@ def load_initial_data(self):
         self.log.info("Loading same docs back into collections")
         doc_loading_spec = \
             self.bucket_util.get_crud_template_from_package("initial_load")
+        CollectionBase.over_ride_doc_loading_template_params(
+            self, doc_loading_spec)
 
         doc_loading_task = \
             self.bucket_util.run_scenario_from_spec(
@@ -228,7 +231,7 @@ def test_flush_bucket_during_mutations(self):
             MetaCrudParams.COLLECTIONS_TO_ADD_PER_BUCKET: 0,
 
             "doc_crud": {
-                MetaCrudParams.DocCrud.COMMON_DOC_KEY: "test_collections",
+                MetaCrudParams.DocCrud.COMMON_DOC_KEY: self.key,
                 MetaCrudParams.DocCrud.CREATE_PERCENTAGE_PER_COLLECTION: 20,
                 MetaCrudParams.DocCrud.READ_PERCENTAGE_PER_COLLECTION: 20,
                 MetaCrudParams.DocCrud.UPDATE_PERCENTAGE_PER_COLLECTION: 20,
@@ -257,11 +260,13 @@ def test_flush_bucket_during_mutations(self):
 
         self.log.info("Running mutations with doc_ttl: %s, durability: %s"
                       % (doc_ttl, durability_level))
-        mutate_task = \
-            self.bucket_util.run_scenario_from_spec(
+        CollectionBase.over_ride_doc_loading_template_params(
+            self, mutate_spec)
+        mutate_task = self.bucket_util.run_scenario_from_spec(
                 self.task,
                 self.cluster,
                 self.cluster.buckets,
+
                 mutate_spec,
                 mutation_num=0,
                 async_load=True,
@@ -325,6 +330,8 @@ def test_data_post_flush(self):
             doc_ttl, durability_level = \
                 self.__get_random_doc_ttl_and_durability_level()
             mutate_spec = self.__get_mutate_spec(doc_ttl, durability_level)
+            CollectionBase.over_ride_doc_loading_template_params(
+                self, mutate_spec)
             mutate_spec[
                 MetaCrudParams.DocCrud.UPDATE_PERCENTAGE_PER_COLLECTION] = 50
             mutate_spec[

pytests/bucket_collections/rollback_tests.py

@@ -80,8 +80,9 @@ def __rewind_doc_index(self, doc_loading_task):
     def load_docs(self, doc_ops):
         load_spec = dict()
         load_spec["doc_crud"] = dict()
-        load_spec["doc_crud"][MetaCrudParams.DocCrud.COMMON_DOC_KEY] \
-            = "test_collections"
+        load_spec["doc_crud"][MetaCrudParams.DocCrud.COMMON_DOC_KEY] = self.key
+
+
         load_spec[MetaCrudParams.TARGET_VBUCKETS] = self.target_vbuckets
         load_spec[MetaCrudParams.COLLECTIONS_CONSIDERED_FOR_CRUD] = 3
         load_spec[MetaCrudParams.SCOPES_CONSIDERED_FOR_CRUD] = "all"

pytests/rebalance_new/auto_retry_failed_rebalance.py

@@ -71,6 +71,9 @@ def async_data_load(self):
         doc_loading_spec = self.bucket_util.get_crud_template_from_package(
             "volume_test_load")
         self.set_retry_exceptions(doc_loading_spec)
+        CollectionBase.over_ride_doc_loading_template_params(
+            self, doc_loading_spec)
+
         tasks = self.bucket_util.run_scenario_from_spec(
             self.task,
             self.cluster,

pytests/volumetests/Collections.py

@@ -592,6 +592,9 @@ def data_load_collection(self, async_load=True, skip_read_success_results=True):
         doc_loading_spec = \
             self.bucket_util.get_crud_template_from_package(self.data_load_spec)
         self.set_retry_exceptions(doc_loading_spec)
+        CollectionBase.over_ride_doc_loading_template_params(
+            self, doc_loading_spec)
+
         doc_loading_spec[MetaCrudParams.DURABILITY_LEVEL] = self.durability_level
         doc_loading_spec[MetaCrudParams.SKIP_READ_SUCCESS_RESULTS] = skip_read_success_results
         task = self.bucket_util.run_scenario_from_spec(self.task,