Forward load_docs_using to doc_generator for tiny doc_size callers

The default_loader path in doc_generator subtracts a ~80-byte JSON
template skeleton from doc_size and rejects anything <= 0. Tests
that intentionally use tiny doc_size (0, 1, 5, 10) trip ValueError
unless they pass load_using=self.load_docs_using so sirius_java_sdk's
SiriusJavaDocGen accepts the raw payload size directly.

Change-Id: Id5ef20e242925d90d2d7e9dcdae387ae362c8528
Reviewed-on: https://review.couchbase.org/c/TAF/+/244720
Tested-by: Build Bot <build@couchbase.com>
Reviewed-by: Balakumaran G <balakumaran.gopal@couchbase.com>

Author: shivanand-adky

pytests/bucket_param_tests/dgm.py

@@ -228,7 +228,8 @@ def check_replica_eviction():
                 mutation_type="ADD")
             read_gen = doc_generator(
                 self.key, op_index, op_index+op_batch_size,
-                key_size=self.key_size, doc_size=0)
+                key_size=self.key_size, doc_size=0,
+                load_using=self.load_docs_using)
 
             create_task = self.task.async_load_gen_docs(
                 self.cluster, bucket, create_gen, "create", 0,

pytests/epengine/basic_ops.py

@@ -727,7 +727,8 @@ def test_MB_40967(self):
         max_gets = 2500000000
         bucket = self.cluster.buckets[0]
         doc_gen = doc_generator(self.key, 0, self.num_items,
-                                doc_size=1)
+                                doc_size=1,
+                                load_using=self.load_docs_using)
         create_task = self.task.async_load_gen_docs(
             self.cluster, bucket, doc_gen, DocLoading.Bucket.DocOps.CREATE, 0,
             batch_size=100, process_concurrency=self.process_concurrency,
@@ -830,11 +831,12 @@ def get_timings(bucket_name):
         for node in kv_nodes:
             cb_stat_obj[node] = Cbstats(node)
 
-        doc_gen = doc_generator(self.key, 0, self.num_items, doc_size=1)
+        doc_gen = doc_generator(self.key, 0, self.num_items, doc_size=1,
+                                load_using=self.load_docs_using)
         create_task = self.task.async_load_gen_docs(
             self.cluster, bucket, doc_gen, DocLoading.Bucket.DocOps.CREATE, 0,
             batch_size=500, process_concurrency=self.process_concurrency,
-            timeout_secs=self.sdk_timeout)
+            timeout_secs=self.sdk_timeout, load_using=self.load_docs_using)
         self.task_manager.get_task_result(create_task)
 
         mc_stat_reset_thread = Thread(target=reset_mcstat, args=[bucket.name])
@@ -1384,7 +1386,8 @@ def test_mb_47267(self):
         small_bucket = self.cluster.buckets[1]
 
         # Big bucket docs generation
-        doc_gen = doc_generator(self.key, 0, self.num_items, doc_size=10)
+        doc_gen = doc_generator(self.key, 0, self.num_items, doc_size=10,
+                                load_using=self.load_docs_using)
         load_task = self.task.async_load_gen_docs(
             self.cluster, big_bucket, doc_gen,
             DocLoading.Bucket.DocOps.CREATE, 0,
@@ -1400,7 +1403,8 @@ def test_mb_47267(self):
         self.task_manager.get_task_result(load_task)
 
         # Small bucket docs generation
-        doc_gen_small = doc_generator(self.key, 0, 500, doc_size=10)
+        doc_gen_small = doc_generator(self.key, 0, 500, doc_size=10,
+                                      load_using=self.load_docs_using)
         load_task_2 = self.task.async_load_gen_docs(
             self.cluster, small_bucket, doc_gen_small,
             DocLoading.Bucket.DocOps.CREATE, 0,
@@ -1484,7 +1488,8 @@ def test_MB_41942(self):
                                           vbucket_type="active")
         doc_gen = doc_generator(self.key, 0, 10000,
                                 doc_size=1, vbuckets=bucket.numVBuckets,
-                                target_vbucket=active_vbs)
+                                target_vbucket=active_vbs,
+                                load_using=self.load_docs_using)
 
         # Load with doc_ttl set
         self.log.info("Setting doc_ttl=1 for %s docs" % 10000)

pytests/epengine/durability_success.py

@@ -589,7 +589,8 @@ def test_buffer_ack_during_dcp_commit(self):
             self.fail("Test requires valid durability level for sync_writes")
 
         doc_gen = doc_generator(self.key, self.num_items, self.num_items*3,
-                                key_size=10, doc_size=5)
+                                key_size=10, doc_size=5,
+                                load_using=self.load_docs_using)
         self.log.info("Loading %s keys into the bucket" % (self.num_items*2))
         load_task = self.task.async_load_gen_docs(
             self.cluster, self.bucket, doc_gen,