mirror of
https://github.com/freedombox/FreedomBox.git
synced 2026-05-13 10:30:16 +00:00
utils: Fix some flake8 warnings
Signed-off-by: Sunil Mohan Adapa <sunil@medhas.org> Reviewed-by: James Valleroy <jvalleroy@mailbox.org>
This commit is contained in:
parent
d3bdaf0729
commit
96e7fd3e40
@ -94,32 +94,32 @@ class TestYAMLFileUtil(TestCase):
|
|||||||
"""
|
"""
|
||||||
Update an empty YAML file with content.
|
Update an empty YAML file with content.
|
||||||
"""
|
"""
|
||||||
fp = tempfile.NamedTemporaryFile()
|
test_file = tempfile.NamedTemporaryFile()
|
||||||
conf = {'property1': self.kv_pair}
|
conf = {'property1': self.kv_pair}
|
||||||
|
|
||||||
with YAMLFile(fp.name) as file_conf:
|
with YAMLFile(test_file.name) as file_conf:
|
||||||
for key in conf.keys():
|
for key in conf:
|
||||||
file_conf[key] = conf[key]
|
file_conf[key] = conf[key]
|
||||||
|
|
||||||
with open(fp.name, 'r') as retrieved_conf:
|
with open(test_file.name, 'r') as retrieved_conf:
|
||||||
assert retrieved_conf.read() == ruamel.yaml.round_trip_dump(conf)
|
assert retrieved_conf.read() == ruamel.yaml.round_trip_dump(conf)
|
||||||
|
|
||||||
def test_update_non_empty_yaml_file(self):
|
def test_update_non_empty_yaml_file(self):
|
||||||
"""
|
"""
|
||||||
Update a non-empty YAML file with modifications
|
Update a non-empty YAML file with modifications
|
||||||
"""
|
"""
|
||||||
fp = tempfile.NamedTemporaryFile()
|
test_file = tempfile.NamedTemporaryFile()
|
||||||
|
|
||||||
with open(fp.name, 'w') as conf_file:
|
with open(test_file.name, 'w') as conf_file:
|
||||||
conf_file.write(
|
conf_file.write(
|
||||||
ruamel.yaml.round_trip_dump({
|
ruamel.yaml.round_trip_dump({
|
||||||
'property1': self.kv_pair
|
'property1': self.kv_pair
|
||||||
}))
|
}))
|
||||||
|
|
||||||
with YAMLFile(fp.name) as file_conf:
|
with YAMLFile(test_file.name) as file_conf:
|
||||||
file_conf['property2'] = self.kv_pair
|
file_conf['property2'] = self.kv_pair
|
||||||
|
|
||||||
with open(fp.name, 'r') as retrieved_conf:
|
with open(test_file.name, 'r') as retrieved_conf:
|
||||||
file_conf = ruamel.yaml.round_trip_load(retrieved_conf)
|
file_conf = ruamel.yaml.round_trip_load(retrieved_conf)
|
||||||
assert file_conf == {'property1': self.kv_pair,
|
assert file_conf == {'property1': self.kv_pair,
|
||||||
'property2': self.kv_pair}
|
'property2': self.kv_pair}
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user