Update unit tests

Author: janbuchar

tests/unit/actor/test_actor_env_helpers.py

@@ -3,6 +3,7 @@
 import random
 import string
 from datetime import datetime, timedelta
+from decimal import Decimal
 from typing import TYPE_CHECKING, Any
 
 from pydantic_core import TzInfo
@@ -30,15 +31,7 @@ async def test_actor_is_not_at_home_when_local() -> None:
         assert is_at_home is False
 
 
-async def test_actor_is_at_home_on_apify(monkeypatch: pytest.MonkeyPatch) -> None:
-    print('setenv')
-    monkeypatch.setenv(ApifyEnvVars.IS_AT_HOME, 'true')
-    async with Actor as actor:
-        is_at_home = actor.is_at_home()
-        assert is_at_home is True
-
-
-async def test_get_env_with_randomized_env_vars(monkeypatch: pytest.MonkeyPatch) -> None:
+async def test_get_env_with_randomized_env_vars(monkeypatch: pytest.MonkeyPatch) -> None:  # noqa: PLR0912
     ignored_env_vars = {
         ApifyEnvVars.INPUT_KEY,
         ApifyEnvVars.MEMORY_MBYTES,
@@ -82,7 +75,10 @@ async def test_get_env_with_randomized_env_vars(monkeypatch: pytest.MonkeyPatch)
             continue
 
         float_get_env_var = float_env_var.name.lower()
-        expected_get_env[float_get_env_var] = random.random()
+        if float_env_var == ActorEnvVars.MAX_TOTAL_CHARGE_USD:
+            expected_get_env[float_get_env_var] = Decimal(random.random())
+        else:
+            expected_get_env[float_get_env_var] = random.random()
         monkeypatch.setenv(float_env_var, f'{expected_get_env[float_get_env_var]}')
 
     for bool_env_var in BOOL_ENV_VARS:
@@ -93,6 +89,9 @@ async def test_get_env_with_randomized_env_vars(monkeypatch: pytest.MonkeyPatch)
         expected_get_env[bool_get_env_var] = random.choice([True, False])
         monkeypatch.setenv(bool_env_var, f'{"true" if expected_get_env[bool_get_env_var] else "false"}')
 
+    expected_get_env[ApifyEnvVars.IS_AT_HOME.name.lower()] = False
+    monkeypatch.setenv(ApifyEnvVars.IS_AT_HOME, 'false')
+
     for datetime_env_var in DATETIME_ENV_VARS:
         if datetime_env_var in ignored_env_vars:
             continue

tests/unit/actor/test_actor_log.py

@@ -14,12 +14,8 @@
     import pytest
 
 
-async def test_actor_logs_messages_correctly(
-    caplog: pytest.LogCaptureFixture,
-    monkeypatch: pytest.MonkeyPatch,
-) -> None:
+async def test_actor_logs_messages_correctly(caplog: pytest.LogCaptureFixture) -> None:
     caplog.set_level(logging.DEBUG, logger='apify')
-    monkeypatch.setenv('APIFY_IS_AT_HOME', '1')
 
     with contextlib.suppress(RuntimeError):
         async with Actor(configure_logging=False):
@@ -43,7 +39,7 @@ async def test_actor_logs_messages_correctly(
             # Test that exception in Actor.main is logged with the traceback
             raise RuntimeError('Dummy RuntimeError')
 
-    assert len(caplog.records) == 12
+    assert len(caplog.records) == 13
 
     assert caplog.records[0].levelno == logging.INFO
     assert caplog.records[0].message == 'Initializing Actor...'
@@ -56,39 +52,42 @@ async def test_actor_logs_messages_correctly(
     assert getattr(caplog.records[1], 'os', None) == sys.platform
 
     assert caplog.records[2].levelno == logging.DEBUG
-    assert caplog.records[2].message.startswith('APIFY_ACTOR_EVENTS_WS_URL env var not set')
+    assert caplog.records[2].message == 'Event manager initialized'
 
     assert caplog.records[3].levelno == logging.DEBUG
-    assert caplog.records[3].message == 'Debug message'
+    assert caplog.records[3].message == 'Charging manager initialized'
 
-    assert caplog.records[4].levelno == logging.INFO
-    assert caplog.records[4].message == 'Info message'
+    assert caplog.records[4].levelno == logging.DEBUG
+    assert caplog.records[4].message == 'Debug message'
 
-    assert caplog.records[5].levelno == logging.WARNING
-    assert caplog.records[5].message == 'Warning message'
+    assert caplog.records[5].levelno == logging.INFO
+    assert caplog.records[5].message == 'Info message'
 
-    assert caplog.records[6].levelno == logging.ERROR
-    assert caplog.records[6].message == 'Error message'
+    assert caplog.records[6].levelno == logging.WARNING
+    assert caplog.records[6].message == 'Warning message'
 
     assert caplog.records[7].levelno == logging.ERROR
-    assert caplog.records[7].message == 'Exception message'
-    assert caplog.records[7].exc_info is not None
-    assert caplog.records[7].exc_info[0] is ValueError
-    assert isinstance(caplog.records[7].exc_info[1], ValueError)
-    assert str(caplog.records[7].exc_info[1]) == 'Dummy ValueError'
-
-    assert caplog.records[8].levelno == logging.INFO
-    assert caplog.records[8].message == 'Multi\nline\nlog\nmessage'
-
-    assert caplog.records[9].levelno == logging.ERROR
-    assert caplog.records[9].message == 'Actor failed with an exception'
-    assert caplog.records[9].exc_info is not None
-    assert caplog.records[9].exc_info[0] is RuntimeError
-    assert isinstance(caplog.records[9].exc_info[1], RuntimeError)
-    assert str(caplog.records[9].exc_info[1]) == 'Dummy RuntimeError'
-
-    assert caplog.records[10].levelno == logging.INFO
-    assert caplog.records[10].message == 'Exiting Actor'
-
-    assert caplog.records[11].levelno == logging.DEBUG
-    assert caplog.records[11].message == 'Not calling sys.exit(91) because Actor is running in an unit test'
+    assert caplog.records[7].message == 'Error message'
+
+    assert caplog.records[8].levelno == logging.ERROR
+    assert caplog.records[8].message == 'Exception message'
+    assert caplog.records[8].exc_info is not None
+    assert caplog.records[8].exc_info[0] is ValueError
+    assert isinstance(caplog.records[8].exc_info[1], ValueError)
+    assert str(caplog.records[8].exc_info[1]) == 'Dummy ValueError'
+
+    assert caplog.records[9].levelno == logging.INFO
+    assert caplog.records[9].message == 'Multi\nline\nlog\nmessage'
+
+    assert caplog.records[10].levelno == logging.ERROR
+    assert caplog.records[10].message == 'Actor failed with an exception'
+    assert caplog.records[10].exc_info is not None
+    assert caplog.records[10].exc_info[0] is RuntimeError
+    assert isinstance(caplog.records[10].exc_info[1], RuntimeError)
+    assert str(caplog.records[10].exc_info[1]) == 'Dummy RuntimeError'
+
+    assert caplog.records[11].levelno == logging.INFO
+    assert caplog.records[11].message == 'Exiting Actor'
+
+    assert caplog.records[12].levelno == logging.DEBUG
+    assert caplog.records[12].message == 'Not calling sys.exit(91) because Actor is running in an unit test'