From: Ethan Lynn <ethanlynnl@vmware.com>
Date: Mon, 26 Mar 2018 03:13:10 +0000 (-0700)
Subject: Add test_getHeadersKeys
X-Git-Tag: v1.1.2~38^2
X-Git-Url: https://gerrit.onap.org/r/gitweb?p=multicloud%2Fframework.git;a=commitdiff_plain;h=a3763450bcb4b606f08756478dffd5ab1175c9e0;hp=0b2cfa524556bf4e1541706ecb94e807cf0995bb

Add test_getHeadersKeys

Add test_getHeadersKeys for syscomm.py

Change-Id: I462b65b13975f9943e77f061a2d2891eee01858b
Issue-ID: MULTICLOUD-198
Signed-off-by: Ethan Lynn <ethanlynnl@vmware.com>
---

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))