smartrack-sdk 1.0.0__py3-none-any.whl

smartrack_sdk/tests/test_callback.py

@@ -0,0 +1,295 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+"""
+回调处理器测试
+"""
+
+import unittest
+from datetime import datetime
+
+from ..callback.handler import CellEventCallbackService
+from ..models.callback import CellEventHandleRequest, CellEventHandleResponse, CellEventData
+from ..models.enums import LedColors, TriggerState, OperationResultCode
+
+
+class TestCellEventCallbackService(unittest.TestCase):
+    """储位事件回调服务测试"""
+
+    def setUp(self):
+        """测试前准备"""
+        self.service = CellEventCallbackService()
+
+    def test_handle_put_in_event_success(self):
+        """测试处理入库事件成功"""
+        # 创建请求
+        request = self._create_test_request()
+
+        # 创建入库事件数据
+        event_data = CellEventData()
+        event_data.code = OperationResultCode.SUCCESS.value
+        event_data.message = "入库成功"
+        event_data.cell_id = "A0010001"
+        event_data.cell_state = TriggerState.PUT_IN
+        event_data.grn = "PRODUCT001"
+        event_data.sheet_id = "SHEET001"
+        event_data.led_color = LedColors.GREEN
+        event_data.is_blink = False
+        event_data.rack_id = "A001"
+
+        request.add_event_data(event_data)
+
+        # 执行处理
+        response = self.service.handle(request)
+
+        # 验证结果
+        self.assertIsInstance(response, CellEventHandleResponse)
+        self.assertEqual(response.code, 0)
+        self.assertEqual(response.message, "OK")
+        self.assertEqual(response.session_id, request.session_id)
+        self.assertTrue(response.is_success)
+
+    def test_handle_take_out_event_success(self):
+        """测试处理出库事件成功"""
+        # 创建请求
+        request = self._create_test_request()
+
+        # 创建出库事件数据
+        event_data = CellEventData()
+        event_data.code = OperationResultCode.SUCCESS.value
+        event_data.message = "出库成功"
+        event_data.cell_id = "A0010001"
+        event_data.cell_state = TriggerState.TAKE_OUT
+        event_data.grn = "PRODUCT001"
+        event_data.sheet_id = "SHEET001"
+        event_data.led_color = LedColors.RED
+        event_data.is_blink = False
+        event_data.rack_id = "A001"
+
+        request.add_event_data(event_data)
+
+        # 执行处理
+        response = self.service.handle(request)
+
+        # 验证结果
+        self.assertEqual(response.code, 0)
+        self.assertTrue(response.is_success)
+
+    def test_handle_multiple_events(self):
+        """测试处理多个事件"""
+        # 创建请求
+        request = self._create_test_request()
+
+        # 第一个事件 - 入库
+        event1 = CellEventData()
+        event1.code = OperationResultCode.SUCCESS.value
+        event1.message = "入库成功"
+        event1.cell_id = "A0010001"
+        event1.cell_state = TriggerState.PUT_IN
+        event1.grn = "PRODUCT001"
+        event1.led_color = LedColors.GREEN
+        event1.is_blink = False
+
+        # 第二个事件 - 出库
+        event2 = CellEventData()
+        event2.code = OperationResultCode.SUCCESS.value
+        event2.message = "出库成功"
+        event2.cell_id = "A0010002"
+        event2.cell_state = TriggerState.TAKE_OUT
+        event2.grn = "PRODUCT002"
+        event2.led_color = LedColors.RED
+        event2.is_blink = False
+
+        request.add_event_data(event1)
+        request.add_event_data(event2)
+
+        # 执行处理
+        response = self.service.handle(request)
+
+        # 验证结果
+        self.assertEqual(response.code, 0)
+        self.assertTrue(response.is_success)
+
+    def test_handle_operation_error(self):
+        """测试处理操作错误"""
+        # 创建请求
+        request = self._create_test_request()
+
+        # 创建错误事件数据
+        event_data = CellEventData()
+        event_data.code = OperationResultCode.ILLEGAL_STOCK_IN.value
+        event_data.message = "非法上架"
+        event_data.cell_id = "A0010001"
+        event_data.cell_state = TriggerState.PUT_IN
+        event_data.grn = "PRODUCT001"
+        event_data.led_color = LedColors.RED
+        event_data.is_blink = True
+        event_data.rack_id = "A001"
+
+        request.add_event_data(event_data)
+
+        # 执行处理
+        response = self.service.handle(request)
+
+        # 验证结果 - 即使是操作错误，回调处理也应该成功
+        self.assertEqual(response.code, 0)
+        self.assertTrue(response.is_success)
+
+    def test_handle_empty_event_data(self):
+        """测试处理空事件数据"""
+        # 创建请求但不添加事件数据
+        request = self._create_test_request()
+        request.data = []
+
+        # 执行处理
+        response = self.service.handle(request)
+
+        # 验证结果
+        self.assertEqual(response.code, 100102)
+        self.assertEqual(response.message, "没有事件数据")
+        self.assertFalse(response.is_success)
+
+    def test_handle_null_event_data(self):
+        """测试处理None事件数据"""
+        # 创建请求
+        request = self._create_test_request()
+        request.data = None
+
+        # 执行处理
+        response = self.service.handle(request)
+
+        # 验证结果
+        self.assertEqual(response.code, 100102)
+        self.assertEqual(response.message, "没有事件数据")
+        self.assertFalse(response.is_success)
+
+    def test_handle_put_in_event_with_empty_grn(self):
+        """测试处理空GRN的入库事件"""
+        # 创建请求
+        request = self._create_test_request()
+
+        # 创建空GRN的事件数据
+        event_data = CellEventData()
+        event_data.code = OperationResultCode.SUCCESS.value
+        event_data.message = "入库成功"
+        event_data.cell_id = "A0010001"
+        event_data.cell_state = TriggerState.PUT_IN
+        event_data.grn = ""  # 空的GRN
+        event_data.led_color = LedColors.GREEN
+        event_data.is_blink = False
+
+        request.add_event_data(event_data)
+
+        # 执行处理
+        response = self.service.handle(request)
+
+        # 验证结果 - 应该返回错误
+        self.assertEqual(response.code, 100103)
+        self.assertIn("处理事件失败", response.message)
+        self.assertFalse(response.is_success)
+
+    def test_handle_take_out_event_with_empty_grn(self):
+        """测试处理空GRN的出库事件"""
+        # 创建请求
+        request = self._create_test_request()
+
+        # 创建空GRN的事件数据
+        event_data = CellEventData()
+        event_data.code = OperationResultCode.SUCCESS.value
+        event_data.message = "出库成功"
+        event_data.cell_id = "A0010001"
+        event_data.cell_state = TriggerState.TAKE_OUT
+        event_data.grn = None  # None的GRN
+        event_data.led_color = LedColors.RED
+        event_data.is_blink = False
+
+        request.add_event_data(event_data)
+
+        # 执行处理
+        response = self.service.handle(request)
+
+        # 验证结果 - 应该返回错误
+        self.assertEqual(response.code, 100103)
+        self.assertIn("处理事件失败", response.message)
+        self.assertFalse(response.is_success)
+
+    def test_handle_illegal_stock_in_error(self):
+        """测试处理非法上架错误"""
+        # 创建请求
+        request = self._create_test_request()
+
+        # 创建非法上架事件数据
+        event_data = CellEventData()
+        event_data.code = OperationResultCode.ILLEGAL_STOCK_IN.value
+        event_data.message = "非法上架"
+        event_data.cell_id = "A0010001"
+        event_data.cell_state = TriggerState.PUT_IN
+        event_data.grn = "PRODUCT001"
+        event_data.led_color = LedColors.RED
+        event_data.is_blink = True
+        event_data.rack_id = "A001"
+
+        request.add_event_data(event_data)
+
+        # 执行处理
+        response = self.service.handle(request)
+
+        # 验证结果 - 回调处理成功，但会记录错误日志
+        self.assertEqual(response.code, 0)
+        self.assertTrue(response.is_success)
+
+    def test_handle_illegal_stock_out_error(self):
+        """测试处理非法下架错误"""
+        # 创建请求
+        request = self._create_test_request()
+
+        # 创建非法下架事件数据
+        event_data = CellEventData()
+        event_data.code = OperationResultCode.ILLEGAL_STOCK_OUT.value
+        event_data.message = "非法下架"
+        event_data.cell_id = "A0010001"
+        event_data.cell_state = TriggerState.TAKE_OUT
+        event_data.grn = "PRODUCT001"
+        event_data.led_color = LedColors.RED
+        event_data.is_blink = True
+        event_data.rack_id = "A001"
+
+        request.add_event_data(event_data)
+
+        # 执行处理
+        response = self.service.handle(request)
+
+        # 验证结果 - 回调处理成功，但会记录错误日志
+        self.assertEqual(response.code, 0)
+        self.assertTrue(response.is_success)
+
+    def test_event_data_properties(self):
+        """测试事件数据属性"""
+        event_data = CellEventData()
+        event_data.code = OperationResultCode.SUCCESS.value
+        event_data.cell_state = TriggerState.PUT_IN
+        event_data.grn = "TEST-GRN"
+
+        # 测试属性
+        self.assertTrue(event_data.is_success)
+        self.assertTrue(event_data.is_put_in)
+        self.assertFalse(event_data.is_take_out)
+        self.assertEqual(event_data.operation_result, OperationResultCode.SUCCESS)
+
+        # 测试字符串表示
+        event_str = str(event_data)
+        self.assertIn("TEST-GRN", event_str)
+        self.assertIn("PutIn", event_str)
+
+    def _create_test_request(self) -> CellEventHandleRequest:
+        """创建测试请求"""
+        request = CellEventHandleRequest()
+        request.user_id = "test-user"
+        request.client_id = "test-client"
+        request.session_id = f"test-session-{int(datetime.now().timestamp())}"
+        request.timestamp = datetime.now().isoformat()
+        return request
+
+
+if __name__ == '__main__':
+    unittest.main()

