def test_merge_env_variable_underscore_name():
prefix = "PYTHONJSONCONFIG"
variables = {
f"{prefix}_TEST__VALUE1": "bla",
f"{prefix}_TEST_VALUE2": "1",
f"{prefix}_TEST_TEST2__VALUES_VALUE2": "3"
}
for key, value in variables.items():
os.environ[key] = value
config = ConfigNode({
"test_value1": "blub",
"test": {
"value2": "5",
"test2_values": {
"value2": 1
}
}
})
config.merge_with_env_variables(prefix)
assert config.test_value1 == "bla"
assert config.test.value2 == "1"
assert config.test.test2_values.value2 == "3"
for key, value in variables.items():
del os.environ[key]
def test_keys(config_dict):
config = ConfigNode(config_dict)
keys = list(config.keys())
expected_keys = [
"key1", "key2.key3", "key2.key4.key5", "key2.key6", "key7"
]
assert keys == expected_keys
def test_nested_update():
node = ConfigNode({"key1": 1})
with pytest.raises(AttributeError):
node.key2
with pytest.raises(RuntimeError):
node.update("key2.key3", "asd", upsert=False)
def test_nested_upsert():
node = ConfigNode({"key1": 1})
with pytest.raises(AttributeError):
node.key2
node.update("key2.key3", "asd", upsert=True)
assert node.key2.key3 == "asd"
def test_access_options_in_new_nodes(config_dict):
config = ConfigNode(config_dict, strict_access=False)
assert config.key1 == 1
assert config.key10 is None
config.update("key10.key11", "testme", upsert=True)
assert config.key10.key11 == "testme"
assert config.key10.key12 is None
def test_equality():
node1 = ConfigNode({"a": 2, "b": 4})
node2 = ConfigNode({"a": 2, "b": 4})
node3 = ConfigNode({"a": 2, "c": 4})
node4 = ConfigNode({"a": 2, "c": 5})
assert node1 == node2
assert node1 != node3
assert node1 != node4
assert node3 != node4
def test_contains(config_dict):
config = ConfigNode(config_dict)
assert "key1" in config
assert "key1.key2" not in config
assert "key2.key3" in config
assert "key2.key4.key5" in config
assert "key2.key4.key6" not in config
def test_merge_env_variable():
prefix = "PYTHONJSONCONFIG"
variables = {f"{prefix}_TESTVALUE1": "bla", f"{prefix}_TESTVALUE2": "1"}
for key, value in variables.items():
os.environ[key] = value
invalid_prefix = "PYTHONJSONCONFIGTEST"
os.environ[invalid_prefix] = "invalid"
config = ConfigNode({"testvalue1": "blub", "testvalue3": 5})
config.merge_with_env_variables(prefix)
assert config.testvalue1 == "bla"
assert config.testvalue2 == "1"
assert config.testvalue3 == 5
assert "test" not in config
for key, value in variables.items():
del os.environ[key]
def test_strict_access(config_dict):
config = ConfigNode(config_dict,
strict_access=True,
optional_fields=["nokey", ["key2", "nokey"], "key2.nokey2"])
assert config.nokey is None
assert config.key2.nokey is None
assert config.key2.nokey2 is None
with pytest.raises(AttributeError):
config.key2.nokey3
config = ConfigNode(config_dict,
strict_access=False,
required_fields=["nokey", ["key2", "nokey"], "key2.nokey2"])
assert config.key2.nokey3 is None
with pytest.raises(AttributeError):
assert config.nokey is None
with pytest.raises(AttributeError):
assert config.key2.nokey is None
with pytest.raises(AttributeError):
assert config.key2.nokey2 is None
def test_update(config_dict):
node = ConfigNode(config_dict)
assert node.key1 == 1
node.update("key1", 2)
assert node.key1 == 2
node.update("key1", {"newkey": 1})
assert isinstance(node.key1, ConfigNode)
assert node.key1.newkey == 1
assert isinstance(node.key2.key4, ConfigNode)
node.update("key2.key4", 1337)
assert node.key2.key4 == 1337
def test_creation(config_dict):
node = ConfigNode(config_dict)
assert node.key1 == 1
assert node.__dict__["_ConfigNode__path"] == []
assert isinstance(node.key2, ConfigNode)
nested_node = node.key2
assert nested_node.key3 == 3
assert nested_node.__dict__["_ConfigNode__path"] == ["key2"]
assert isinstance(nested_node.key4, ConfigNode)
nested_node = nested_node.key4
assert nested_node.key5 == 5
assert nested_node.__dict__["_ConfigNode__path"] == ["key2", "key4"]
def test_iteration(config_dict):
config = ConfigNode(config_dict)
for iter_key, key, value, item in zip(config, config.keys(),
config.values(), config.items()):
item_key, item_value = item
assert iter_key == key
assert item_key == key
assert item_value == value
assert config.get(key) == value
def test_get(config_dict):
node = ConfigNode(config_dict)
assert node.key1 == 1
with pytest.raises(AttributeError):
node.nokey
assert node.get("key1") == 1
with pytest.raises(AttributeError):
node.get("nokey")
assert node.get("key2.key3") == 3
def test_upsert():
node = ConfigNode({"key1": 1})
with pytest.raises(AttributeError):
node.key2
with pytest.raises(RuntimeError):
node.update("key2", "asd", upsert=False)
node.update("key2", "asd", upsert=True)
assert node.key2 == "asd"
def test_add():
node = ConfigNode({"key1": 1})
assert node.key1 == 1
with pytest.raises(AttributeError):
node.key2
node.add("key2", 2)
assert node.key2 == 2
with pytest.raises(AttributeError):
node.key3.key4
node.add("key3.key4", "test")
assert node.key3.key4 == "test"
def test_add_overwrite():
node = ConfigNode({"key1": 1})
assert node.key1 == 1
with warnings.catch_warnings(record=True) as warning_log:
assert len(warning_log) == 0
node.add("key1", 2, overwrite=False)
assert len(warning_log) == 1
assert node.key1 == 2
node.add("key1", 3)
assert len(warning_log) == 1
assert node.key1 == 3
def test_string():
node = ConfigNode({1: 2, 3: 4})
node_str = "ConfigNode(path=[], values={1: 2, 3: 4}, strict_access=True, required_fields=[], optional_fields" \
"=[])"
assert str(node) == node_str
def test_iter(config_dict):
config = ConfigNode(config_dict)
expected_keys = [
"key1", "key2.key3", "key2.key4.key5", "key2.key6", "key7"
]
assert list(config) == expected_keys
def test_values(config_dict):
config = ConfigNode(config_dict)
values = list(config.values())
assert values == [1, 3, 5, 6, 7]
def test_items(config_dict):
config = ConfigNode(config_dict)
items = list(config.items())
expected_items = [("key1", 1), ("key2.key3", 3), ("key2.key4.key5", 5),
("key2.key6", 6), ("key7", 7)]
assert items == expected_items