Skip to content

Commit 211bfe7

Browse files
haackedhaacked
committed
Run black . 
1 parent 625563e commit 211bfe7

File tree

6 files changed

+63
-69
lines changed

6 files changed

+63
-69
lines changed

‎posthog/__init__.py‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,7 @@
66
from posthog.exception_capture import Integrations # noqa: F401
77
from posthog.version import VERSION
88
from posthog.types import FlagsAndPayloads
9+
910
__version__ = VERSION
1011

1112
"""Settings."""

‎posthog/test/test_client.py‎

Lines changed: 10 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,7 @@
1313
from posthog.version import VERSION
1414
from posthog.types import FeatureFlag, LegacyFlagMetadata
1515

16+
1617
class TestClient(unittest.TestCase):
1718
@classmethod
1819
def setUpClass(cls):
@@ -1213,11 +1214,11 @@ def test_mock_system_context(
12131214

12141215
assert context == expected_context
12151216

1216-
@mock.patch("posthog.client.decide")
1217+
@mock.patch("posthog.client.decide")
12171218
def test_get_decide_returns_normalized_decide_response(self, patch_decide):
12181219
patch_decide.return_value = {
12191220
"featureFlags": {"beta-feature": "random-variant", "alpha-feature": True, "off-feature": False},
1220-
"featureFlagPayloads": {"beta-feature": "{\"some\": \"data\"}"},
1221+
"featureFlagPayloads": {"beta-feature": '{"some": "data"}'},
12211222
"errorsWhileComputingFlags": False,
12221223
"requestId": "test-id",
12231224
}
@@ -1226,9 +1227,9 @@ def test_get_decide_returns_normalized_decide_response(self, patch_decide):
12261227
distinct_id = "test_distinct_id"
12271228
groups = {"test_group_type": "test_group_id"}
12281229
person_properties = {"test_property": "test_value"}
1229-
1230+
12301231
response = client.get_decide(distinct_id, groups, person_properties)
1231-
1232+
12321233
assert response == {
12331234
"flags": {
12341235
"beta-feature": FeatureFlag(
@@ -1237,8 +1238,8 @@ def test_get_decide_returns_normalized_decide_response(self, patch_decide):
12371238
variant="random-variant",
12381239
reason=None,
12391240
metadata=LegacyFlagMetadata(
1240-
payload="{\"some\": \"data\"}",
1241-
)
1241+
payload='{"some": "data"}',
1242+
),
12421243
),
12431244
"alpha-feature": FeatureFlag(
12441245
key="alpha-feature",
@@ -1247,7 +1248,7 @@ def test_get_decide_returns_normalized_decide_response(self, patch_decide):
12471248
reason=None,
12481249
metadata=LegacyFlagMetadata(
12491250
payload=None,
1250-
)
1251+
),
12511252
),
12521253
"off-feature": FeatureFlag(
12531254
key="off-feature",
@@ -1256,8 +1257,8 @@ def test_get_decide_returns_normalized_decide_response(self, patch_decide):
12561257
reason=None,
12571258
metadata=LegacyFlagMetadata(
12581259
payload=None,
1259-
)
1260-
)
1260+
),
1261+
),
12611262
},
12621263
"errorsWhileComputingFlags": False,
12631264
"requestId": "test-id",

‎posthog/test/test_feature_flags.py‎

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -767,8 +767,7 @@ def test_get_all_flags_and_payloads_with_no_fallback(self, patch_decide, patch_c
767767
disabled_flag,
768768
]
769769
self.assertEqual(
770-
client.get_all_flags_and_payloads("distinct_id")["featureFlagPayloads"],
771-
{"beta-feature": "new"}
770+
client.get_all_flags_and_payloads("distinct_id")["featureFlagPayloads"], {"beta-feature": "new"}
772771
)
773772
# decide not called because this can be evaluated locally
774773
self.assertEqual(patch_decide.call_count, 0)
@@ -1624,7 +1623,7 @@ def test_boolean_feature_flag_payloads_local(self, patch_decide):
16241623
},
16251624
}
16261625
self.client.feature_flags = [basic_flag]
1627-
1626+
16281627
self.assertEqual(
16291628
self.client.get_feature_flag_payload(
16301629
"person-flag", "some-distinct-id", person_properties={"region": "USA"}
@@ -1691,7 +1690,7 @@ def test_multivariate_feature_flag_payloads(self, patch_decide):
16911690
},
16921691
}
16931692
self.client.feature_flags = [multivariate_flag]
1694-
1693+
16951694
self.assertEqual(
16961695
self.client.get_feature_flag_payload(
16971696
"beta-feature", "test_id", person_properties={"email": "[email protected]"}
@@ -2362,7 +2361,7 @@ def test_capture_is_called_with_flag_details(self, patch_decide, patch_capture):
23622361
},
23632362
"requestId": "18043bf7-9cf6-44cd-b959-9662ee20d371",
23642363
}
2365-
client = Client(FAKE_TEST_API_KEY)
2364+
client = Client(FAKE_TEST_API_KEY)
23662365

