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'