|
1 | | -import json |
2 | 1 | import logging |
3 | 2 | import os |
4 | 3 | import site |
|
18 | 17 | from django.core.cache import caches |
19 | 18 | from django.db.backends.base.creation import TEST_DATABASE_PREFIX |
20 | 19 | from django.test.utils import setup_databases |
21 | | -from django.urls import reverse |
22 | 20 | from flag_engine.segments.constants import EQUAL |
23 | 21 | from moto import mock_dynamodb # type: ignore[import-untyped] |
24 | 22 | from mypy_boto3_dynamodb.service_resource import DynamoDBServiceResource, Table |
|
27 | 25 | from pytest_django.fixtures import SettingsWrapper |
28 | 26 | from pytest_django.plugin import blocking_manager_key |
29 | 27 | from pytest_mock import MockerFixture |
30 | | -from rest_framework import status |
31 | 28 | from rest_framework.authtoken.models import Token |
32 | 29 | from rest_framework.test import APIClient |
33 | 30 | from task_processor.task_run_method import TaskRunMethod |
|
86 | 83 | from tests.test_helpers import fix_issue_3869 |
87 | 84 | from tests.types import ( |
88 | 85 | AdminClientAuthType, |
89 | | - GetEnvironmentFlagsResponseJSONCallable, |
90 | | - GetIdentityFlagsResponseJSONCallable, |
91 | 86 | WithEnvironmentPermissionsCallable, |
92 | 87 | WithOrganisationPermissionsCallable, |
93 | 88 | WithProjectPermissionsCallable, |
@@ -572,13 +567,6 @@ def api_client() -> APIClient: |
572 | 567 | return APIClient() |
573 | 568 |
|
574 | 569 |
|
575 | | -@pytest.fixture() |
576 | | -def sdk_client(environment: Environment) -> APIClient: |
577 | | - client = APIClient() |
578 | | - client.credentials(HTTP_X_ENVIRONMENT_KEY=environment.api_key) |
579 | | - return client |
580 | | - |
581 | | - |
582 | 570 | @pytest.fixture() |
583 | 571 | def feature(project: Project, environment: Environment) -> Feature: |
584 | 572 | return Feature.objects.create(name="Test Feature1", project=project) # type: ignore[no-any-return] |
@@ -1277,46 +1265,3 @@ def set_github_webhook_secret() -> None: |
1277 | 1265 | from django.conf import settings |
1278 | 1266 |
|
1279 | 1267 | settings.GITHUB_WEBHOOK_SECRET = "secret-key" |
1280 | | - |
1281 | | - |
1282 | | -# TODO: move _flags_response_json to common? |
1283 | | -@pytest.fixture() |
1284 | | -def get_environment_flags_response_json( |
1285 | | - sdk_client: APIClient, |
1286 | | -) -> GetEnvironmentFlagsResponseJSONCallable: |
1287 | | - get_environment_flags_url = reverse("api-v1:flags") |
1288 | | - |
1289 | | - def _get_environment_flags_response_json(num_expected_flags: int) -> typing.Dict: # type: ignore[type-arg] |
1290 | | - _response = sdk_client.get(get_environment_flags_url) |
1291 | | - assert _response.status_code == status.HTTP_200_OK |
1292 | | - _response_json = _response.json() |
1293 | | - assert len(_response_json) == num_expected_flags |
1294 | | - return _response_json # type: ignore[no-any-return] |
1295 | | - |
1296 | | - return _get_environment_flags_response_json |
1297 | | - |
1298 | | - |
1299 | | -@pytest.fixture() |
1300 | | -def get_identity_flags_response_json( |
1301 | | - sdk_client: APIClient, identity: Identity |
1302 | | -) -> GetIdentityFlagsResponseJSONCallable: |
1303 | | - identities_url = reverse("api-v1:sdk-identities") |
1304 | | - |
1305 | | - def _get_identity_flags_response_json( # type: ignore[no-untyped-def] |
1306 | | - num_expected_flags: int, identifier: str = identity.identifier, **traits |
1307 | | - ) -> typing.Dict: # type: ignore[type-arg] |
1308 | | - traits = traits or {} |
1309 | | - data = { |
1310 | | - "identifier": identifier, |
1311 | | - "traits": [{"trait_key": k, "trait_value": v} for k, v in traits.items()], |
1312 | | - } |
1313 | | - |
1314 | | - _response = sdk_client.post( |
1315 | | - identities_url, data=json.dumps(data), content_type="application/json" |
1316 | | - ) |
1317 | | - assert _response.status_code == status.HTTP_200_OK |
1318 | | - _response_json = _response.json() |
1319 | | - assert len(_response_json["flags"]) == num_expected_flags |
1320 | | - return _response_json # type: ignore[no-any-return] |
1321 | | - |
1322 | | - return _get_identity_flags_response_json # type: ignore[return-value] |
0 commit comments