From 2af21bc4eed5e07c121b0911c9b2148f0240d9e3 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E8=83=A1=E5=B0=A7?= <huyao@jikimo.com>
Date: Mon, 21 Oct 2024 11:37:01 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9=E5=8D=95=E5=85=83=E6=B5=8B?=
 =?UTF-8?q?=E8=AF=95?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../controllers/main.py                       |  2 +-
 .../test_jikimo_workorder_exception_notify.py | 55 ++++++++++++++++++-
 2 files changed, 54 insertions(+), 3 deletions(-)

diff --git a/jikimo_workorder_exception/controllers/main.py b/jikimo_workorder_exception/controllers/main.py
index 6a17f958..0bd3d762 100644
--- a/jikimo_workorder_exception/controllers/main.py
+++ b/jikimo_workorder_exception/controllers/main.py
@@ -21,7 +21,7 @@ class WorkorderExceptionConroller(http.Controller):
         try:
             res = {'Succeed': True, 'ErrorCode': 0, 'Error': ''}
             datas = request.httprequest.data
-            ret = json.loads(datas)['Datas']
+            ret = json.loads(datas)
             if not ret.get('RfidCode') or not ret.get('ErrorType'):
                 res = {'Succeed': False, 'ErrorCode': 400, 'Error': '参数错误'}
                 return json.JSONEncoder().encode(res)
diff --git a/jikimo_workorder_exception_notify/tests/test_jikimo_workorder_exception_notify.py b/jikimo_workorder_exception_notify/tests/test_jikimo_workorder_exception_notify.py
index 85d304be..92886ff3 100644
--- a/jikimo_workorder_exception_notify/tests/test_jikimo_workorder_exception_notify.py
+++ b/jikimo_workorder_exception_notify/tests/test_jikimo_workorder_exception_notify.py
@@ -14,12 +14,25 @@ class TestJikimoWorkorderExceptionNotify(TestJikimoWorkorderExceptionNotifyCommo
             ('model', '=', 'jikimo.workorder.exception')
         ]))
         self.assertTrue(self.env['jikimo.message.template'].search([
-            ('name', '=', '加工失败'),
+            ('name', '=', '工单加工失败提醒'),
             ('model', '=', 'jikimo.workorder.exception')
         ]))
 
+    def test_create_message_queue_yc0001(self):
+        exception_record = self.env['jikimo.workorder.exception'].create({
+            'workorder_id': self.workorder.id,
+            'exception_code': 'YC0001',
+            'exception_content': '无CNC程序'
+        })
 
-    def test_create_message_queue(self):
+        message_record = self.env['jikimo.message.queue'].search([
+            ('res_id', '=', exception_record.id),
+            ('model', '=', 'jikimo.workorder.exception'),
+            ('message_status', '=', 'pending')
+        ])
+        self.assertFalse(message_record)
+
+    def test_create_message_queue_yc0002(self):
         exception_record = self.env['jikimo.workorder.exception'].create({
             'workorder_id': self.workorder.id,
             'exception_code': 'YC0002',
@@ -43,7 +56,45 @@ class TestJikimoWorkorderExceptionNotify(TestJikimoWorkorderExceptionNotifyCommo
             ('message_template_id', '=', message_template.id)
         ])
         self.assertTrue(message_record)
+
+    def test_create_message_queue_yc0003(self):
+        exception_record = self.env['jikimo.workorder.exception'].create({
+            'workorder_id': self.workorder.id,
+            'exception_code': 'YC0003',
+            'exception_content': '无定位数据'
+        })
+
+        bussiness_node = self.env['jikimo.message.bussiness.node'].search([
+            ('name', '=', '无定位数据'),
+            ('model', '=', 'jikimo.workorder.exception')
+        ])
+
+        message_template = self.env['jikimo.message.template'].search([
+            ('bussiness_node_id', '=', bussiness_node.id),
+            ('model', '=', 'jikimo.workorder.exception')
+        ])
+
+        message_record = self.env['jikimo.message.queue'].search([
+            ('res_id', '=', exception_record.id),
+            ('model', '=', 'jikimo.workorder.exception'),
+            ('message_status', '=', 'pending'),
+            ('message_template_id', '=', message_template.id)
+        ])
+        self.assertTrue(message_record)
     
+    def test_create_message_queue_yc0004(self):
+        exception_record = self.env['jikimo.workorder.exception'].create({
+            'workorder_id': self.workorder.id,
+            'exception_code': 'YC0004',
+            'exception_content': '无CNC程序'
+        })
+
+        message_record = self.env['jikimo.message.queue'].search([
+            ('res_id', '=', exception_record.id),
+            ('model', '=', 'jikimo.workorder.exception'),
+            ('message_status', '=', 'pending')
+        ])
+        self.assertFalse(message_record)
 
     def test_get_message(self):
         exception_record = self.env['jikimo.workorder.exception'].create({