1 files changed, 70 insertions, 0 deletions
diff --git a/tests/api/ApiProxy.py b/tests/api/ApiProxy.py
new file mode 100644
index 000000000..0da79a204
--- /dev/null
+++ b/tests/api/ApiProxy.py
@@ -0,0 +1,70 @@
+# -*- coding: utf-8 -*-
+
+
+from pyload.remote.apitypes_debug import classes, methods
+
+from tests.helper.config import credentials
+
+class ApiProxy:
+    """ Proxy that does type checking on the api """
+
+    def __init__(self, api, user=credentials[0], pw=credentials[1]):
+        self.api = api
+        self.user = user
+        self.pw = pw
+
+        if user and pw is not None:
+            self.api.login(user, pw)
+
+    def assert_type(self, result, type):
+        if not type: return # void
+        try:
+            # Complex attribute
+            if isinstance(type, tuple):
+                # Optional result
+                if type[0] is None:
+                    # Only check if not None
+                    if result is not None: self.assert_type(result, type[1])
+
+                # List
+                elif type[0] == list:
+                    assert isinstance(result, list)
+                    for item in result:
+                        self.assert_type(item, type[1])
+                # Dict
+                elif type[0] == dict:
+                    assert isinstance(result, dict)
+                    for k, v in result.iteritems():
+                        self.assert_type(k, type[1])
+                        self.assert_type(v, type[2])
+
+            # Struct - Api class
+            elif hasattr(result, "__name__") and result.__name__ in classes:
+                for attr, atype in zip(result.__slots__, classes[result.__name__]):
+                    self.assert_type(getattr(result, attr), atype)
+            else: # simple object
+                assert isinstance(result, type)
+        except AssertionError:
+            print "Assertion for %s as %s failed" % (result, type)
+            raise
+
+
+    def call(self, func, *args, **kwargs):
+        result = getattr(self.api, func)(*args, **kwargs)
+        self.assert_type(result, methods[func])
+
+        return result
+
+
+    def __getattr__(self, item):
+        def call(*args, **kwargs):
+            return self.call(item, *args, **kwargs)
+
+        return call
+
+if __name__ == "__main__":
+
+    from pyload.remote.JSONClient import JSONClient
+
+    api = ApiProxy(JSONClient(), "User", "test")
+    api.getServerVersion()
+\ No newline at end of file