Add total requests field to tests

Author: johnugeorge

pkg/suggestion/v1beta1/chocolate/base_service.py

@@ -157,7 +157,8 @@ def getSuggestions(self, trials, request_number, total_request_number):
                     logger.info("New record in sqlite DB is updated")
                     logger.info("{}\n".format(
                         self.created_trials[new_trial_loss_idx]))
-
+        # Assuming that created_trials are already populated
+        # TODO: Handle Restart of algorithm pod
         logger.info("{} Trials created in DB".format(len(self.created_trials)))
         new_actual_requested_no = total_request_number - len(self.created_trials)
         prev_generated_no = request_number - new_actual_requested_no

test/suggestion/v1beta1/test_chocolate_service.py

@@ -169,6 +169,7 @@ def test_get_suggestion(self):
             experiment=experiment,
             trials=trials,
             request_number=2,
+            total_request_number=2,
         )
 
         get_suggestion = self.test_server.invoke_unary_unary(