recreate_module: set __spec__ to the new module

celery/local.py

@@ -517,6 +517,7 @@ def recreate_module(name, compat_modules=None, by_module=None, direct=None,
     new_module.__dict__.update({
         mod: get_compat_module(new_module, mod) for mod in compat_modules
     })
+    new_module.__spec__ = old_module.__spec__
     return old_module, new_module
 
 

t/unit/utils/test_local.py

@@ -1,3 +1,5 @@
+import sys
+from importlib.util import find_spec
 from unittest.mock import Mock
 
 import pytest
@@ -339,3 +341,15 @@ def test_maybe_evaluate(self):
 
         assert maybe_evaluate(30) == 30
         assert x.__evaluated__()
+
+
+class test_celery_import:
+    def test_import_celery(self, monkeypatch):
+        monkeypatch.delitem(sys.modules, "celery", raising=False)
+        spec = find_spec("celery")
+        assert spec
+
+        import celery
+
+        assert celery.__spec__ == spec
+        assert find_spec("celery") == spec