diff --git a/roles/common/molecule/default/tests/test_default.py b/roles/common/molecule/default/tests/test_default.py index 1cbd19f3b3ff08ab03d6e66b02f471e39bcd8c6a..9bdc107f561ba4fb34bc2f41cfe0f247b6a445c2 100644 --- a/roles/common/molecule/default/tests/test_default.py +++ b/roles/common/molecule/default/tests/test_default.py @@ -250,6 +250,25 @@ def test_pipreqcheck_virtualenv(host): assert virtualenv_activate.mode == 0o644 +def test_pipreqcheck_virtualenv_prompt(host): + """ + Tests if Python virtual environment prompt has been set-up + correctly. + """ + + with host.sudo("pipreqcheck"): + prompt = host.run('bash -c "source /var/lib/pipreqcheck/virtualenv/bin/activate; printenv PS1"') + + # Chop off trailing newline if present (this is from the + # host.run itself). + if prompt.stdout.endswith("\n"): + prompt_stdout = prompt.stdout[:-1] + else: + prompt_stdout = prompt.stdout + + assert prompt_stdout == "(pipreqcheck) " + + def test_pipreqcheck_directories(host): """ Tests creation of directories used for storing configuration used by script