Optimize getdict Message method always return dict

panoramisk/message.py

@@ -109,9 +109,10 @@ def getdict(self, key):
             sip:[email protected]:4242
         """
         values = self.get(key, None)
-        if not isinstance(values, list):
-            raise TypeError("{0} must be a list. got {1}".format(key, values))
         result = utils.CaseInsensitiveDict()
+        if not isinstance(values, list):
+            result[key] = values
+            return result
         for item in values:
             k, v = item.split('=', 1)
             result[k] = v

tests/test_message.py

@@ -28,3 +28,25 @@ def test_content(message):
 --- blah ---
 ''')
     assert m.content == '--- blah ---'
+
+
+@pytest.mark.parametrize('msg', [
+    Message(
+        {
+            'Response': 'Success',
+            'ChanVariable': ['FROM_DID=', 'SIPURI=sip:[email protected]:4242'],
+        }
+    ),
+    Message(
+        {
+            'Response': 'Success',
+            'ChanVariable': 'var',
+        }
+    )
+]
+                         )
+def test_getdict(msg):
+    assert isinstance(msg.getdict('chanvariable'), utils.CaseInsensitiveDict)
+    for k, v in msg.getdict('chanvariable').items():
+        assert isinstance(k, str)
+        assert isinstance(v, str)