429 lines
16 KiB
Python
429 lines
16 KiB
Python
#!/usr/bin/env python
|
||
# -*- coding: utf-8 -*-
|
||
|
||
"""
|
||
OPC UA模块测试
|
||
测试VWED OPC UA内置函数模块的功能
|
||
"""
|
||
|
||
import sys
|
||
import os
|
||
import time
|
||
import threading
|
||
import unittest
|
||
|
||
# 添加项目根目录到路径
|
||
sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..'))
|
||
|
||
from services.online_script.built_in_modules.opc_ua_module import VWEDOpcUaModule
|
||
from tests.test_opc_ua_server import OpcUaTestServer
|
||
|
||
|
||
class TestOpcUaModule(unittest.TestCase):
|
||
"""OPC UA模块测试类"""
|
||
|
||
@classmethod
|
||
def setUpClass(cls):
|
||
"""设置测试类 - 启动OPC UA服务器"""
|
||
print("\n" + "="*60)
|
||
print("启动OPC UA测试服务器...")
|
||
print("="*60)
|
||
|
||
# 启动测试服务器
|
||
cls.server = OpcUaTestServer(host='localhost', port=4840)
|
||
cls.server.start_server()
|
||
|
||
# 等待服务器完全启动
|
||
time.sleep(2)
|
||
|
||
# 获取服务器信息
|
||
cls.server_info = cls.server.get_server_info()
|
||
cls.endpoint = cls.server_info['endpoint']
|
||
cls.namespace_idx = cls.server_info['namespace_index']
|
||
|
||
print(f"服务器端点: {cls.endpoint}")
|
||
print(f"命名空间索引: {cls.namespace_idx}")
|
||
|
||
# 显示测试节点信息
|
||
node_info = cls.server.get_node_info()
|
||
if node_info:
|
||
print("\n可用测试节点:")
|
||
for name, info in node_info.items():
|
||
print(f" {name}: ns={info['namespace_index']};{info['node_id']} = {info['value']}")
|
||
|
||
@classmethod
|
||
def tearDownClass(cls):
|
||
"""清理测试类 - 停止OPC UA服务器"""
|
||
print("\n" + "="*60)
|
||
print("停止OPC UA测试服务器...")
|
||
print("="*60)
|
||
|
||
if hasattr(cls, 'server'):
|
||
cls.server.stop_server()
|
||
time.sleep(1)
|
||
|
||
print("测试完成")
|
||
|
||
def setUp(self):
|
||
"""设置测试方法"""
|
||
self.opc_module = VWEDOpcUaModule("test_script")
|
||
|
||
def tearDown(self):
|
||
"""清理测试方法"""
|
||
if hasattr(self, 'opc_module'):
|
||
self.opc_module.cleanup()
|
||
|
||
def test_read_opc_value_simple(self):
|
||
"""测试简单读取OPC值（使用默认命名空间）"""
|
||
print("\n测试: 简单读取OPC值")
|
||
|
||
# 注意：由于我们的测试节点都在自定义命名空间中，这个测试可能会失败
|
||
# 这是预期的，因为默认命名空间(0)中通常没有我们的测试节点
|
||
try:
|
||
# 尝试读取标准的服务器状态节点
|
||
value = self.opc_module.read_opc_value("i=2259", self.endpoint) # ServerStatus.State
|
||
print(f"读取服务器状态: {value}")
|
||
self.assertIsNotNone(value)
|
||
except RuntimeError as e:
|
||
print(f"预期的错误（默认命名空间中没有测试节点）: {e}")
|
||
self.assertEqual(str(e), "read_opc_value error")
|
||
|
||
def test_read_opc_value_with_namespace(self):
|
||
"""测试带命名空间的OPC值读取"""
|
||
print("\n测试: 带命名空间的OPC值读取")
|
||
|
||
# 测试读取整数节点
|
||
try:
|
||
value = self.opc_module.read_opc_value_with_namespace(
|
||
self.namespace_idx, "i=1001", self.endpoint
|
||
)
|
||
print(f"读取整数节点: ns={self.namespace_idx};i=1001 = {value}")
|
||
self.assertIsNotNone(value)
|
||
self.assertEqual(value, 1001)
|
||
except Exception as e:
|
||
self.fail(f"读取整数节点失败: {e}")
|
||
|
||
# 测试读取字符串节点
|
||
try:
|
||
value = self.opc_module.read_opc_value_with_namespace(
|
||
self.namespace_idx, "s=StringNodeId", self.endpoint
|
||
)
|
||
print(f"读取字符串节点: ns={self.namespace_idx};s=StringNodeId = {value}")
|
||
self.assertIsNotNone(value)
|
||
self.assertEqual(value, "String Node Value")
|
||
except Exception as e:
|
||
self.fail(f"读取字符串节点失败: {e}")
|
||
|
||
# 测试读取GUID节点
|
||
try:
|
||
value = self.opc_module.read_opc_value_with_namespace(
|
||
self.namespace_idx, "g=550e8400-e29b-41d4-a716-446655440000", self.endpoint
|
||
)
|
||
print(f"读取GUID节点: ns={self.namespace_idx};g=550e8400-e29b-41d4-a716-446655440000 = {value}")
|
||
self.assertIsNotNone(value)
|
||
self.assertEqual(value, "GUID Node Value")
|
||
except Exception as e:
|
||
self.fail(f"读取GUID节点失败: {e}")
|
||
|
||
def test_read_dynamic_values(self):
|
||
"""测试读取动态变化的值"""
|
||
print("\n测试: 读取动态变化的值")
|
||
|
||
# 读取计数器值
|
||
try:
|
||
value1 = self.opc_module.read_opc_value_with_namespace(
|
||
self.namespace_idx, "Counter", self.endpoint
|
||
)
|
||
print(f"第一次读取计数器: {value1}")
|
||
|
||
# 等待一段时间
|
||
time.sleep(2)
|
||
|
||
value2 = self.opc_module.read_opc_value_with_namespace(
|
||
self.namespace_idx, "Counter", self.endpoint
|
||
)
|
||
print(f"第二次读取计数器: {value2}")
|
||
|
||
# 验证值发生了变化
|
||
self.assertNotEqual(value1, value2)
|
||
|
||
except Exception as e:
|
||
self.fail(f"读取动态值失败: {e}")
|
||
|
||
def test_write_opc_value(self):
|
||
"""测试写入OPC值"""
|
||
print("\n测试: 写入OPC值")
|
||
|
||
# 测试写入整数值
|
||
try:
|
||
# 先读取当前值
|
||
original_value = self.opc_module.read_opc_value_with_namespace(
|
||
self.namespace_idx, "i=1001", self.endpoint
|
||
)
|
||
print(f"原始值: {original_value}")
|
||
|
||
# 写入新值
|
||
new_value = 9999
|
||
result = self.opc_module.write_opc_value_with_namespace(
|
||
self.namespace_idx, "i=1001", new_value, self.endpoint
|
||
)
|
||
print(f"写入结果: {result}")
|
||
self.assertTrue(result)
|
||
|
||
# 读取验证
|
||
written_value = self.opc_module.read_opc_value_with_namespace(
|
||
self.namespace_idx, "i=1001", self.endpoint
|
||
)
|
||
print(f"写入后的值: {written_value}")
|
||
self.assertEqual(written_value, new_value)
|
||
|
||
# 恢复原始值
|
||
self.opc_module.write_opc_value_with_namespace(
|
||
self.namespace_idx, "i=1001", original_value, self.endpoint
|
||
)
|
||
|
||
except Exception as e:
|
||
self.fail(f"写入值失败: {e}")
|
||
|
||
def test_write_different_types(self):
|
||
"""测试写入不同类型的值"""
|
||
print("\n测试: 写入不同类型的值")
|
||
|
||
test_cases = [
|
||
("BooleanVariable", True, False),
|
||
("IntegerVariable", 42, 100),
|
||
("FloatVariable", 3.14159, 2.71828),
|
||
("StringVariable", "Hello VWED", "Hello Test"),
|
||
]
|
||
|
||
for node_name, original_expected, new_value in test_cases:
|
||
try:
|
||
print(f"\n测试节点: {node_name}")
|
||
|
||
# 读取当前值
|
||
current_value = self.opc_module.read_opc_value_with_namespace(
|
||
self.namespace_idx, node_name, self.endpoint
|
||
)
|
||
print(f" 当前值: {current_value} ({type(current_value).__name__})")
|
||
|
||
# 写入新值
|
||
result = self.opc_module.write_opc_value_with_namespace(
|
||
self.namespace_idx, node_name, new_value, self.endpoint
|
||
)
|
||
print(f" 写入 {new_value} 结果: {result}")
|
||
self.assertTrue(result)
|
||
|
||
# 验证写入
|
||
written_value = self.opc_module.read_opc_value_with_namespace(
|
||
self.namespace_idx, node_name, self.endpoint
|
||
)
|
||
print(f" 写入后的值: {written_value} ({type(written_value).__name__})")
|
||
self.assertEqual(written_value, new_value)
|
||
|
||
# 恢复原始值（如果知道的话）
|
||
if current_value is not None:
|
||
self.opc_module.write_opc_value_with_namespace(
|
||
self.namespace_idx, node_name, current_value, self.endpoint
|
||
)
|
||
|
||
except Exception as e:
|
||
print(f" 测试 {node_name} 失败: {e}")
|
||
|
||
def test_subscription_read(self):
|
||
"""测试订阅方式读取值"""
|
||
print("\n测试: 订阅方式读取值")
|
||
|
||
try:
|
||
# 第一次读取（创建订阅）
|
||
value1 = self.opc_module.read_opc_value_by_subscription(
|
||
self.namespace_idx, "Counter", self.endpoint
|
||
)
|
||
print(f"订阅读取计数器（第一次）: {value1}")
|
||
self.assertIsNotNone(value1)
|
||
|
||
# 等待一段时间让值发生变化
|
||
time.sleep(3)
|
||
|
||
# 第二次读取（使用已有订阅）
|
||
value2 = self.opc_module.read_opc_value_by_subscription(
|
||
self.namespace_idx, "Counter", self.endpoint
|
||
)
|
||
print(f"订阅读取计数器（第二次）: {value2}")
|
||
self.assertIsNotNone(value2)
|
||
|
||
# 验证值发生了变化（或至少订阅在工作）
|
||
print(f"值变化: {value1} -> {value2}")
|
||
|
||
except Exception as e:
|
||
self.fail(f"订阅读取失败: {e}")
|
||
|
||
def test_error_handling(self):
|
||
"""测试错误处理"""
|
||
print("\n测试: 错误处理")
|
||
|
||
# 测试连接到不存在的服务器
|
||
try:
|
||
self.opc_module.read_opc_value_with_namespace(
|
||
0, "i=1", "opc.tcp://nonexistent:4840"
|
||
)
|
||
self.fail("应该抛出RuntimeError异常")
|
||
except RuntimeError as e:
|
||
print(f"预期的连接错误: {e}")
|
||
self.assertEqual(str(e), "read_opc_value error")
|
||
|
||
# 测试读取不存在的节点
|
||
try:
|
||
self.opc_module.read_opc_value_with_namespace(
|
||
self.namespace_idx, "i=99999", self.endpoint
|
||
)
|
||
self.fail("应该抛出RuntimeError异常")
|
||
except RuntimeError as e:
|
||
print(f"预期的节点不存在错误: {e}")
|
||
self.assertEqual(str(e), "read_opc_value error")
|
||
|
||
def test_cleanup(self):
|
||
"""测试资源清理"""
|
||
print("\n测试: 资源清理")
|
||
|
||
# 创建一些连接和订阅
|
||
try:
|
||
self.opc_module.read_opc_value_with_namespace(
|
||
self.namespace_idx, "i=1001", self.endpoint
|
||
)
|
||
self.opc_module.read_opc_value_by_subscription(
|
||
self.namespace_idx, "Counter", self.endpoint
|
||
)
|
||
|
||
# 验证有连接和订阅
|
||
self.assertGreater(len(self.opc_module._clients), 0)
|
||
self.assertGreater(len(self.opc_module._subscriptions), 0)
|
||
|
||
# 清理资源
|
||
self.opc_module.cleanup()
|
||
|
||
# 验证资源已清理
|
||
self.assertEqual(len(self.opc_module._clients), 0)
|
||
self.assertEqual(len(self.opc_module._subscriptions), 0)
|
||
self.assertEqual(len(self.opc_module._subscription_values), 0)
|
||
|
||
print("资源清理成功")
|
||
|
||
except Exception as e:
|
||
self.fail(f"资源清理失败: {e}")
|
||
|
||
|
||
def run_interactive_test():
|
||
"""运行交互式测试"""
|
||
print("\n" + "="*60)
|
||
print("OPC UA模块交互式测试")
|
||
print("="*60)
|
||
|
||
# 启动服务器
|
||
server = OpcUaTestServer(host='localhost', port=4840)
|
||
server.start_server()
|
||
time.sleep(2)
|
||
|
||
# 创建模块实例
|
||
opc_module = VWEDOpcUaModule("interactive_test")
|
||
|
||
server_info = server.get_server_info()
|
||
endpoint = server_info['endpoint']
|
||
namespace_idx = server_info['namespace_index']
|
||
|
||
print(f"服务器端点: {endpoint}")
|
||
print(f"命名空间索引: {namespace_idx}")
|
||
|
||
try:
|
||
# 显示可用节点
|
||
node_info = server.get_node_info()
|
||
print(f"\n可用测试节点:")
|
||
for name, info in node_info.items():
|
||
print(f" {name}: ns={info['namespace_index']};{info['node_id']} = {info['value']}")
|
||
|
||
print(f"\n开始交互式测试...")
|
||
|
||
# 测试各种操作
|
||
print(f"\n1. 读取各种类型的节点:")
|
||
test_reads = [
|
||
("i=1001", "整数节点"),
|
||
("s=StringNodeId", "字符串节点"),
|
||
("BooleanVariable", "布尔变量"),
|
||
("Temperature", "温度传感器"),
|
||
("Counter", "计数器"),
|
||
]
|
||
|
||
for node_id, description in test_reads:
|
||
try:
|
||
value = opc_module.read_opc_value_with_namespace(namespace_idx, node_id, endpoint)
|
||
print(f" {description} ({node_id}): {value}")
|
||
except Exception as e:
|
||
print(f" {description} ({node_id}): 读取失败 - {e}")
|
||
|
||
print(f"\n2. 测试写入操作:")
|
||
# 写入整数值
|
||
try:
|
||
result = opc_module.write_opc_value_with_namespace(namespace_idx, "i=1001", 8888, endpoint)
|
||
print(f" 写入整数节点 8888: {result}")
|
||
|
||
# 验证写入
|
||
value = opc_module.read_opc_value_with_namespace(namespace_idx, "i=1001", endpoint)
|
||
print(f" 读取验证: {value}")
|
||
except Exception as e:
|
||
print(f" 写入失败: {e}")
|
||
|
||
print(f"\n3. 测试订阅读取:")
|
||
try:
|
||
value1 = opc_module.read_opc_value_by_subscription(namespace_idx, "Counter", endpoint)
|
||
print(f" 订阅计数器（第一次）: {value1}")
|
||
|
||
time.sleep(3)
|
||
|
||
value2 = opc_module.read_opc_value_by_subscription(namespace_idx, "Counter", endpoint)
|
||
print(f" 订阅计数器（第二次）: {value2}")
|
||
print(f" 值变化: {value1} -> {value2}")
|
||
except Exception as e:
|
||
print(f" 订阅失败: {e}")
|
||
|
||
print(f"\n4. 监控动态值变化 (10秒):")
|
||
start_time = time.time()
|
||
while time.time() - start_time < 10:
|
||
try:
|
||
counter = opc_module.read_opc_value_with_namespace(namespace_idx, "Counter", endpoint)
|
||
temp = opc_module.read_opc_value_with_namespace(namespace_idx, "Temperature", endpoint)
|
||
print(f" [{time.strftime('%H:%M:%S')}] 计数器: {counter}, 温度: {temp:.2f}")
|
||
time.sleep(2)
|
||
except Exception as e:
|
||
print(f" 监控错误: {e}")
|
||
break
|
||
|
||
except KeyboardInterrupt:
|
||
print("\n用户中断测试")
|
||
finally:
|
||
print(f"\n清理资源...")
|
||
opc_module.cleanup()
|
||
server.stop_server()
|
||
print("交互式测试完成")
|
||
|
||
|
||
def main():
|
||
"""主函数"""
|
||
print("VWED OPC UA模块测试")
|
||
print("请选择测试模式:")
|
||
print("1. 运行单元测试")
|
||
print("2. 运行交互式测试")
|
||
|
||
choice = input("请输入选择 (1 或 2): ").strip()
|
||
|
||
if choice == "1":
|
||
# 运行单元测试
|
||
unittest.main(argv=[''], exit=False, verbosity=2)
|
||
elif choice == "2":
|
||
# 运行交互式测试
|
||
run_interactive_test()
|
||
else:
|
||
print("无效选择，运行单元测试")
|
||
unittest.main(argv=[''], exit=False, verbosity=2)
|
||
|
||
|
||
if __name__ == "__main__":
|
||
main() |