Update test case to account for context handling

NeonGeckoCom · Dec 19, 2023 · 32c6441 · 32c6441
1 parent 23b2f6a
commit 32c6441
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/test/test_skill.py b/test/test_skill.py
@@ -246,10 +246,10 @@ def test_get_attachments(self):
                         "data": {"key": "val"}}
 
         content = self.skill._get_support_info(test_message, test_profile)
-        self.assertEqual(set(content.keys()),
-                         {'user_profile', 'message_context', 'module_status',
-                          'loaded_skills', 'packages', 'host_device',
-                          'generated_time_utc'})
+        for key in {'user_profile', 'message_context', 'module_status',
+                    'loaded_skills', 'packages', 'host_device',
+                    'generated_time_utc'}:
+            self.assertIsNotNone(content[key])
         self.assertEqual(content['user_profile'], test_profile)
         self.assertEqual(content['message_context'], original_context)
         self.assertEqual(content['module_status'], {'test': True})