From 6b618f4d94d6d5e435fad36a7372a12aba37b0c3 Mon Sep 17 00:00:00 2001
From: Jason Fried <fried@meta.com>
Date: Fri, 5 Jan 2024 08:07:42 -0800
Subject: [PATCH] fix DebugWipeLoader being used for zipimporter

Summary:
X-link: https://github.com/facebookincubator/zstrong/pull/664

Need to insure DebugWipeLoader doesn't get used for zipimporter files.  The original imp.find_module that I replaced did not support zipimporter the new importlib find_spec does find the them.

Reviewed By: itamaro

Differential Revision: D52556833

fbshipit-source-id: bf5892a950a965bb3b17ab951910db2e33b69a34
---
 build/fbcode_builder/CMake/fb_py_test_main.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/build/fbcode_builder/CMake/fb_py_test_main.py b/build/fbcode_builder/CMake/fb_py_test_main.py
index 2a9b5b4b9eed..a9499e221a4d 100644
--- a/build/fbcode_builder/CMake/fb_py_test_main.py
+++ b/build/fbcode_builder/CMake/fb_py_test_main.py
@@ -120,7 +120,8 @@ def get_code(self, fullname):
                     code_impl.ob_size = 0
                 return code
 
-        spec.loader = DebugWipeLoader(fullname, spec.origin)
+        if isinstance(spec.loader, SourceFileLoader):
+            spec.loader = DebugWipeLoader(fullname, spec.origin)
         return spec