Mock systemctl Wrapper in Python for Service Testing

import subprocess
import sys

class ServiceManager:
    def __init__(self, service_name):
        self.service_name = service_name
        self.status = "inactive"
    
    def start(self):
        self.status = "active"
        print(f"Starting {self.service_name}... OK")
    
    def stop(self):
        self.status = "inactive"
        print(f"Stopping {self.service_name}... OK")
    
    def restart(self):
        print(f"Restarting {self.service_name}...")
        self.stop()
        self.start()
        print(f"{self.service_name} restarted successfully")
    
    def status_report(self):
        return f"{self.service_name} is {self.status}"

def mock_systemctl(action, service_name):
    manager = ServiceManager(service_name)
    
    if action == "restart":
        manager.restart()
    elif action == "start":
        manager.start()
    elif action == "stop":
        manager.stop()
    elif action == "status":
        print(manager.status_report())
    else:
        print(f"Unknown action: {action}", file=sys.stderr)
        return 1
    return 0

if __name__ == "__main__":
    service = "nginx"
    print(mock_systemctl("start", service))
    print(mock_systemctl("restart", service))
    print(mock_systemctl("status", service))
    print(mock_systemctl("stop", service))