@pytest.mark.parametrize("virtualenv_dir, admin_user, expected_prompt", [

    ('/var/www/parameters-mandatory/virtualenv', 'admin-parameters-mandatory', '(parameters-mandatory) '),

    ('/var/www/parameters-optional.local/virtualenv', 'admin-parameters-optional_local', '(parameters-optional.local) '),

    ('/var/www/parameters-paste-req/virtualenv', 'admin-parameters-paste-req', '(parameters-paste-req) '),

])

def test_python_virtualenv_prompt(host, virtualenv_dir, admin_user, expected_prompt):

    """

    Tests if Python virtual environment prompt has been set-up

    correctly.

    """

    with host.sudo(admin_user):

        prompt = host.run('bash -c "source %s/bin/activate; printenv PS1"', virtualenv_dir)

        # 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 == expected_prompt