update: adds init tests

Author: namsnath

tests/plugins/plugins.py

@@ -49,8 +49,7 @@ def init(
         all_plugins: List[SuperTokensPublicPlugin],
         sdk_version: str,
     ):
-        # TODO: Test this
-        print(f"{identifier} init")
+        PluginTestRecipe.init_calls.append(identifier)
 
     return init
 
@@ -78,6 +77,7 @@ def plugin_factory(
     override_functions: bool = False,
     override_apis: bool = False,
     deps: Optional[List[SuperTokensPlugin]] = None,
+    add_init: bool = False,
 ):
     override_map_obj: OverrideMap = {PluginTestRecipe.recipe_id: OverrideConfig()}
 
@@ -90,24 +90,57 @@ def plugin_factory(
             identifier
         )
 
+    init_fn = None
+    if add_init:
+        init_fn = init_factory(identifier)
+
     class Plugin(SuperTokensPlugin):
         id: str = identifier
         compatible_sdk_versions: Union[str, List[str]] = ["0.30.0"]
         override_map: Optional[OverrideMap] = override_map_obj
-        init: Any = init_factory(identifier)
+        init: Any = init_fn
         dependencies: Optional[SuperTokensPluginDependencies] = dependency_factory(deps)
 
     return Plugin()
 
 
-Plugin1 = plugin_factory("plugin1", override_functions=True)
-Plugin2 = plugin_factory("plugin2", override_functions=True)
-Plugin3Dep1 = plugin_factory("plugin3dep1", override_functions=True, deps=[Plugin1])
+Plugin1 = plugin_factory(
+    "plugin1",
+    override_functions=True,
+    add_init=True,
+)
+Plugin2 = plugin_factory(
+    "plugin2",
+    override_functions=True,
+    add_init=True,
+)
+Plugin3Dep1 = plugin_factory(
+    "plugin3dep1",
+    override_functions=True,
+    deps=[Plugin1],
+    add_init=True,
+)
 Plugin3Dep2_1 = plugin_factory(
-    "plugin3dep2_1", override_functions=True, deps=[Plugin2, Plugin1]
+    "plugin3dep2_1",
+    override_functions=True,
+    deps=[Plugin2, Plugin1],
+    add_init=True,
+)
+Plugin4Dep1 = plugin_factory(
+    "plugin4dep1",
+    override_functions=True,
+    deps=[Plugin1],
+    add_init=True,
+)
+Plugin4Dep2 = plugin_factory(
+    "plugin4dep2",
+    override_functions=True,
+    deps=[Plugin2],
+    add_init=True,
 )
-Plugin4Dep1 = plugin_factory("plugin4dep1", override_functions=True, deps=[Plugin1])
-Plugin4Dep2 = plugin_factory("plugin4dep2", override_functions=True, deps=[Plugin2])
 Plugin4Dep3__2_1 = plugin_factory(
-    "plugin4dep3__2_1", override_functions=True, deps=[Plugin3Dep2_1]
+    "plugin4dep3__2_1",
+    override_functions=True,
+    deps=[Plugin3Dep2_1],
+    add_init=True,
 )

tests/plugins/recipe.py

@@ -26,6 +26,7 @@
 
 class PluginTestRecipe(RecipeModule):
     __instance: Optional["PluginTestRecipe"] = None
+    init_calls: List[str] = []
     recipe_id = "plugin_test"
 
     config: NormalizedPluginTestConfig
@@ -94,6 +95,7 @@ def func(app_info: AppInfo, plugins: List[OverrideMap]):
     @staticmethod
     def reset():
         PluginTestRecipe.__instance = None
+        PluginTestRecipe.init_calls = []
 
     def get_all_cors_headers(self) -> List[str]:
         return []

tests/plugins/test_plugins.py

@@ -15,6 +15,7 @@
     PluginRouteHandlerFunctionOkResponse,
     SuperTokensPlugin,
 )
+from supertokens_python.post_init_callbacks import PostSTInitCallbacks
 from supertokens_python.supertokens import SupertokensPublicConfig
 
 from tests.utils import outputs, reset
@@ -41,9 +42,11 @@
 def setup_and_teardown():
     reset()
     PluginTestRecipe.reset()
+    PostSTInitCallbacks.reset()
     yield
     reset()
     PluginTestRecipe.reset()
+    PostSTInitCallbacks.reset()
 
 
 def recipe_factory(override_functions: bool = False, override_apis: bool = False):
@@ -214,36 +217,41 @@ def test_overrides(
 
 # TODO: Figure out a way to add circular dependencies and test them
 @mark.parametrize(
-    ("plugins", "recipe_expectation", "api_expectation"),
+    ("plugins", "recipe_expectation", "api_expectation", "init_expectation"),
     [
         param(
             [Plugin1, Plugin1],
             outputs(["plugin1", "original"]),
             outputs(["original"]),
+            outputs(["plugin1"]),
             id="1,1 => 1",
         ),
         param(
             [Plugin1, Plugin2],
             outputs(["plugin2", "plugin1", "original"]),
             outputs(["original"]),
+            outputs(["plugin1", "plugin2"]),
             id="1,2 => 2,1",
         ),
         param(
             [Plugin3Dep1],
             outputs(["plugin3dep1", "plugin1", "original"]),
             outputs(["original"]),
+            outputs(["plugin1", "plugin3dep1"]),
             id="3->1 => 3,1",
         ),
         param(
             [Plugin3Dep2_1],
             outputs(["plugin3dep2_1", "plugin1", "plugin2", "original"]),
             outputs(["original"]),
+            outputs(["plugin2", "plugin1", "plugin3dep2_1"]),
             id="3->(2,1) => 3,2,1",
         ),
         param(
             [Plugin3Dep1, Plugin4Dep2],
             outputs(["plugin4dep2", "plugin2", "plugin3dep1", "plugin1", "original"]),
             outputs(["original"]),
+            outputs(["plugin1", "plugin3dep1", "plugin2", "plugin4dep2"]),
             id="3->1,4->2 => 4,2,3,1",
         ),
         param(
@@ -252,20 +260,23 @@ def test_overrides(
                 ["plugin4dep3__2_1", "plugin3dep2_1", "plugin1", "plugin2", "original"]
             ),
             outputs(["original"]),
+            outputs(["plugin2", "plugin1", "plugin3dep2_1", "plugin4dep3__2_1"]),
             id="4->3->(2,1) => 4,3,1,2",
         ),
         param(
             [Plugin3Dep1, Plugin4Dep1],
             outputs(["plugin4dep1", "plugin3dep1", "plugin1", "original"]),
             outputs(["original"]),
+            outputs(["plugin1", "plugin3dep1", "plugin4dep1"]),
             id="3->1,4->1 => 4,3,1",
         ),
     ],
 )
-def test_depdendencies(
+def test_depdendencies_and_init(
     plugins: List[SuperTokensPlugin],
     recipe_expectation: Any,
     api_expectation: Any,
+    init_expectation: Any,
 ):
     partial_init(
         recipe_list=[
@@ -298,8 +309,10 @@ def test_depdendencies(
             message="msg",
         )
 
+    with init_expectation as expected_stack:
+        assert PluginTestRecipe.init_calls == expected_stack
+
 
-# TODO: Add tests for init, recipe config override
 def test_st_config_override():
     plugin = plugin_factory("plugin1", override_functions=False, override_apis=False)