X-Git-Url: https://osm.etsi.org/gitweb/?a=blobdiff_plain;ds=sidebyside;f=osm_common%2Ftests%2Ftest_msgbase.py;h=665968e6b07f05310ad49bf051181f4a72872dbd;hb=98fc8f01d18d1a05c16fed7ccee355611a6e20ce;hp=c3b97261abd60f8a572245cdefd1e8c96e201918;hpb=4d611d38236ddb5834c48d9228ba78c86b5cbfc5;p=osm%2Fcommon.git

diff --git a/osm_common/tests/test_msgbase.py b/osm_common/tests/test_msgbase.py
index c3b9726..665968e 100644
--- a/osm_common/tests/test_msgbase.py
+++ b/osm_common/tests/test_msgbase.py
@@ -1,3 +1,22 @@
+# Copyright 2018 Whitestack, LLC
+# Copyright 2018 Telefonica S.A.
+#
+# Licensed under the Apache License, Version 2.0 (the "License"); you may
+# not use this file except in compliance with the License. You may obtain
+# a copy of the License at
+#
+#         http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
+# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
+# License for the specific language governing permissions and limitations
+# under the License.
+#
+# For those usages not covered by the Apache License, Version 2.0 please
+# contact: esousa@whitestack.com or alfonso.tiernosepulveda@telefonica.com
+##
+
 import http
 import pytest
 
@@ -7,42 +26,59 @@ 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 msgbase is not None
     assert isinstance(msgbase, MsgBase)
 
+
 def test_connect(msg_base):
-    msg_base.connect()
+    msg_base.connect(None)
+
 
 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 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 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"))
+        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 str(excinfo.value).startswith(
+        exception_message("Method 'aioread' not implemented")
+    )
     assert excinfo.value.http_code == http.HTTPStatus.INTERNAL_SERVER_ERROR