def test_unset_key(tmpdir):
    env_file = write_testenv(tmpdir.join('.env'))
    stage = serverless_helpers.get_key(env_file, 'SERVERLESS_STAGE')
    assert stage == 'dev'
    success, _ = serverless_helpers.unset_key(env_file, 'SERVERLESS_STAGE')
    assert success
    stage = serverless_helpers.get_key(env_file, 'SERVERLESS_STAGE')
    assert stage is None
def test_override_key(tmpdir):
    env_file = write_testenv(tmpdir.join('.env'))
    data_stage = serverless_helpers.get_key(env_file, 'SERVERLESS_DATA_MODEL_STAGE')
    assert data_stage == 'dev'

    serverless_helpers.set_key(env_file, 'SERVERLESS_DATA_MODEL_STAGE', 'overridden')
    data_stage = serverless_helpers.get_key(env_file, 'SERVERLESS_DATA_MODEL_STAGE')

    assert data_stage == 'overridden'
def test_get_nonexistent(tmpdir):
    env_file = write_testenv(tmpdir.join('.env'))
    data = serverless_helpers.get_key(env_file, 'NOT_A_THING')
    assert data is None
def test_read_nonexistent(tmpdir):
    env_file = write_testenv(tmpdir.join('.env'))
    data_stage = serverless_helpers.get_key(env_file + 'fooooo', 'SERVERLESS_DATA_MODEL_STAGE')
    assert data_stage is None