Adding tests for base classes

[osm/common.git] / osm_common / tests / test_msgbase.py

diff --git a/osm_common/tests/test_msgbase.py b/osm_common/tests/test_msgbase.py
new file mode 100644 (file)
index 0000000..c3b9726
--- /dev/null
+++ b/osm_common/tests/test_msgbase.py
@@ -0,0 +1,48 @@
+import http
+import pytest
+
+from osm_common.msgbase import MsgBase, MsgException
+
+
+def exception_message(message):
+    return "messaging exception " + message
+
+@pytest.fixture
+def msg_base():
+    return MsgBase()
+
+def test_constructor():
+    msgbase = MsgBase()
+
+    assert msgbase != None
+    assert isinstance(msgbase, MsgBase)
+
+def test_connect(msg_base):
+    msg_base.connect()
+
+def test_disconnect(msg_base):
+    msg_base.disconnect()
+
+def test_write(msg_base):
+    with pytest.raises(MsgException) as excinfo:
+        msg_base.write("test", "test", "test")
+    assert str(excinfo.value).startswith(exception_message("Method 'write' not implemented"))
+    assert excinfo.value.http_code == http.HTTPStatus.INTERNAL_SERVER_ERROR
+
+def test_read(msg_base):
+    with pytest.raises(MsgException) as excinfo:
+        msg_base.read("test")
+    assert str(excinfo.value).startswith(exception_message("Method 'read' not implemented"))
+    assert excinfo.value.http_code == http.HTTPStatus.INTERNAL_SERVER_ERROR
+
+def test_aiowrite(msg_base, event_loop):
+    with pytest.raises(MsgException) as excinfo:
+        event_loop.run_until_complete(msg_base.aiowrite("test", "test", "test", event_loop))
+    assert str(excinfo.value).startswith(exception_message("Method 'aiowrite' not implemented"))
+    assert excinfo.value.http_code == http.HTTPStatus.INTERNAL_SERVER_ERROR
+
+def test_aioread(msg_base, event_loop):
+    with pytest.raises(MsgException) as excinfo:
+        event_loop.run_until_complete(msg_base.aioread("test", event_loop))
+    assert str(excinfo.value).startswith(exception_message("Method 'aioread' not implemented"))
+    assert excinfo.value.http_code == http.HTTPStatus.INTERNAL_SERVER_ERROR