summaryrefslogtreecommitdiff
path: root/tests/plugins
diff options
context:
space:
mode:
Diffstat (limited to 'tests/plugins')
-rw-r--r--tests/plugins/yamlrdr.py174
1 files changed, 138 insertions, 36 deletions
diff --git a/tests/plugins/yamlrdr.py b/tests/plugins/yamlrdr.py
index 7f65624..103d6e8 100644
--- a/tests/plugins/yamlrdr.py
+++ b/tests/plugins/yamlrdr.py
@@ -16,26 +16,54 @@ class TestYamlReader(ChrysalideTestCase):
super(TestYamlReader, cls).setUpClass()
- cls._simple = tempfile.NamedTemporaryFile()
-
- cls.simple_data = b'''
-meta:
- id: java_class
- endian: be
- file-extension: class
- xref:
- justsolve: Java
- pronom: x-fmt/415
- 'wiki"data': Q2193155
- license: CC0-1.0
+ cls._simple_map = tempfile.NamedTemporaryFile()
+
+ cls._simple_map_data = b'''
+a: av
+b: bv
+c: cv
+
+'''
+
+ cls._simple_seq = tempfile.NamedTemporaryFile()
+
+ cls._simple_seq_data = b'''
+- a: av
+- b: bv
+- c: cv
'''
- cls._simple.write(cls.simple_data)
+ cls._nested = tempfile.NamedTemporaryFile()
+
+ cls._nested_data = b'''
+root:
+ a: v0
+ b: v1
+ c: v2
+ sub:
+ aa: v00
+ bb: v01
+ cc: v02
+ - i: w
+ - j: x
+ - k: c
+ d: v3
- cls._simple.flush()
+'''
- cls.log('Using temporary file "%s"' % cls._simple.name)
+ tmp = [
+ [ cls._simple_map, cls._simple_map_data ],
+ [ cls._simple_seq, cls._simple_seq_data ],
+ [ cls._nested, cls._nested_data ]
+ ]
+
+ for f, d in tmp:
+
+ f.write(d)
+ f.flush()
+
+ cls.log('Using temporary file "%s"' % f.name)
@classmethod
@@ -43,52 +71,126 @@ meta:
super(TestYamlReader, cls).tearDownClass()
- cls.log('Delete file "%s"' % cls._simple.name)
+ tmp = [
+ cls._simple_map,
+ cls._simple_seq,
+ cls._nested,
+ ]
- cls._simple.close()
+ for f in tmp:
+ cls.log('Delete file "%s"' % f.name)
- def testSimpleYamlContent(self):
- """Validate simple Yaml content reader."""
+ f.close()
- reader = YamlReader.new_from_path(self._simple.name)
- self.assertIsNotNone(reader)
+ def testSimpleYamlContent(self):
+ """Validate Yaml content readers."""
def _build_node_desc(node, left):
line = node.yaml_line
- desc = left + line.key + ':' + (' ' + line.value if line.value else '') + '\n'
+ if line:
+ prefix = '- ' if line.is_list_item else ''
+ desc = left + prefix + line.key + ':' + (' ' + line.value if line.value else '') + '\n'
+ indent = ' '
+ else:
+ desc = ''
+ indent = ''
- for child in node.children:
- desc += _build_node_desc(child, left + ' ')
+ if node.collection:
+ for child in node.collection.nodes:
+ desc += _build_node_desc(child, left + indent)
return desc
- fulldesc = ''
+ reader = YamlReader.new_from_path(self._simple_map.name)
+ self.assertIsNotNone(reader)
+ self.assertIsNotNone(reader.tree)
+
+ fulldesc = _build_node_desc(reader.tree.root, '')
+
+ self.assertEqual('\n' + fulldesc + '\n', self._simple_map_data.decode('ascii'))
+
+ reader = YamlReader.new_from_path(self._simple_seq.name)
+ self.assertIsNotNone(reader)
+ self.assertIsNotNone(reader.tree)
+
+ fulldesc = _build_node_desc(reader.tree.root, '')
+
+ self.assertEqual('\n' + fulldesc + '\n', self._simple_seq_data.decode('ascii'))
+
+ reader = YamlReader.new_from_path(self._nested.name)
+ self.assertIsNotNone(reader)
+ self.assertIsNotNone(reader.tree)
- for rn in reader.tree.root_nodes:
- fulldesc += _build_node_desc(rn, '')
+ fulldesc = _build_node_desc(reader.tree.root, '')
- self.assertEqual('\n' + fulldesc + '\n', self.simple_data.decode('ascii'))
+ self.assertEqual('\n' + fulldesc + '\n', self._nested_data.decode('ascii'))
def testSimpleYamlContentFinder(self):
- """Validate simple Yaml content search."""
+ """Validate Yaml nested content search."""
- reader = YamlReader.new_from_path(self._simple.name)
+ reader = YamlReader.new_from_path(self._nested.name)
self.assertIsNotNone(reader)
- found = reader.tree.find_node_by_path('/meta/xref')
+ found = reader.tree.find_by_path('/root/sub')
+
+ self.assertEqual(len(found), 1)
+
+ if len(found) == 1:
+ self.assertEqual(found[0].yaml_line.key, 'sub')
+
+ found = reader.tree.find_by_path('/root/sub/')
+
+ self.assertEqual(len(found), 3)
+
+ found = reader.tree.find_by_path('/root/sub/xx')
+
+ self.assertEqual(len(found), 0)
+
+ found = reader.tree.find_by_path('/root/sub/cc/i')
+
+ self.assertEqual(len(found), 1)
+
+ if len(found) == 1:
+ self.assertEqual(found[0].yaml_line.key, 'i')
+ self.assertEqual(found[0].yaml_line.is_list_item, True)
+
+ found = reader.tree.find_by_path('/root/sub/cc')
+
+ self.assertEqual(len(found), 1)
+
+ if len(found) == 1:
+
+ root = found[0]
+
+ found = root.find_by_path('cc/i')
+
+ self.assertEqual(len(found), 1)
+
+ if len(found) == 1:
+
+ self.assertEqual(found[0].yaml_line.key, 'i')
+ self.assertEqual(found[0].yaml_line.is_list_item, True)
+
+ found = root.find_by_path('/cc/i')
+
+ self.assertEqual(len(found), 1)
+
+ if len(found) == 1:
- self.assertEqual(found.yaml_line.key, 'xref')
+ self.assertEqual(found[0].yaml_line.key, 'i')
+ self.assertEqual(found[0].yaml_line.is_list_item, True)
- found = reader.tree.find_node_by_path('/meta/xref/')
+ found = root.find_by_path('//i')
- self.assertEqual(found.yaml_line.key, 'xref')
+ self.assertEqual(len(found), 1)
- found = reader.tree.find_node_by_path('/meta/xref/aa')
+ if len(found) == 1:
- self.assertIsNone(found)
+ self.assertEqual(found[0].yaml_line.key, 'i')
+ self.assertEqual(found[0].yaml_line.is_list_item, True)