diff --git a/tests/test_protect_call.py b/tests/test_protect_call.py new file mode 100644 index 000000000..dc818e7f4 --- /dev/null +++ b/tests/test_protect_call.py @@ -0,0 +1,30 @@ +import unittest +import common + +from bitkeylib import proto + +class TestProtectCall(common.BitkeyTest): + def test_features(self): + features = self.bitkey.call(proto.Initialize(session_id=self.bitkey.session_id)) + + # Result is the same as reported by BitkeyClient class + self.assertEqual(features, self.bitkey.features) + + def test_ping(self): + ping = self.bitkey.call(proto.Ping(message='ahoj!')) + + # Ping results in Success(message='Ahoj!') + self.assertEqual(ping, proto.Success(message='ahoj!')) + + def test_uuid(self): + uuid1 = self.bitkey.get_uuid() + uuid2 = self.bitkey.get_uuid() + + # UUID must be longer than 10 characters + self.assertGreater(len(uuid1), 10) + + # Every resulf of UUID must be the same + self.assertEqual(uuid1, uuid2) + +if __name__ == '__main__': + unittest.main() \ No newline at end of file