removed irrelevant tested and added new tests

Author: iresharma

tests/telemetry/test_telemetry.py

@@ -12,17 +12,20 @@
 # License for the specific language governing permissions and limitations
 # under the License.
 
+import asyncio
+import os
+
 import pytest
+
 from supertokens_python import InputAppInfo, Supertokens, SupertokensConfig, init
 from supertokens_python.recipe import session
-import asyncio
 from tests.utils import reset
 
 
 @pytest.mark.asyncio
-async def test_asgi_telemetry():
+async def test_telemetry():
     reset()
-    with pytest.warns(None) as record:
+    with pytest.warns(None) as _:
         init(
             supertokens_config=SupertokensConfig("http://localhost:3567"),
             app_info=InputAppInfo(
@@ -44,20 +47,16 @@ async def test_asgi_telemetry():
         )
         await asyncio.sleep(1)
 
-    for warn in record:
-        if warn.category is RuntimeWarning:
-            if "telemetry" in str(warn.message) and "was never awaited" in str(
-                warn.message
-            ):
-                assert False, "Asyncio error"
+        assert Supertokens.get_instance().telemetry is not None
 
-    assert Supertokens.get_instance()._telemetry_status == "SKIPPED"  # type: ignore pylint: disable=W0212
+        assert Supertokens.get_instance().telemetry
 
 
 @pytest.mark.asyncio
-async def test_asgi_telemetry_with_wrong_mode():
+async def test_read_from_env():
     reset()
-    with pytest.warns(None) as record:
+    os.environ["TEST_MODE"] = "testing"
+    with pytest.warns(None) as _:
         init(
             supertokens_config=SupertokensConfig("http://localhost:3567"),
             app_info=InputAppInfo(
@@ -67,75 +66,6 @@ async def test_asgi_telemetry_with_wrong_mode():
                 api_base_path="/auth",
             ),
             framework="fastapi",
-            mode="wsgi",
-            recipe_list=[
-                session.init(
-                    anti_csrf="VIA_TOKEN",
-                    cookie_domain="supertokens.io",
-                    override=session.InputOverrideConfig(),
-                )
-            ],
-            telemetry=True,
-        )
-        await asyncio.sleep(1)
-
-    found_warning = False
-    for warn in record:
-        if warn.category is RuntimeWarning:
-            found_warning = found_warning or "Inconsistent mode detected" in str(
-                warn.message
-            )
-
-    assert found_warning, "Asyncio error"
-
-    assert Supertokens.get_instance()._telemetry_status == "SKIPPED"  # type: ignore pylint: disable=W0212
-
-
-def test_wsgi_telemetry():
-    reset()
-    with pytest.warns(None) as record:
-        init(
-            supertokens_config=SupertokensConfig("http://localhost:3567"),
-            app_info=InputAppInfo(
-                app_name="SuperTokens Demo",
-                api_domain="http://api.supertokens.io",
-                website_domain="http://supertokens.io",
-                api_base_path="/auth",
-            ),
-            framework="flask",
-            mode="wsgi",
-            recipe_list=[
-                session.init(
-                    anti_csrf="VIA_TOKEN",
-                    cookie_domain="supertokens.io",
-                    override=session.InputOverrideConfig(),
-                )
-            ],
-            telemetry=True,
-        )
-
-    for warn in record:
-        if warn.category is RuntimeWarning:
-            if "telemetry" in str(warn.message) and "was never awaited" in str(
-                warn.message
-            ):
-                assert False, "Asyncio error"
-
-    assert Supertokens.get_instance()._telemetry_status == "SKIPPED"  # type: ignore pylint: disable=W0212
-
-
-def test_wsgi_telemetry_with_wrong_mode():
-    reset()
-    with pytest.warns(None) as record:
-        init(
-            supertokens_config=SupertokensConfig("http://localhost:3567"),
-            app_info=InputAppInfo(
-                app_name="SuperTokens Demo",
-                api_domain="http://api.supertokens.io",
-                website_domain="http://supertokens.io",
-                api_base_path="/auth",
-            ),
-            framework="flask",
             mode="asgi",
             recipe_list=[
                 session.init(
@@ -144,16 +74,7 @@ def test_wsgi_telemetry_with_wrong_mode():
                     override=session.InputOverrideConfig(),
                 )
             ],
-            telemetry=True,
         )
+        await asyncio.sleep(1)
 
-    found_warning = False
-    for warn in record:
-        if warn.category is RuntimeWarning:
-            found_warning = found_warning or "Inconsistent mode detected" in str(
-                warn.message
-            )
-
-    assert found_warning, "Asyncio error"
-
-    assert Supertokens.get_instance()._telemetry_status == "SKIPPED"  # type: ignore pylint: disable=W0212
+        assert not Supertokens.get_instance().telemetry