[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Author: pre-commit-ci[bot]

tests/test_analytics.py

@@ -2,7 +2,7 @@
 from datetime import datetime, timedelta
 from unittest import mock
 
-from hamcrest import assert_that, is_, equal_to
+from hamcrest import assert_that, equal_to, is_
 
 from flagsmith.analytics import ANALYTICS_TIMER, AnalyticsProcessor
 

tests/test_flagsmith.py

@@ -8,7 +8,18 @@
 import responses
 from flag_engine.environments.models import EnvironmentModel
 from flag_engine.features.models import FeatureModel, FeatureStateModel
-from hamcrest import assert_that, none, not_none, is_, has_entry, has_length, empty, equal_to, same_instance, is_not
+from hamcrest import (
+    assert_that,
+    empty,
+    equal_to,
+    has_entry,
+    has_length,
+    is_,
+    is_not,
+    none,
+    not_none,
+    same_instance,
+)
 from pytest_mock import MockerFixture
 from responses import matchers
 
@@ -22,7 +33,7 @@
 
 
 def test_flagsmith_starts_polling_manager_on_init_if_enabled(
-        mocker: MockerFixture, server_api_key: str, requests_session_response_ok: None
+    mocker: MockerFixture, server_api_key: str, requests_session_response_ok: None
 ) -> None:
     # Given
     mock_polling_manager = mocker.MagicMock()
@@ -40,7 +51,7 @@ def test_flagsmith_starts_polling_manager_on_init_if_enabled(
 
 @responses.activate()
 def test_update_environment_sets_environment(
-        flagsmith: Flagsmith, environment_json: str, environment_model: EnvironmentModel
+    flagsmith: Flagsmith, environment_json: str, environment_model: EnvironmentModel
 ) -> None:
     # Given
     responses.add(method="GET", url=flagsmith.environment_url, body=environment_json)
@@ -56,7 +67,7 @@ def test_update_environment_sets_environment(
 
 @responses.activate()
 def test_get_environment_flags_calls_api_when_no_local_environment(
-        api_key: str, flagsmith: Flagsmith, flags_json: str
+    api_key: str, flagsmith: Flagsmith, flags_json: str
 ) -> None:
     # Given
     responses.add(method="GET", url=flagsmith.environment_flags_url, body=flags_json)
@@ -79,7 +90,7 @@ def test_get_environment_flags_calls_api_when_no_local_environment(
 
 @responses.activate()
 def test_get_environment_flags_uses_local_environment_when_available(
-        flagsmith: Flagsmith, environment_model: EnvironmentModel
+    flagsmith: Flagsmith, environment_model: EnvironmentModel
 ) -> None:
     # Given
     flagsmith._environment = environment_model
@@ -91,14 +102,19 @@ def test_get_environment_flags_uses_local_environment_when_available(
     # Then
     assert_that(responses.calls, empty())
     assert_that(len(all_flags), equal_to(1))
-    assert_that(all_flags[0].feature_name, equal_to(environment_model.feature_states[0].feature.name))
+    assert_that(
+        all_flags[0].feature_name,
+        equal_to(environment_model.feature_states[0].feature.name),
+    )
     assert_that(all_flags[0].enabled, is_(environment_model.feature_states[0].enabled))
-    assert_that(all_flags[0].value, equal_to(environment_model.feature_states[0].get_value()))
+    assert_that(
+        all_flags[0].value, equal_to(environment_model.feature_states[0].get_value())
+    )
 
 
 @responses.activate()
 def test_get_identity_flags_calls_api_when_no_local_environment_no_traits(
-        flagsmith: Flagsmith, identities_json: str
+    flagsmith: Flagsmith, identities_json: str
 ) -> None:
     # Given
     responses.add(method="POST", url=flagsmith.identities_url, body=identities_json)
@@ -122,7 +138,7 @@ def test_get_identity_flags_calls_api_when_no_local_environment_no_traits(
 
 @responses.activate()
 def test_get_identity_flags_calls_api_when_no_local_environment_with_traits(
-        flagsmith: Flagsmith, identities_json: str
+    flagsmith: Flagsmith, identities_json: str
 ) -> None:
     # Given
     responses.add(method="POST", url=flagsmith.identities_url, body=identities_json)
@@ -138,12 +154,19 @@ def test_get_identity_flags_calls_api_when_no_local_environment_with_traits(
         # Decode 'body' from bytes to string if it is in bytes format.
         body = body.decode()
 
-    assert_that(body, equal_to(json.dumps(
-        {
-            "identifier": identifier,
-            "traits": [{"trait_key": k, "trait_value": v} for k, v in traits.items()],
-        }
-    )))
+    assert_that(
+        body,
+        equal_to(
+            json.dumps(
+                {
+                    "identifier": identifier,
+                    "traits": [
+                        {"trait_key": k, "trait_value": v} for k, v in traits.items()
+                    ],
+                }
+            )
+        ),
+    )
 
     # Taken from hard coded values in tests/data/identities.json
     assert_that(identity_flags.all_flags()[0].enabled, is_(True))
@@ -153,7 +176,7 @@ def test_get_identity_flags_calls_api_when_no_local_environment_with_traits(
 
 @responses.activate()
 def test_get_identity_flags_uses_local_environment_when_available(
-        flagsmith: Flagsmith, environment_model: EnvironmentModel, mocker: MockerFixture
+    flagsmith: Flagsmith, environment_model: EnvironmentModel, mocker: MockerFixture
 ) -> None:
     # Given
     flagsmith._environment = environment_model
@@ -180,8 +203,8 @@ def test_get_identity_flags_uses_local_environment_when_available(
 
 @responses.activate()
 def test_get_identity_flags__transient_identity__calls_expected(
-        flagsmith: Flagsmith,
-        identities_json: str,
+    flagsmith: Flagsmith,
+    identities_json: str,
 ) -> None:
     # Given
     responses.add(
@@ -211,10 +234,10 @@ def test_get_identity_flags__transient_identity__calls_expected(
 
 @responses.activate()
 def test_get_identity_flags__transient_trait_keys__calls_expected(
-        flagsmith: Flagsmith,
-        identities_json: str,
-        environment_model: EnvironmentModel,
-        mocker: MockerFixture,
+    flagsmith: Flagsmith,
+    identities_json: str,
+    environment_model: EnvironmentModel,
+    mocker: MockerFixture,
 ) -> None:
     # Given
     responses.add(
@@ -245,7 +268,7 @@ def test_get_identity_flags__transient_trait_keys__calls_expected(
 
 
 def test_request_connection_error_raises_flagsmith_api_error(
-        mocker: MockerFixture, api_key: str
+    mocker: MockerFixture, api_key: str
 ) -> None:
     """
     Test the behaviour when session.<method> raises a ConnectionError. Note that this
@@ -316,7 +339,7 @@ def default_flag_handler(feature_name: str) -> DefaultFlag:
 
 @responses.activate()
 def test_default_flag_is_not_used_when_environment_flags_returned(
-        api_key: str, flags_json: str
+    api_key: str, flags_json: str
 ) -> None:
     # Given
     feature_name = "some_feature"
@@ -342,7 +365,9 @@ def default_flag_handler(feature_name: str) -> DefaultFlag:
     flag = flags.get_flag(feature_name)
     assert_that(flag.is_default, is_not(True))
     assert_that(flag.value, is_not(equal_to(default_flag.value)))
-    assert_that(flag.value, equal_to("some-value"))  # hard coded value in tests/data/flags.json
+    assert_that(
+        flag.value, equal_to("some-value")
+    )  # hard coded value in tests/data/flags.json
 
 
 @responses.activate()
@@ -382,7 +407,7 @@ def default_flag_handler(feature_name: str) -> DefaultFlag:
 
 @responses.activate()
 def test_default_flag_is_not_used_when_identity_flags_returned(
-        api_key: str, identities_json: str
+    api_key: str, identities_json: str
 ) -> None:
     # Given
     feature_name = "some_feature"
@@ -408,11 +433,13 @@ def default_flag_handler(feature_name: str) -> DefaultFlag:
     flag = flags.get_flag(feature_name)
     assert_that(flag.is_default, is_not(True))
     assert_that(flag.value, is_not(equal_to(default_flag.value)))
-    assert_that(flag.value, equal_to("some-value"))  # hard coded value in tests/data/identities.json
+    assert_that(
+        flag.value, equal_to("some-value")
+    )  # hard coded value in tests/data/identities.json
 
 
 def test_default_flags_are_used_if_api_error_and_default_flag_handler_given(
-        mocker: MockerFixture,
+    mocker: MockerFixture,
 ) -> None:
     # Given
     # a default flag and associated handler
@@ -438,7 +465,7 @@ def default_flag_handler(feature_name: str) -> DefaultFlag:
 
 
 def test_get_identity_segments_no_traits(
-        local_eval_flagsmith: Flagsmith, environment_model: EnvironmentModel
+    local_eval_flagsmith: Flagsmith, environment_model: EnvironmentModel
 ) -> None:
     # Given
     identifier = "identifier"
@@ -451,7 +478,7 @@ def test_get_identity_segments_no_traits(
 
 
 def test_get_identity_segments_with_valid_trait(
-        local_eval_flagsmith: Flagsmith, environment_model: EnvironmentModel
+    local_eval_flagsmith: Flagsmith, environment_model: EnvironmentModel
 ) -> None:
     # Given
     identifier = "identifier"
@@ -462,7 +489,9 @@ def test_get_identity_segments_with_valid_trait(
 
     # Then
     assert_that(len(segments), equal_to(1))
-    assert_that(segments[0].name, is_("Test segment"))  # obtained from data/environment.json
+    assert_that(
+        segments[0].name, is_("Test segment")
+    )  # obtained from data/environment.json
 
 
 def test_local_evaluation_requires_server_key() -> None:
@@ -501,7 +530,7 @@ def get_environment(self) -> EnvironmentModel:
 
 @responses.activate()
 def test_flagsmith_uses_offline_handler_if_set_and_no_api_response(
-        mocker: MockerFixture, environment_model: EnvironmentModel
+    mocker: MockerFixture, environment_model: EnvironmentModel
 ) -> None:
     # Given
     api_url = "http://some.flagsmith.com/api/v1/"
@@ -533,9 +562,9 @@ def test_flagsmith_uses_offline_handler_if_set_and_no_api_response(
 
 @responses.activate()
 def test_offline_mode__local_evaluation__correct_fallback(
-        mocker: MockerFixture,
-        environment_model: EnvironmentModel,
-        caplog: pytest.LogCaptureFixture,
+    mocker: MockerFixture,
+    environment_model: EnvironmentModel,
+    caplog: pytest.LogCaptureFixture,
 ) -> None:
     # Given
     api_url = "http://some.flagsmith.com/api/v1/"
@@ -577,7 +606,10 @@ def test_cannot_use_offline_mode_without_offline_handler() -> None:
         Flagsmith(offline_mode=True, offline_handler=None)
 
     # Then
-    assert_that(e.exconly(), is_("ValueError: offline_handler must be provided to use offline mode."))
+    assert_that(
+        e.exconly(),
+        is_("ValueError: offline_handler must be provided to use offline mode."),
+    )
 
 
 def test_cannot_use_default_handler_and_offline_handler(mocker: MockerFixture) -> None:
@@ -591,7 +623,10 @@ def test_cannot_use_default_handler_and_offline_handler(mocker: MockerFixture) -
         )
 
     # Then
-    assert_that(e.exconly(), is_("ValueError: Cannot use both default_flag_handler and offline_handler."))
+    assert_that(
+        e.exconly(),
+        is_("ValueError: Cannot use both default_flag_handler and offline_handler."),
+    )
 
 
 def test_cannot_create_flagsmith_client_in_remote_evaluation_without_api_key() -> None:
@@ -604,7 +639,7 @@ def test_cannot_create_flagsmith_client_in_remote_evaluation_without_api_key() -
 
 
 def test_stream_not_used_by_default(
-        requests_session_response_ok: None, server_api_key: str
+    requests_session_response_ok: None, server_api_key: str
 ) -> None:
     # When
     flagsmith = Flagsmith(
@@ -617,7 +652,7 @@ def test_stream_not_used_by_default(
 
 
 def test_stream_used_when_enable_realtime_updates_is_true(
-        requests_session_response_ok: None, server_api_key: str
+    requests_session_response_ok: None, server_api_key: str
 ) -> None:
     # When
     flagsmith = Flagsmith(
@@ -631,7 +666,7 @@ def test_stream_used_when_enable_realtime_updates_is_true(
 
 
 def test_error_raised_when_realtime_updates_is_true_and_local_evaluation_false(
-        requests_session_response_ok: None, server_api_key: str
+    requests_session_response_ok: None, server_api_key: str
 ) -> None:
     with pytest.raises(ValueError):
         Flagsmith(
@@ -643,8 +678,8 @@ def test_error_raised_when_realtime_updates_is_true_and_local_evaluation_false(
 
 @responses.activate()
 def test_flagsmith_client_get_identity_flags__local_evaluation__returns_expected(
-        environment_json: str,
-        server_api_key: str,
+    environment_json: str,
+    server_api_key: str,
 ) -> None:
     # Given
     identifier = "overridden-id"
@@ -669,7 +704,7 @@ def test_flagsmith_client_get_identity_flags__local_evaluation__returns_expected
 
 
 def test_custom_feature_error_raised_when_invalid_feature(
-        requests_session_response_ok: None, server_api_key: str
+    requests_session_response_ok: None, server_api_key: str
 ) -> None:
     # Given
     flagsmith = Flagsmith(

tests/test_offline_handlers.py

@@ -17,6 +17,7 @@ def test_local_file_handler(environment_json: str) -> None:
 
         # Then
         assert isinstance(environment_model, EnvironmentModel)
-        assert_that(environment_model.api_key,
-                    equal_to("B62qaMZNwfiqT76p38ggrQ"))  # hard coded from json file
+        assert_that(
+            environment_model.api_key, equal_to("B62qaMZNwfiqT76p38ggrQ")
+        )  # hard coded from json file
         mock_file.assert_called_once_with(environment_document_file_path)

tests/test_polling_manager.py

@@ -3,7 +3,7 @@
 
 import requests
 import responses
-from hamcrest import assert_that, is_, equal_to
+from hamcrest import assert_that, equal_to, is_
 from pytest_mock import MockerFixture
 
 from flagsmith import Flagsmith