Change option from --yes to --force
This commit is contained in:
Родитель
c21032df0a
Коммит
7466c6ea0d
|
@ -15,15 +15,15 @@ class Command(BaseCommand):
|
|||
def add_arguments(self, parser: ArgumentParser) -> None:
|
||||
parser.add_argument("fxa_id", help="The user's FxA ID")
|
||||
parser.add_argument(
|
||||
"-y",
|
||||
"--yes",
|
||||
"-f",
|
||||
"--force",
|
||||
action="store_true",
|
||||
help="Skip confirmation and delete any found data",
|
||||
)
|
||||
|
||||
def handle(self, *args: Any, **kwargs: Any) -> None | str:
|
||||
fxa_id: str = kwargs["fxa_id"]
|
||||
skip_confirmation: bool = kwargs["yes"]
|
||||
skip_confirmation: bool = kwargs["force"]
|
||||
|
||||
try:
|
||||
data = _PhoneData.from_fxa(fxa_id)
|
||||
|
|
|
@ -78,7 +78,7 @@ def test_active_user(phone_user: User) -> None:
|
|||
assert InboundContact.objects.filter(relay_number=relay_number).count() == 2
|
||||
|
||||
stdout = StringIO()
|
||||
call_command(THE_COMMAND, phone_user.profile.fxa.uid, "--yes", stdout=stdout)
|
||||
call_command(THE_COMMAND, phone_user.profile.fxa.uid, "--force", stdout=stdout)
|
||||
|
||||
expected_stdout = f"""\
|
||||
Found a matching user:
|
||||
|
@ -107,7 +107,7 @@ def test_no_contacts(phone_user: User) -> None:
|
|||
InboundContact.objects.filter(relay_number=relay_number).delete()
|
||||
|
||||
stdout = StringIO()
|
||||
call_command(THE_COMMAND, phone_user.profile.fxa.uid, "--yes", stdout=stdout)
|
||||
call_command(THE_COMMAND, phone_user.profile.fxa.uid, "--force", stdout=stdout)
|
||||
|
||||
expected_stdout = f"""\
|
||||
Found a matching user:
|
||||
|
@ -135,7 +135,7 @@ def test_no_relay_phone(phone_user: User) -> None:
|
|||
RelayNumber.objects.filter(user=phone_user).delete()
|
||||
|
||||
stdout = StringIO()
|
||||
call_command(THE_COMMAND, phone_user.profile.fxa.uid, "--yes", stdout=stdout)
|
||||
call_command(THE_COMMAND, phone_user.profile.fxa.uid, "--force", stdout=stdout)
|
||||
|
||||
expected_stdout = f"""\
|
||||
Found a matching user:
|
||||
|
@ -162,7 +162,7 @@ def test_no_real_phone(phone_user: User) -> None:
|
|||
RealPhone.objects.filter(user=phone_user).delete()
|
||||
|
||||
stdout = StringIO()
|
||||
call_command(THE_COMMAND, phone_user.profile.fxa.uid, "--yes", stdout=stdout)
|
||||
call_command(THE_COMMAND, phone_user.profile.fxa.uid, "--force", stdout=stdout)
|
||||
|
||||
expected_stdout = f"""\
|
||||
Found a matching user:
|
||||
|
@ -188,7 +188,7 @@ def test_user_not_found() -> None:
|
|||
"""The command fails if a matching user is not found."""
|
||||
stdout = StringIO()
|
||||
with pytest.raises(CommandError) as excinfo:
|
||||
call_command(THE_COMMAND, "unknown_fxa_id", "--yes", stdout=stdout)
|
||||
call_command(THE_COMMAND, "unknown_fxa_id", "--force", stdout=stdout)
|
||||
assert str(excinfo.value) == "No user with FxA ID 'unknown_fxa_id'."
|
||||
assert stdout.getvalue() == ""
|
||||
|
||||
|
|
Загрузка…
Ссылка в новой задаче