almet/ihatemoney
1
0
Fork 0
mirror of https://github.com/spiral-project/ihatemoney.git synced 2025-05-06 13:01:50 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

add option to select hash method

Browse source 
this will fix tests
This commit is contained in:
Glandos 2021-06-06 15:44:48 +02:00
parent 3894f1568a
commit 8847894421
2 changed files with 8 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
ihatemoney/manage.py
View file

@ -34,10 +34,14 @@ def runserver(ctx):
@click.command(name="generate_password_hash") @click.command(name="generate_password_hash")
def password_hash(): @click.option(
"--method",
help="""Password method. See https://werkzeug.palletsprojects.com/en/2.0.x/utils/#werkzeug.security.generate_password_hash""", # NOQA
)
def password_hash(method):
"""Get password from user and hash it without printing it in clear text.""" """Get password from user and hash it without printing it in clear text."""
password = getpass.getpass(prompt="Password: ") password = getpass.getpass(prompt="Password: ")
print(generate_password_hash(password)) print(generate_password_hash(password, method=method))
@click.command() @click.command()

4
ihatemoney/tests/main_test.py
View file

@ -89,8 +89,8 @@ class CommandTestCase(BaseTestCase):
def test_generate_password_hash(self): def test_generate_password_hash(self):
runner = self.app.test_cli_runner() runner = self.app.test_cli_runner()
with patch("getpass.getpass", new=lambda prompt: "secret"): with patch("getpass.getpass", new=lambda prompt: "secret"):
result = runner.invoke(password_hash) result = runner.invoke(password_hash, ["--method", "pbkdf2:sha256:260000"])
self.assertEqual(len(result.output.strip()), 94) self.assertEqual(len(result.output.strip()), 102)
def test_demo_project_deletion(self): def test_demo_project_deletion(self):
self.create_project("demo") self.create_project("demo")