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_write_nonexistent(tmpdir):
env_file = write_testenv(tmpdir.join('.env'))
success, key, val = serverless_helpers.set_key(env_file + 'fooooo', 'WRITE', 'nope')
assert success is None
assert key == 'WRITE'