smartrack_sdk/tests/test_client.py

@@ -0,0 +1,264 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+"""
+SmartRack客户端测试
+"""
+
+import unittest
+from unittest.mock import Mock, patch, MagicMock
+import json
+import requests
+
+from ..client.smart_rack_client import SmartRackClient
+from ..exceptions import SmartRackConfigurationException, SmartRackApiException, SmartRackNetworkException
+from ..models.requests import StockInRequest, ClearErrorRequest
+from ..models.responses import BaseResponse, ClearErrorResponse
+from ..models.enums import LedColors, RackTestMode
+
+
+class TestSmartRackClient(unittest.TestCase):
+    """SmartRack客户端测试"""
+
+    def setUp(self):
+        """测试前准备"""
+        self.base_url = "http://test.com"
+        self.client = SmartRackClient(self.base_url)
+
+    def tearDown(self):
+        """测试后清理"""
+        self.client.close()
+
+    def test_init_with_valid_url(self):
+        """测试使用有效URL初始化"""
+        client = SmartRackClient("http://example.com")
+        self.assertEqual(client.base_url, "http://example.com")
+        client.close()
+
+    def test_init_with_empty_url(self):
+        """测试使用空URL初始化"""
+        with self.assertRaises(SmartRackConfigurationException):
+            SmartRackClient("")
+
+    def test_init_with_none_url(self):
+        """测试使用None URL初始化"""
+        with self.assertRaises(SmartRackConfigurationException):
+            SmartRackClient(None)
+
+    def test_set_auth_token(self):
+        """测试设置认证Token"""
+        token = "test-token"
+        self.client.set_auth_token(token)
+        self.assertEqual(self.client.auth_token, token)
+        self.assertEqual(self.client.session.headers['Authorization'], f'Bearer {token}')
+
+    def test_clear_auth_token(self):
+        """测试清除认证Token"""
+        self.client.set_auth_token("test-token")
+        self.client.clear_auth_token()
+        self.assertIsNone(self.client.auth_token)
+        self.assertNotIn('Authorization', self.client.session.headers)
+
+    @patch('requests.Session.request')
+    def test_stock_in_request_success(self, mock_request):
+        """测试入库请求成功"""
+        # 模拟响应
+        mock_response = Mock()
+        mock_response.json.return_value = {
+            "code": 0,
+            "message": "Success",
+            "sessionId": "test-session",
+            "timestamp": "2024-01-01T00:00:00"
+        }
+        mock_response.raise_for_status.return_value = None
+        mock_request.return_value = mock_response
+
+        # 创建请求
+        request = StockInRequest("test", "MES")
+        request.set_sheet_id("IN001")
+        request.set_rack_id("A001")
+        request.set_grn("PRODUCT001")
+        request.set_led_color(LedColors.GREEN)
+        request.set_blink(True)
+
+        # 执行请求
+        response = self.client.stock_in_request(request)
+
+        # 验证结果
+        self.assertIsInstance(response, BaseResponse)
+        self.assertEqual(response.code, 0)
+        self.assertEqual(response.message, "Success")
+
+        # 验证请求调用
+        mock_request.assert_called_once()
+        call_args = mock_request.call_args
+        self.assertEqual(call_args[0][0], 'POST')  # method is first positional argument
+        self.assertIn('/api/HJWMS/External/SmartRackExternal/StockInRequest', call_args[0][1])  # url is second positional argument
+        self.assertEqual(call_args[1]['json']['userId'], 'test')
+        self.assertEqual(call_args[1]['json']['clientId'], 'MES')
+        self.assertEqual(call_args[1]['json']['sheetId'], 'IN001')
+        self.assertEqual(call_args[1]['json']['rackId'], 'A001')
+        self.assertEqual(call_args[1]['json']['grn'], 'PRODUCT001')
+        self.assertEqual(call_args[1]['json']['ledColor'], LedColors.GREEN.value)
+        self.assertTrue(call_args[1]['json']['isBlink'])
+
+    @patch('requests.Session.request')
+    def test_stock_in_request_api_error(self, mock_request):
+        """测试入库请求API错误"""
+        # 模拟API错误响应
+        mock_response = Mock()
+        mock_response.json.return_value = {
+            "code": 100101,
+            "message": "入库超时",
+            "sessionId": "test-session",
+            "timestamp": "2024-01-01T00:00:00"
+        }
+        mock_response.raise_for_status.return_value = None
+        mock_request.return_value = mock_response
+
+        # 创建请求
+        request = StockInRequest("test", "MES")
+        request.set_sheet_id("IN001")
+        request.set_rack_id("A001")
+        request.set_grn("PRODUCT001")
+
+        # 执行请求并验证异常
+        with self.assertRaises(SmartRackApiException) as context:
+            self.client.stock_in_request(request)
+
+        self.assertEqual(context.exception.api_code, 100101)
+        self.assertEqual(context.exception.message, "入库超时")
+        self.assertEqual(context.exception.session_id, "test-session")
+
+    @patch('requests.Session.request')
+    def test_stock_in_request_network_error(self, mock_request):
+        """测试入库请求网络错误"""
+        # 模拟网络错误
+        mock_request.side_effect = requests.exceptions.ConnectionError("Connection failed")
+
+        # 创建请求
+        request = StockInRequest("test", "MES")
+        request.set_sheet_id("IN001")
+        request.set_rack_id("A001")
+        request.set_grn("PRODUCT001")
+
+        # 执行请求并验证异常
+        with self.assertRaises(SmartRackNetworkException) as context:
+            self.client.stock_in_request(request)
+
+        self.assertIn("连接错误", context.exception.message)
+
+    @patch('requests.Session.request')
+    def test_clear_error_success(self, mock_request):
+        """测试清错请求成功"""
+        # 模拟响应
+        mock_response = Mock()
+        mock_response.json.return_value = {
+            "code": 0,
+            "message": "Success",
+            "sessionId": "test-session",
+            "timestamp": "2024-01-01T00:00:00",
+            "resultData": [
+                {
+                    "grn": "PRODUCT001",
+                    "cell": "A0010001",
+                    "state": 0
+                }
+            ]
+        }
+        mock_response.raise_for_status.return_value = None
+        mock_request.return_value = mock_response
+
+        # 创建请求
+        request = ClearErrorRequest("test", "MES")
+        request.set_cell_id("A0010001")
+
+        # 执行请求
+        response = self.client.clear_error(request)
+
+        # 验证结果
+        self.assertIsInstance(response, ClearErrorResponse)
+        self.assertEqual(response.code, 0)
+        self.assertEqual(response.message, "Success")
+        self.assertEqual(len(response.result_data), 1)
+        self.assertEqual(response.result_data[0].grn, "PRODUCT001")
+        self.assertEqual(response.result_data[0].cell, "A0010001")
+        self.assertEqual(response.result_data[0].state, 0)
+
+    @patch('requests.Session.request')
+    def test_get_stock_data_success(self, mock_request):
+        """测试查询库存成功"""
+        # 模拟响应
+        mock_response = Mock()
+        mock_response.json.return_value = {
+            "code": 0,
+            "message": "Success",
+            "sessionId": "test-session",
+            "timestamp": "2024-01-01T00:00:00",
+            "resultData": [
+                {
+                    "rackId": "A001",
+                    "grn": "PRODUCT001",
+                    "cellId": "A0010001",
+                    "ledColor": LedColors.GREEN.value,
+                    "isBlink": False,
+                    "status": 1
+                }
+            ],
+            "total": 1
+        }
+        mock_response.raise_for_status.return_value = None
+        mock_request.return_value = mock_response
+
+        # 执行请求
+        response = self.client.get_stock_data(
+            grn="PRODUCT001",
+            rack_id="A001",
+            cell_id="A0010001"
+        )
+
+        # 验证结果
+        self.assertEqual(response.code, 0)
+        self.assertEqual(response.total, 1)
+        self.assertEqual(len(response.result_data), 1)
+        self.assertEqual(response.result_data[0].rack_id, "A001")
+        self.assertEqual(response.result_data[0].grn, "PRODUCT001")
+        self.assertEqual(response.result_data[0].cell_id, "A0010001")
+
+        # 验证请求参数
+        call_args = mock_request.call_args
+        self.assertEqual(call_args[0][0], 'GET')  # method is first positional argument
+        params = call_args[1]['params']
+        self.assertEqual(params['GRN'], 'PRODUCT001')
+        self.assertEqual(params['RackId'], 'A001')
+        self.assertEqual(params['CellId'], 'A0010001')
+
+    def test_context_manager(self):
+        """测试上下文管理器"""
+        with SmartRackClient("http://test.com") as client:
+            self.assertEqual(client.base_url, "http://test.com")
+        # 客户端应该自动关闭
+
+    def test_validation_error_empty_sheet_id(self):
+        """测试验证错误 - 空的入库单号"""
+        request = StockInRequest("test", "MES")
+        # 不设置sheet_id
+
+        with self.assertRaises(Exception) as context:
+            self.client.stock_in_request(request)
+        self.assertIn("sheetId", str(context.exception))
+
+    def test_validation_error_empty_grn_list(self):
+        """测试验证错误 - 空的GRN列表"""
+        from ..models.requests import StockOutRequest
+
+        request = StockOutRequest("test", "MES")
+        # 不添加GRN到列表
+
+        with self.assertRaises(Exception) as context:
+            self.client.stock_out_request(request)
+        self.assertIn("grnList", str(context.exception))
+
+
+if __name__ == '__main__':
+    unittest.main()