23672366
self.assertEqual(client.get_feature_flag("decide-flag", "some-distinct-id"), "decide-variant")
23682367
self.assertEqual(patch_capture.call_count, 1)
@@ -2400,15 +2399,17 @@ def test_capture_is_called_with_flag_details_and_payload(self, patch_decide, pat
24002399
"metadata": {
24012400
"id": 23,
24022401
"version": 42,
2403-
"payload": "{\"foo\": \"bar\"}",
2402+
"payload": '{"foo": "bar"}',
24042403
},
24052404
}
24062405
},
24072406
"requestId": "18043bf7-9cf6-44cd-b959-9662ee20d371",
24082407
}
2409-
client = Client(FAKE_TEST_API_KEY)
2408+
client = Client(FAKE_TEST_API_KEY)
24102409

2411-
self.assertEqual(client.get_feature_flag_payload("decide-flag-with-payload", "some-distinct-id"), "{\"foo\": \"bar\"}")
2410+
self.assertEqual(
2411+
client.get_feature_flag_payload("decide-flag-with-payload", "some-distinct-id"), '{"foo": "bar"}'
2412+
)
24122413
self.assertEqual(patch_capture.call_count, 1)
24132414
patch_capture.assert_called_with(
24142415
"some-distinct-id",
@@ -2422,13 +2423,12 @@ def test_capture_is_called_with_flag_details_and_payload(self, patch_decide, pat
24222423
"$feature_flag_id": 23,
24232424
"$feature_flag_version": 42,
24242425
"$feature_flag_request_id": "18043bf7-9cf6-44cd-b959-9662ee20d371",
2425-
"$feature_flag_payload": "{\"foo\": \"bar\"}",
2426+
"$feature_flag_payload": '{"foo": "bar"}',
24262427
},
24272428
groups={},
24282429
disable_geoip=None,
24292430
)
24302431

2431-
24322432
@mock.patch("posthog.client.decide")
24332433
def test_capture_is_called_but_does_not_add_all_flags(self, patch_decide):
24342434
patch_decide.return_value = {"featureFlags": {"decide-flag": "decide-value"}}

‎posthog/test/test_request.py‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -84,6 +84,7 @@ def test_normal_decide_response(self):
8484
response = decide("fake_key", "fake_host")
8585
self.assertEqual(response["featureFlags"], {"flag1": True})
8686

87+
8788
@pytest.mark.parametrize(
8889
"host, expected",
8990
[

‎posthog/test/test_types.py‎

Lines changed: 23 additions & 39 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,14 @@
11
import unittest
22
from parameterized import parameterized
3-
from posthog.types import FeatureFlag, FlagMetadata, FlagReason, LegacyFlagMetadata, normalize_decide_response, to_flags_and_payloads
3+
from posthog.types import (
4+
FeatureFlag,
5+
FlagMetadata,
6+
FlagReason,
7+
LegacyFlagMetadata,
8+
normalize_decide_response,
9+
to_flags_and_payloads,
10+
)
11+
412

513
class TestTypes(unittest.TestCase):
614
@parameterized.expand([(True,), (False,)])
@@ -22,7 +30,7 @@ def test_normalize_decide_response_v4(self, has_errors: bool):
2230
}
2331

2432
result = normalize_decide_response(resp)
25-
33+
2634
flag = result["flags"]["my-flag"]
2735
self.assertEqual(flag.key, "my-flag")
2836
self.assertTrue(flag.enabled)
@@ -41,7 +49,7 @@ def test_normalize_decide_response_legacy(self):
4149
# Test legacy response format with "featureFlags" and "featureFlagPayloads"
4250
resp = {
4351
"featureFlags": {"my-flag": "test-variant"},
44-
"featureFlagPayloads": {"my-flag": "{\"some\": \"json-payload\"}"},
52+
"featureFlagPayloads": {"my-flag": '{"some": "json-payload"}'},
4553
"errorsWhileComputingFlags": False,
4654
"requestId": "test-id",
4755
}
@@ -54,9 +62,7 @@ def test_normalize_decide_response_legacy(self):
5462
self.assertEqual(flag.variant, "test-variant")
5563
self.assertEqual(flag.get_value(), "test-variant")
5664
self.assertIsNone(flag.reason)
57-
self.assertEqual(
58-
flag.metadata, LegacyFlagMetadata(payload='{"some": "json-payload"}')
59-
)
65+
self.assertEqual(flag.metadata, LegacyFlagMetadata(payload='{"some": "json-payload"}'))
6066
self.assertFalse(result["errorsWhileComputingFlags"])
6167
self.assertEqual(result["requestId"], "test-id")
6268
# Verify legacy fields are removed
@@ -65,10 +71,7 @@ def test_normalize_decide_response_legacy(self):
6571

6672
def test_normalize_decide_response_boolean_flag(self):
6773
# Test legacy response with boolean flag
68-
resp = {
69-
"featureFlags": {"my-flag": True},
70-
"errorsWhileComputingFlags": False
71-
}
74+
resp = {"featureFlags": {"my-flag": True}, "errorsWhileComputingFlags": False}
7275

7376
result = normalize_decide_response(resp)
7477

@@ -80,9 +83,7 @@ def test_normalize_decide_response_boolean_flag(self):
8083
self.assertTrue(flag.enabled)
8184
self.assertIsNone(flag.variant)
8285
self.assertIsNone(flag.reason)
83-
self.assertEqual(
84-
flag.metadata, LegacyFlagMetadata(payload=None)
85-
)
86+
self.assertEqual(flag.metadata, LegacyFlagMetadata(payload=None))
8687
self.assertFalse(result["errorsWhileComputingFlags"])
8788
self.assertNotIn("featureFlags", result)
8889
self.assertNotIn("featureFlagPayloads", result)
@@ -115,22 +116,21 @@ def test_to_flags_and_payloads_v4(self):
115116
variant=None,
116117
reason=None,
117118
metadata=LegacyFlagMetadata(payload=None),
118-
)
119+
),
119120
},
120121
"errorsWhileComputingFlags": False,
121122
"requestId": "test-id",
122123
}
123124

