adding asciidoc generator tests

Author: ebeahan

scripts/tests/test_asciidoc_fields.py

@@ -0,0 +1,93 @@
+import os
+import sys
+import unittest
+
+sys.path.append(os.path.join(os.path.dirname(__file__), '..'))
+
+from scripts.generators import asciidoc_fields
+from scripts.generators import intermediate_files
+from scripts.schema import cleaner
+from scripts.schema import loader
+from scripts.schema import finalizer
+
+
+class TestGeneratorsAsciiFields(unittest.TestCase):
+
+    @classmethod
+    def setUpClass(cls):
+        fields = loader.load_schemas()
+        cleaner.clean(fields)
+        finalizer.finalize(fields)
+        cls.nested = intermediate_files.generate_nested_fields(fields)
+
+    def setUp(self):
+        self.fieldsets = TestGeneratorsAsciiFields.nested
+        self.event_fieldset = self.fieldsets['event']
+        self.foo_fieldset = self.dummy_fieldset()
+
+    def dummy_fieldset(self):
+        return {
+	        'description': 'foo',
+	        'fields': {
+	        	'foo.type': {
+	        		'dashed_name': 'foo-type',
+	        		'description': 'describes the foo',
+	        		'example': '2016-05-23T08:05:34.853Z',
+	        		'flat_name': 'foo.type',
+	        		'level': 'core',
+	        		'name': 'type',
+	        		'normalize': ['array'],
+	        		'short': 'describes the foo',
+	        		'ignore_above': 1024,
+	        		'type': 'keyword',
+	        		'allowed_values': [{
+	        				'description': 'fluffy foo',
+	        				'name': 'fluffy',
+	        			},
+	        			{
+	        				'description': 'coarse foo',
+	        				'name': 'coarse',
+	        			}
+	        		]
+	        	},
+          	    'foo.id': {
+			        'dashed_name': 'foo-id',
+			        'description': 'Unique ID of the foo.',
+			        'example': 'foo123',
+			        'flat_name': 'foo.id',
+			        'ignore_above': 1024,
+			        'level': 'core',
+			        'name': 'id',
+			        'normalize': [],
+			        'short': 'Unique ID of the foo.',
+			        'type': 'keyword'
+                }
+            },
+	        'group': 2,
+	        'name': 'foo',
+	        'prefix': 'foo.',
+	        'short': 'Foo fields',
+	        'title': 'Foo',
+	        'type': 'group'
+        }
+
+    def test_validate_sort_event_fieldset(self):
+        sorted_event_fields = asciidoc_fields.sort_fields(self.event_fieldset)
+        # Is a list
+        self.assertIsInstance(sorted_event_fields, list)
+
+        # `allowed_value_names` key and list value is present
+        for field in sorted_event_fields:
+            self.assertIsInstance(field.get('allowed_value_names'), list)
+
+    def test_validate_sort_dummy_fieldset(self):
+        sorted_foo_fields = asciidoc_fields.sort_fields(self.foo_fieldset)
+        #import pdb;pdb.set_trace()
+        self.assertIsInstance(sorted_foo_fields, list)
+        self.assertEqual('id', sorted_foo_fields[0]['name'])
+        self.assertEqual('type', sorted_foo_fields[1]['name'])
+        self.assertIn('fluffy', sorted_foo_fields[1]['allowed_value_names'])
+        self.assertIn('coarse', sorted_foo_fields[1]['allowed_value_names'])
+
+if __name__ == '__main__':
+    unittest.main()