mirror of
https://github.com/ARM-software/devlib.git
synced 2024-10-06 02:40:50 +01:00
33 lines
880 B
Python
33 lines
880 B
Python
|
import os
|
||
|
|
import shutil
|
||
|
|
import tempfile
|
||
|
|
from unittest import TestCase
|
||
|
|
|
||
|
|
from devlib import LocalLinuxTarget
|
||
|
|
|
||
|
|
|
||
|
|
class TestReadTreeValues(TestCase):
|
||
|
|
|
||
|
|
def test_read_multiline_values(self):
|
||
|
|
data = {
|
||
|
|
'test1': '1',
|
||
|
|
'test2': '2\n\n',
|
||
|
|
'test3': '3\n\n4\n\n',
|
||
|
|
}
|
||
|
|
|
||
|
|
tempdir = tempfile.mkdtemp(prefix='devlib-test-')
|
||
|
|
for key, value in data.items():
|
||
|
|
path = os.path.join(tempdir, key)
|
||
|
|
with open(path, 'w') as wfh:
|
||
|
|
wfh.write(value)
|
||
|
|
|
||
|
|
t = LocalLinuxTarget(connection_settings={'unrooted': True})
|
||
|
|
raw_result = t.read_tree_values_flat(tempdir)
|
||
|
|
result = {os.path.basename(k): v for k, v in raw_result.items()}
|
||
|
|
|
||
|
|
shutil.rmtree(tempdir)
|
||
|
|
|
||
|
|
self.assertEqual({k: v.strip()
|
||
|
|
for k, v in data.items()},
|
||
|
|
result)
|