124125
result = to_flags_and_payloads(resp)
125-
126+
126127
self.assertEqual(result["featureFlags"]["my-variant-flag"], "test-variant")
127128
self.assertEqual(result["featureFlags"]["my-boolean-flag"], True)
128129
self.assertEqual(result["featureFlags"]["disabled-flag"], False)
129130
self.assertEqual(result["featureFlagPayloads"]["my-variant-flag"], '{"some": "json"}')
130131
self.assertEqual(result["featureFlagPayloads"]["my-boolean-flag"], None)
131132
self.assertNotIn("disabled-flag", result["featureFlagPayloads"])
132133

133-
134134
def test_to_flags_and_payloads_empty(self):
135135
# Test empty response
136136
resp = {
@@ -140,7 +140,7 @@ def test_to_flags_and_payloads_empty(self):
140140
}
141141

142142
result = to_flags_and_payloads(resp)
143-
143+
144144
self.assertEqual(result["featureFlags"], {})
145145
self.assertEqual(result["featureFlagPayloads"], {})
146146

@@ -159,7 +159,7 @@ def test_to_flags_and_payloads_with_payload(self):
159159
"metadata": {
160160
"id": 23,
161161
"version": 42,
162-
"payload": "{\"foo\": \"bar\"}",
162+
"payload": '{"foo": "bar"}',
163163
},
164164
}
165165
},
@@ -170,25 +170,16 @@ def test_to_flags_and_payloads_with_payload(self):
170170
result = to_flags_and_payloads(normalized)
171171

172172
self.assertEqual(result["featureFlags"]["decide-flag"], "decide-variant")
173-
self.assertEqual(result["featureFlagPayloads"]["decide-flag"], "{\"foo\": \"bar\"}")
173+
self.assertEqual(result["featureFlagPayloads"]["decide-flag"], '{"foo": "bar"}')
174174

175175
def test_feature_flag_from_json(self):
176176
# Test with full metadata
177177
resp = {
178178
"key": "test-flag",
179179
"enabled": True,
180180
"variant": "test-variant",
181-
"reason": {
182-
"code": "matched_condition",
183-
"condition_index": 0,
184-
"description": "Matched condition set 1"
185-
},
186-
"metadata": {
187-
"id": 1,
188-
"payload": '{"some": "json"}',
189-
"version": 2,
190-
"description": "test-description"
191-
}
181+
"reason": {"code": "matched_condition", "condition_index": 0, "description": "Matched condition set 1"},
182+
"metadata": {"id": 1, "payload": '{"some": "json"}', "version": 2, "description": "test-description"},
192183
}
193184

194185
flag = FeatureFlag.from_json(resp)
@@ -205,10 +196,7 @@ def test_feature_flag_from_json(self):
205196

206197
def test_feature_flag_from_json_minimal(self):
207198
# Test with minimal required fields
208-
resp = {
209-
"key": "test-flag",
210-
"enabled": True
211-
}
199+
resp = {"key": "test-flag", "enabled": True}
212200

213201
flag = FeatureFlag.from_json(resp)
214202
self.assertEqual(flag.key, "test-flag")
@@ -224,11 +212,7 @@ def test_feature_flag_from_json_without_metadata(self):
224212
"key": "test-flag",
225213
"enabled": True,
226214
"variant": "test-variant",
227-
"reason": {
228-
"code": "matched_condition",
229-
"condition_index": 0,
230-
"description": "Matched condition set 1"
231-
}
215+
"reason": {"code": "matched_condition", "condition_index": 0, "description": "Matched condition set 1"},
232216
}
233217

234218
flag = FeatureFlag.from_json(resp)

0 commit comments

Comments
 (0)