From a3763450bcb4b606f08756478dffd5ab1175c9e0 Mon Sep 17 00:00:00 2001 From: Ethan Lynn Date: Sun, 25 Mar 2018 20:13:10 -0700 Subject: [PATCH] Add test_getHeadersKeys Add test_getHeadersKeys for syscomm.py Change-Id: I462b65b13975f9943e77f061a2d2891eee01858b Issue-ID: MULTICLOUD-198 Signed-off-by: Ethan Lynn --- multivimbroker/multivimbroker/tests/test_syscomm.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/multivimbroker/multivimbroker/tests/test_syscomm.py b/multivimbroker/multivimbroker/tests/test_syscomm.py index 1e88aa2..e28c2c7 100644 --- a/multivimbroker/multivimbroker/tests/test_syscomm.py +++ b/multivimbroker/multivimbroker/tests/test_syscomm.py @@ -64,3 +64,17 @@ class TestSyscomm(unittest.TestCase): } ret_headers = syscomm.originHeaders(req) self.assertDictEqual(expect_headers, ret_headers) + + def test_getHeadersKeys_no_connection(self): + resp = {} + self.assertEqual([], syscomm.getHeadersKeys(resp)) + + def test_getHeadersKeys_with_connection(self): + resp = { + "connection": "aa,bb", + "keep-alive": "yes", + "aa": "vaa", + "bb": "vbb", + "cc": "vcc" + } + self.assertEqual(["cc"], syscomm.getHeadersKeys(resp)) -- 2.16.6