1 import osdf.adapters.dcae.message_router as MR
4 from osdf.operation.exceptions import MessageBusConfigurationException
5 from unittest.mock import patch
8 class TestMessageRouter(unittest.TestCase):
10 def test_valid_MR(self):
11 mr = MR.MessageRouterClient(dmaap_url="https://MYHOST:3905")
13 def test_valid_MR_with_base_urls(self):
14 base_urls = ["https://MYHOST1:3905/","https://MYHOST2:3905/"]
15 mr = MR.MessageRouterClient(mr_host_base_urls=base_urls, topic="MY-TOPIC")
17 def test_invalid_valid_MR_with_base_urls(self):
19 base_urls = ["https://MYHOST1:3905/","https://MYHOST2:3905/"]
21 mr = MR.MessageRouterClient(mr_host_base_urls=base_urls)
22 except MessageBusConfigurationException:
25 raise Exception("Allows invalid MR configuration") # if it failed to error out
27 @patch('osdf.adapters.dcae.message_router.MessageRouterClient.http_request', return_value={})
28 def test_mr_http_request_mocked(self, http_request):
29 mr = MR.MessageRouterClient(dmaap_url="https://MYHOST:3905")
30 mr.http_request = http_request
32 assert mr.post("Hello") == {}
34 def test_mr_http_request_non_existent_host(self):
35 mr = MR.MessageRouterClient(dmaap_url="https://MYHOST:3905")
41 raise Exception("Allows invalid host") # if it failed to error out
42 if __name__ == "__main__":