temp cert fixture cleans own tempfile

PrefectHQ · Jun 20, 2024 · c6a2ed5 · c6a2ed5
1 parent c55a351
commit c6a2ed5
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/integrations/prefect-kubernetes/tests/test_credentials.py b/src/integrations/prefect-kubernetes/tests/test_credentials.py
@@ -66,7 +66,9 @@ def create_temp_self_signed_cert(self, tmp_path):
         key_file.close()
 
         # Return file names
-        return (cert_file_name, key_file_name)
+        yield (cert_file_name, key_file_name)
+        os.remove(cert_file_name)
+        os.remove(key_file_name)
 
     @pytest.fixture
     def config_context(self, create_temp_self_signed_cert):