test_proxy.py

#!/usr/bin/env python3
"""Test script for SCADA-PLC proxy communication"""

import socket
import json
import time

def send_proxy_request(command, device=None, value=None):
    """Send a request to the PLC proxy server"""
    try:
        # Connect to PLC proxy
        s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        s.connect(('192.168.1.10', 8080))
        
        # Prepare request
        request = {'command': command}
        if device:
            request['device'] = device
        if value is not None:
            request['value'] = value
        
        # Send request
        s.send(json.dumps(request).encode())
        
        # Get response
        response = s.recv(1024).decode()
        s.close()
        
        return json.loads(response)
    except Exception as e:
        return {'error': str(e)}

if __name__ == "__main__":
    print("Testing PLC Proxy Communication")
    print("===============================")
    
    # Test 1: Get status of all RIO devices
    print("\n1. Getting status of all RIO devices:")
    result = send_proxy_request('status')
    print(f"   Response: {json.dumps(result, indent=2)}")
    
    time.sleep(1)
    
    # Test 2: Set actuator to ON
    print("\n2. Setting actuator to ON (value=1):")
    result = send_proxy_request('write', 'actuator', 1)
    print(f"   Response: {result}")
    
    time.sleep(2)
    
    # Test 3: Set actuator to OFF
    print("\n3. Setting actuator to OFF (value=0):")
    result = send_proxy_request('write', 'actuator', 0)
    print(f"   Response: {result}")
    
    print("\nProxy tests complete!")