mirror of
https://github.com/freedombox/FreedomBox.git
synced 2026-01-21 07:55:00 +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.
|
||||
"""
|
||||
fp = tempfile.NamedTemporaryFile()
|
||||
test_file = tempfile.NamedTemporaryFile()
|
||||
conf = {'property1': self.kv_pair}
|
||||
|
||||
with YAMLFile(fp.name) as file_conf:
|
||||
for key in conf.keys():
|
||||
with YAMLFile(test_file.name) as file_conf:
|
||||
for key in conf:
|
||||
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)
|
||||
|
||||
def test_update_non_empty_yaml_file(self):
|
||||
"""
|
||||
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(
|
||||
ruamel.yaml.round_trip_dump({
|
||||
'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
|
||||
|
||||
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)
|
||||
assert file_conf == {'property1': self.kv_pair,
|
||||
'property2': self.kv_pair}
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user