oncall-engine/engine/apps/base/tests/test_live_settings.py
Joey Orlando afc688feda
upgrade flake8 to v7 (#4141)
# Which issue(s) this PR closes

Fixes [this
issue](https://github.com/grafana/oncall-private/pull/2620/files#diff-0144920543fd191db13f76c9fb797116e26eda2bdd2b79332b61bfbf5846208eR193-R197)
(https://github.com/PyCQA/pycodestyle/issues/334#issuecomment-2027394413)
in `grafana/oncall-private`

## Checklist

- [ ] Unit, integration, and e2e (if applicable) tests updated (N/A)
- [x] Documentation added (or `pr:no public docs` PR label added if not
required)
- [x] Added the relevant release notes label (see labels prefixed w/
`release:`). These labels dictate how your PR will
    show up in the autogenerated release notes.
2024-04-02 14:26:19 +00:00

72 lines
2.6 KiB
Python

from unittest.mock import patch
import pytest
from apps.base.models import LiveSetting
from apps.base.utils import live_settings
from apps.twilioapp.phone_provider import TwilioPhoneProvider
@pytest.mark.django_db
def test_fallback_to_settings(settings):
settings.SOME_NEW_FEATURE_ENABLED = True
with patch.object(LiveSetting, "AVAILABLE_NAMES", ("SOME_NEW_FEATURE_ENABLED",)):
assert LiveSetting.get_setting("SOME_NEW_FEATURE_ENABLED") is True
@pytest.mark.django_db
def test_take_from_db(settings):
settings.SOME_NEW_FEATURE_ENABLED = True
with patch.object(LiveSetting, "AVAILABLE_NAMES", ("SOME_NEW_FEATURE_ENABLED",)):
LiveSetting.objects.create(name="SOME_NEW_FEATURE_ENABLED", value=False)
assert LiveSetting.get_setting("SOME_NEW_FEATURE_ENABLED") is False
@pytest.mark.django_db
def test_restrict_foreign_names():
with pytest.raises(ValueError):
LiveSetting.objects.create(name="SOME_NONEXISTENT_FANCY_FEATURE_ENABLED", value=42)
with pytest.raises(ValueError):
LiveSetting.get_setting("SOME_NONEXISTENT_FANCY_FEATURE_ENABLED")
@pytest.mark.parametrize("value", (True, None, 12, "test string", ["hey", "there", 1]))
@pytest.mark.django_db
def test_multi_type_support(value):
with patch.object(LiveSetting, "AVAILABLE_NAMES", ("SOME_NEW_FEATURE_ENABLED",)):
LiveSetting.objects.create(name="SOME_NEW_FEATURE_ENABLED", value=value)
setting_value = LiveSetting.get_setting("SOME_NEW_FEATURE_ENABLED")
assert type(setting_value) is type(value)
assert setting_value == value
@pytest.mark.django_db
def test_live_settings_proxy(settings, monkeypatch):
settings.SOME_SETTING = 12
monkeypatch.setattr(LiveSetting, "AVAILABLE_NAMES", ("SOME_SETTING",))
assert live_settings.SOME_SETTING == 12
live_settings.SOME_SETTING = 42
assert LiveSetting.objects.get(name="SOME_SETTING").value == 42
assert live_settings.SOME_SETTING == 42
@pytest.mark.django_db
def test_twilio_respects_changed_credentials(settings):
settings.TWILIO_ACCOUNT_SID = "twilio_account_sid"
settings.TWILIO_AUTH_TOKEN = "twilio_auth_token"
settings.TWILIO_NUMBER = "twilio_number"
twilio_client = TwilioPhoneProvider()
live_settings.TWILIO_ACCOUNT_SID = "new_twilio_account_sid"
live_settings.TWILIO_AUTH_TOKEN = "new_twilio_auth_token"
live_settings.TWILIO_NUMBER = "new_twilio_number"
assert twilio_client._default_twilio_api_client.username == "new_twilio_account_sid"
assert twilio_client._default_twilio_api_client.password == "new_twilio_auth_token"
assert twilio_client._default_twilio_number == "new_twilio_number"