Bug fix to address work order receipt flow in work order sync mode execution.

Signed-off-by: Ramakrishna Srinivasamurthy <[email protected]>

Author: Ram-srini

enclave_manager/avalon_enclave_manager/singleton/enclave_manager.py

@@ -299,51 +299,52 @@ def __update_receipt(self, kv_helper, wo_id, wo_json_resp):
             the receipt.
         """
         receipt_entry = kv_helper.get("wo-receipts", wo_id)
-        if receipt_entry:
-            update_type = None
-            if "error" in wo_json_resp and \
-                    wo_json_resp["error"]["code"] != \
-                    WorkOrderStatus.PENDING.value:
-                update_type = ReceiptCreateStatus.FAILED.value
-            else:
-                update_type = ReceiptCreateStatus.PROCESSED.value
-            receipt_obj = WorkOrderReceiptRequest()
-            wo_receipt = receipt_obj.update_receipt(
-                wo_id,
-                update_type,
-                wo_json_resp,
-                self.private_key
-            )
-            updated_receipt = None
-            # load previous updates to receipt
-            updates_to_receipt = kv_helper.get("wo-receipt-updates", wo_id)
-            # If it is first update to receipt
-            if updates_to_receipt is None:
-                updated_receipt = []
-            else:
-                updated_receipt = json.loads(updates_to_receipt)
-                # Get the last update to receipt
-                last_receipt = updated_receipt[len(updated_receipt) - 1]
-
-                # If receipt updateType is completed,
-                # then no further update allowed
-                if last_receipt["updateType"] == \
-                        ReceiptCreateStatus.COMPLETED.value:
-                    logger.info(
-                        "Receipt for the workorder id %s is completed " +
-                        "and no further updates are allowed",
-                        wo_id)
-                    return
-            updated_receipt.append(wo_receipt)
-
-            # Since receipts_json is jrpc request updating only params object.
-            kv_helper.set("wo-receipt-updates", wo_id, json.dumps(
-                updated_receipt))
-            logger.info("Receipt for the workorder id %s is updated to %s",
-                        wo_id, wo_receipt)
+        # If receipt is not created yet, add tag "receiptUpdates" to
+        # Receipt entry and update it
+        if receipt_entry is None:
+            receipt_entry = {
+                "params": {
+                    "receiptUpdates": []
+                }
+            }
+        # load previous updates to receipt
+        receipt_update_entry = receipt_entry["params"]["receiptUpdates"]
+        update_type = None
+        if "error" in wo_json_resp and \
+                wo_json_resp["error"]["code"] != \
+                WorkOrderStatus.PENDING.value:
+            update_type = ReceiptCreateStatus.FAILED.value
         else:
-            logger.info("Work order receipt is not created, " +
-                        "so skipping the update")
+            update_type = ReceiptCreateStatus.PROCESSED.value
+        receipt_obj = WorkOrderReceiptRequest()
+        wo_receipt = receipt_obj.update_receipt(
+            wo_id,
+            update_type,
+            wo_json_resp,
+            self.private_key
+        )
+
+        # If it is first update to receipt
+        if len(receipt_update_entry) > 0:
+            # Get the last update to receipt
+            last_receipt = receipt_update_entry[len(receipt_update_entry) - 1]
+            # If receipt updateType is completed,
+            # then no further update allowed
+            if last_receipt["updateType"] == \
+                    ReceiptCreateStatus.COMPLETED.value:
+                logger.info(
+                    "Receipt for the workorder id %s is completed " +
+                    "and no further updates are allowed",
+                    wo_id)
+                return
+        receipt_update_entry.append(wo_receipt)
+
+        # Since receipts_json is jrpc request updating only params object.
+        receipt_entry["receiptUpdates"] = receipt_update_entry
+        kv_helper.set("wo-receipts", wo_id, json.dumps(
+            receipt_entry))
+        logger.info("Receipt for the workorder id %s is updated to %s",
+                    wo_id, wo_receipt)
 
 
 # -----------------------------------------------------------------

listener/avalon_listener/tcs_workorder_receipt_handler.py

@@ -72,7 +72,7 @@ def WorkOrderReceiptCreate(self, **params):
         wo_id = params["workOrderId"]
         input_json_str = params["raw"]
         input_value = json.loads(input_json_str)
-
+        # Check if work order id exists
         wo_request = self.kv_helper.get("wo-requests", wo_id)
         if wo_request is None:
             raise JSONRPCDispatchException(
@@ -82,12 +82,22 @@ def WorkOrderReceiptCreate(self, **params):
             )
         else:
             wo_receipt = self.kv_helper.get("wo-receipts", wo_id)
-            if wo_receipt is None:
+            wo_receipt = json.loads(wo_receipt)
+            if wo_receipt and "workOrderId" not in wo_receipt["params"]:
                 status, err_msg = \
                     self.__validate_work_order_receipt_create_req(
                         input_value, wo_request)
                 if status is True:
-                    self.kv_helper.set("wo-receipts", wo_id, input_json_str)
+                    # If receiptUpdates doesn't exists then
+                    # create an entry with empty array
+                    if "receiptUpdates" not in wo_receipt["params"]:
+                        input_value["params"]["receiptUpdates"] = []
+                    else:
+                        input_value["params"]["receiptUpdates"] = \
+                            wo_receipt["receiptUpdates"]
+                    self.kv_helper.set("wo-receipts", wo_id, json.dumps(
+                        input_value
+                    ))
                     raise JSONRPCDispatchException(
                         JRPCErrorCodes.SUCCESS,
                         "Receipt created successfully"
@@ -186,11 +196,14 @@ def WorkOrderReceiptUpdate(self, **params):
             status, err_msg = self.__validate_work_order_receipt_update_req(
                 input_value)
             if status is True:
+                value = json.loads(value)
                 # Load previous updates to receipt
-                updates_to_receipt = \
-                    self.kv_helper.get("wo-receipt-updates", wo_id)
+                if "receiptUpdates" in value["params"]:
+                    updates_to_receipt = value["params"]["receiptUpdates"]
+                else:
+                    updates_to_receipt = []
                 # If it is first update to receipt
-                if updates_to_receipt is None:
+                if len(updates_to_receipt) == 0:
                     updated_receipt = []
                 else:
                     updated_receipt = json.loads(updates_to_receipt)
@@ -220,8 +233,9 @@ def WorkOrderReceiptUpdate(self, **params):
                             " is not allowed"
                         )
                 updated_receipt.append(input_value)
-                self.kv_helper.set("wo-receipt-updates", wo_id,
-                                   json.dumps(updated_receipt))
+                input_value["receiptUpdates"] = updated_receipt
+                self.kv_helper.set("wo-receipts", wo_id,
+                                   json.dumps(input_value))
                 raise JSONRPCDispatchException(
                     JRPCErrorCodes.SUCCESS,
                     "Receipt updated successfully"
@@ -377,15 +391,19 @@ def WorkOrderReceiptRetrieve(self, **params):
         value = self.kv_helper.get("wo-receipts", wo_id)
         if value:
             receipt = json.loads(value)
-            receipt_updates = self.kv_helper.get("wo-receipt-updates", wo_id)
-            if receipt_updates is None:
+            if "receiptUpdates" in receipt["params"]:
+                receipt_updates = receipt["params"]["receiptUpdates"]
+            else:
+                receipt_updates = []
+            if len(receipt_updates) == 0:
+                # If there is no updates to receipt
+                # then current status is same as create status
                 receipt["params"]["receiptCurrentStatus"] = \
                     receipt["params"]["receiptCreateStatus"]
             else:
-                receipt_updates_json = json.loads(receipt_updates)
-                # Get the recent update to receipt
-                last_receipt = receipt_updates_json[len(receipt_updates_json)
-                                                    - 1]
+                # Get the latest update to receipt
+                last_receipt = receipt_updates[len(receipt_updates)
+                                               - 1]
                 receipt["params"]["receiptCurrentStatus"] = \
                     last_receipt["updateType"]
             return receipt["params"]
@@ -420,10 +438,12 @@ def WorkOrderReceiptUpdateRetrieve(self, **params):
         # starts from 1
         update_index = input_params["updateIndex"]
         # Load list of updates to the receipt
-        receipt_updates = self.kv_helper.get("wo-receipt-updates", wo_id)
+        receipt_entry = self.kv_helper.get("wo-receipts", wo_id)
 
-        if receipt_updates:
-            receipt_updates_json = json.loads(receipt_updates)
+        if receipt_entry:
+            receipt_entry_json = json.loads(receipt_entry)
+            receipt_updates_json = \
+                receipt_entry_json["params"]["receiptUpdates"]
             total_updates = len(receipt_updates_json)
             if update_index <= 0:
                 raise JSONRPCDispatchException(

tools/run_tests.sh

@@ -99,11 +99,10 @@ done
 
 #yell "Start testing echo client with reading registry from blockchain................"
 #yell "#------------------------------------------------------------------------------------------------"
-#try $echo_client_path/echo_client.py -m "Hello world" -rs -dh
 
 yell "Start testing echo client with service uri ................"
 yell "#------------------------------------------------------------------------------------------------"
-try $echo_client_path/echo_client.py -m "Hello world" -s "http://$LISTENER_URL:1947" -dh
+try $echo_client_path/echo_client.py -m "Hello world" -s "http://$LISTENER_URL:1947" -dh -rs
 
 yell "Start testing generic client for echo workload ................"
 yell "#------------------------------------------------------------------------------------------------"