util/agent: add test to check that the handlers are registered correctly

Signed-off-by: Jan Luebbe <[email protected]>

labgrid/util/agent.py

@@ -16,6 +16,7 @@ class Agent:
     def __init__(self):
         self.methods = {}
         self.register('load', self.load)
+        self.register('list', self.list)
 
         # use real stdin/stdout
         self.stdin = sys.stdin
@@ -38,6 +39,9 @@ def load(self, name, source):
         for k, v in module.methods.items():
             self.register('{}.{}'.format(name, k), v)
 
+    def list(self):
+        return list(self.methods.keys())
+
     def run(self):
         for line in self.stdin:
             if not line:

tests/test_agent.py

@@ -99,3 +99,6 @@ def test_local():
 def test_all_modules():
     aw = AgentWrapper(None)
     aw.load('deditec_relais8')
+    methods = aw.list()
+    assert 'deditec_relais8.set' in methods
+    assert 'deditec_relais8.get' in methods