DEV Community

Serhat Teker
Serhat Teker

Posted on • Originally published at on

Test Environment Variables in Python

If you are working with environment variables you need to set, change or delete the values from the system for testing purposes. So how can we test them?

Let's assume that we have a function that gets a "secret" key from os

import os

def get_secret():
    """Simple retrieval function.
    Returns SECRET or raises OSError.
    secret = os.getenv('SECRET', default=None)

    if secret is None:
        raise OSError("SECRET environment is not set.")

    return secret
Enter fullscreen mode Exit fullscreen mode

One way for testing would be put an env var into config files of test like tox.ini or pytest.ini

# tox.ini
setenv = SECRET=top_secret
Enter fullscreen mode Exit fullscreen mode

Another way would be using pytest-env plugin and adding it into pytest.ini:

# pytest.ini
env =
Enter fullscreen mode Exit fullscreen mode

However these methods are not preferred since it violates isolating tests.

Monkeypatching environment variables

Recommended way comes from pytest itself: monkeypatch. pytest provides a mechanism which gives the setenv and delenv methods.

First we set SECRET env var and test it, then testing OSError when SECRET not set. Since as we know; we always should test both "normal" and "failure" cases:

import pytest

from utilities import get_user

def test_get_user(monkeypatch):
    """Set the SECRET env var to assert the behavior."""
    monkeypatch.setenv("SECRET", "top_secret")

    assert get_user() == "top_secret"

def test_get_user_exception(monkeypatch):
    """Remove the SECRET env var and assert OSError is raised."""
    monkeypatch.delenv("SECRET", raising=False)

    with pytest.raises(OSError):
        _ = get_user()
Enter fullscreen mode Exit fullscreen mode

You can also use fixture structures and share across them tests. For more info please go to: pytest:testing env vars.

All done!

Top comments (0)