diff --git a/src/test_ansible_link.py b/src/test_ansible_link.py index a34d8cc..674f077 100644 --- a/src/test_ansible_link.py +++ b/src/test_ansible_link.py @@ -12,6 +12,8 @@ def setUpClass(cls): cls.app = init_app() cls.client = cls.app.test_client() cls.app.testing = True + + cls.job_storage = cls.app.config['job_storage'] def test_health_check(self): response = self.client.get('/health') @@ -88,9 +90,10 @@ def test_job_creation_and_retrieval(self): for key in expected_keys: self.assertIn(key, file_data, f"Expected key '{key}' not found in job file") + # Verify that the job data matches the initial payload self.assertEqual(file_data['playbook'], payload['playbook']) self.assertEqual(file_data['inventory'], payload['inventory']) self.assertEqual(file_data['vars'], payload['vars']) if __name__ == '__main__': - unittest.main() + unittest.main() \ No newline at end of file