tests done

test.py

@@ -1,6 +1,10 @@
 #!/usr/bin/env python
 
-import pytest
+import os
+import re
+import consumerlocator
+import communicator
+import messagesender
 
 """
 Unit tests for producer module.
@@ -10,3 +14,206 @@ __author__ = "@tormakris"
 __copyright__ = "Copyright 2020, GoldenPogácsa Team"
 __module_name__ = "test"
 __version__text__ = "1"
+
+generateduuid = 'c959ad81-58f9-4445-aab4-8f3d68aee1ad'
+os.environ["KnownConsumer"] = "10.1.2.3"
+
+def test_generate_string(mocker):
+    mocker.patch('communicator.Communicator')
+    comm = communicator.Communicator(
+        currentconsumer="localhost",
+        uuid=generateduuid)
+    mess = messagesender.MessageSender(communicator=comm)
+    msg = mess.randomstring(stringlength=32)
+    assert isinstance(msg, str)
+    assert len(msg) == 32
+
+
+def test_sendmessage(httpserver):
+    httpserver.expect_oneshot_request(
+        uri="/log",
+        method='POST',
+        data="{\"uuid\": \"c959ad81-58f9-4445-aab4-8f3d68aee1ad\", \"message\": \"SENDING\"}").respond_with_json(
+        {
+            "test": "ok"})
+    url = httpserver.url_for("/")
+    port = re.match(r"\W*http[^:]*\D*(\d+)", url).group(1)
+    comm = communicator.Communicator(
+        currentconsumer=f"127.0.0.1:{port}",
+        uuid=generateduuid)
+    mess = "SENDING"
+    ret = comm.sendmessage(message=mess)
+    assert ret is None
+
+
+def test_send_message(mocker):
+    mocker.patch('communicator.Communicator')
+    comm = communicator.Communicator(
+        currentconsumer="127.0.0.1",
+        uuid=generateduuid)
+    mess = messagesender.MessageSender(communicator=comm)
+    messa = "SENDING"
+    msg = mess.sendmessage(message=messa)
+    assert msg is None
+
+
+def test_discoveravailableconsumers(httpserver):
+    httpserver.expect_oneshot_request(
+        uri="/consumer",
+        method='GET',
+        data="").respond_with_json(
+        ["10.69.42.1", "10.10.10.10", "10.20.30.40"])
+    url = httpserver.url_for("/")
+    port = re.match(r"\W*http[^:]*\D*(\d+)", url).group(1)
+    comm = communicator.Communicator(
+        currentconsumer=f"127.0.0.1:{port}",
+        uuid=generateduuid)
+    ret = comm.discoveravailableconsumers()
+    assert type(ret) is list
+    assert ret == ["10.69.42.1", "10.10.10.10", "10.20.30.40"]
+
+
+def test_isconsumeravailable(httpserver):
+    httpserver.expect_oneshot_request(
+        uri="/consumer",
+        method='GET',
+        data="").respond_with_json(
+        ["10.69.42.1", "10.10.10.10", "10.20.30.40"])
+    url = httpserver.url_for("/")
+    port = re.match(r"\W*http[^:]*\D*(\d+)", url).group(1)
+    comm = communicator.Communicator(
+        currentconsumer=f"127.0.0.1:{port}",
+        uuid=generateduuid)
+    ret = comm.isconsumeravailable()
+    assert type(ret) is bool
+    assert ret
+
+    ret2 = comm.isconsumeravailable()
+    assert type(ret2) is bool
+    assert ret2 == False
+
+    comm2 = communicator.Communicator(
+        currentconsumer="127.0.0.1:69",
+        uuid=generateduuid)
+
+    ret3 = comm2.isconsumeravailable()
+    assert type(ret3) is bool
+    assert ret3 == False
+
+
+def test_checkconsumer(httpserver):
+    httpserver.expect_oneshot_request(
+        uri="/consumer",
+        method='GET',
+        data="").respond_with_json(
+        ["10.69.42.1", "10.10.10.10", "10.20.30.40"])
+    url = httpserver.url_for("/")
+    port = re.match(r"\W*http[^:]*\D*(\d+)", url).group(1)
+    comm = communicator.Communicator(
+        currentconsumer="127.0.0.1",
+        uuid=generateduuid)
+    ret = comm.checkconsumer(f"127.0.0.1:{port}")
+    assert type(ret) is bool
+    assert ret
+
+    ret2 = comm.checkconsumer(f"127.0.0.1:{port}")
+    assert type(ret2) is bool
+    assert ret2 == False
+
+    comm2 = communicator.Communicator(
+        currentconsumer="127.0.0.1",
+        uuid=generateduuid)
+
+    ret3 = comm2.checkconsumer(f"127.0.0.1:{port}")
+    assert type(ret3) is bool
+    assert ret3 == False
+
+
+def test_setcurrentconsumer():
+    comm = communicator.Communicator(
+        currentconsumer="127.0.0.1",
+        uuid=generateduuid)
+    comm.set_currentconsumer("10.69.42.1")
+    assert comm.currenctconsumer == "10.69.42.1"
+
+def test_updateconsumerlist(httpserver):
+    httpserver.expect_oneshot_request(
+        uri="/consumer",
+        method='GET',
+        data="").respond_with_json(
+        ["10.69.42.1", "10.10.10.10", "10.20.30.40"])
+    url = httpserver.url_for("/")
+    port = re.match(r"\W*http[^:]*\D*(\d+)", url).group(1)
+    comm = communicator.Communicator(
+        currentconsumer=f"127.0.0.1:{port}",
+        uuid=generateduuid)
+    locator = consumerlocator.ConsumerLocator(uuid=generateduuid, communicator=comm)
+    ret = locator.updateconsumerlist()
+    assert ret is None
+
+def test_learnconsumerlist(httpserver):
+    httpserver.expect_oneshot_request(
+        uri="/consumer",
+        method='GET',
+        data="").respond_with_json(
+        ["10.69.42.1", "10.10.10.10", "10.20.30.40"])
+    url = httpserver.url_for("/")
+    port = re.match(r"\W*http[^:]*\D*(\d+)", url).group(1)
+    comm = communicator.Communicator(
+        currentconsumer=f"127.0.0.1:{port}",
+        uuid=generateduuid)
+    locator = consumerlocator.ConsumerLocator(uuid=generateduuid, communicator=comm)
+    ret = locator.learnconsumerlist()
+    assert ret is None
+
+def test_getcurrentconsumer(mocker):
+    mocker.patch('communicator.Communicator')
+    comm = communicator.Communicator(
+        currentconsumer="127.0.0.1",
+        uuid=generateduuid)
+    locator = consumerlocator.ConsumerLocator(uuid=generateduuid, communicator=comm)
+    assert locator.getcurrentconsumer() == "10.1.2.3"
+
+def test_checkcurrentconsumer(httpserver):
+    httpserver.expect_oneshot_request(
+        uri="/consumer",
+        method='GET',
+        data="").respond_with_json(
+        ["10.69.42.1", "10.10.10.10", "10.20.30.40"])
+    url = httpserver.url_for("/")
+    port = re.match(r"\W*http[^:]*\D*(\d+)", url).group(1)
+    comm = communicator.Communicator(
+        currentconsumer=f"127.0.0.1:{port}",
+        uuid=generateduuid)
+    locator = consumerlocator.ConsumerLocator(uuid=generateduuid, communicator=comm)
+    ret = locator.checkcurrentconsumer()
+    assert ret == False
+
+def test_updateconsumer(httpserver):
+    httpserver.expect_oneshot_request(
+        uri="/consumer",
+        method='GET',
+        data="").respond_with_json(
+        ["10.69.42.1", "10.10.10.10", "10.20.30.40"])
+    url = httpserver.url_for("/")
+    port = re.match(r"\W*http[^:]*\D*(\d+)", url).group(1)
+    comm = communicator.Communicator(
+        currentconsumer=f"127.0.0.1:{port}",
+        uuid=generateduuid)
+    locator = consumerlocator.ConsumerLocator(uuid=generateduuid, communicator=comm)
+    ret = locator.updateconsumer()
+    assert ret == "10.69.42.1"
+
+    httpserver.expect_oneshot_request(
+        uri="/consumer",
+        method='GET',
+        data="").respond_with_json(
+        [])
+    url2 = httpserver.url_for("/")
+    port2 = re.match(r"\W*http[^:]*\D*(\d+)", url2).group(1)
+    comm2 = communicator.Communicator(
+        currentconsumer=f"127.0.0.1:{port2}",
+        uuid=generateduuid)
+    locator2 = consumerlocator.ConsumerLocator(uuid=generateduuid, communicator=comm2)
+    ret2 = locator2.updateconsumer()
+    assert ret2 is None