forked from huangxin/ailai
34 lines
1.1 KiB
Python
34 lines
1.1 KiB
Python
#!/usr/bin/env python3
|
||
import unittest
|
||
from unittest.mock import patch, MagicMock
|
||
|
||
# 假设你的模块名为 CU，Catch 定义在 CU/Catch.py
|
||
from CU.Catch import Catch, CatchStatus
|
||
|
||
class TestCatchRelayControl(unittest.TestCase):
|
||
|
||
@patch('CU.EMV.close') # 只mock close函数，因为我们只关心关闭电磁阀1
|
||
def test_catch_status_ctake_calls_close_relay(self, mock_close):
|
||
# 创建一个 mock 的 RobotClient（不需要真实实现）
|
||
mock_robot_client = MagicMock()
|
||
mock_robot_client.time_delay_put = 0.5
|
||
mock_robot_client.con_ios = [0, 1, 2]
|
||
|
||
# 实例化 Catch 类
|
||
catch_instance = Catch(mock_robot_client)
|
||
|
||
# 设置状态为 CTake
|
||
catch_instance.catch_status = CatchStatus.CTake
|
||
|
||
# 第一次 run，应该触发 close(1, 0, 0)
|
||
catch_instance.run()
|
||
|
||
# 验证 close 是否被调用，并且是 close(1, 0, 0)
|
||
mock_close.assert_called_once_with(1, 0, 0)
|
||
|
||
# 再次 run，不应再次调用 close
|
||
catch_instance.run()
|
||
self.assertEqual(mock_close.call_count, 1) # 确保只调用了一次
|
||
|
||
if __name__ == '__main__':
|
||
unittest.main() |