[CORD-1440] Moving the generative toolchain in a library
Change-Id: Ifa8e8f930ac34e1f8952099b7e34842a52f4664d
diff --git a/lib/xos-genx/tests/test_generator.py b/lib/xos-genx/tests/test_generator.py
new file mode 100644
index 0000000..a13fb38
--- /dev/null
+++ b/lib/xos-genx/tests/test_generator.py
@@ -0,0 +1,212 @@
+import unittest
+import os
+from helpers import FakeArgs, OUTPUT_DIR
+from xosgenx.generator import XOSGenerator
+
+TEST_EXPECTED_OUTPUT = """
+ name: XOSModel
+ fields:
+ name:
+ type: string
+ description: "Help Name"
+ files:
+ type: string
+ description: "Help Files"
+"""
+
+BASE_XPROTO = os.path.abspath(os.path.dirname(os.path.realpath(__file__)) + "/xproto/base.xproto")
+TEST_XPROTO = os.path.abspath(os.path.dirname(os.path.realpath(__file__)) + "/xproto/test.xproto")
+SKIP_DJANGO_XPROTO = os.path.abspath(os.path.dirname(os.path.realpath(__file__)) + "/xproto/skip_django.xproto")
+VROUTER_XPROTO = os.path.abspath(os.path.dirname(os.path.realpath(__file__)) + "/xproto/vrouterport.xproto")
+TEST_TARGET = os.path.abspath(os.path.dirname(os.path.realpath(__file__)) + "/xtarget/test.xtarget")
+SPLIT_TARGET = os.path.abspath(os.path.dirname(os.path.realpath(__file__)) + "/xtarget/split.xtarget")
+
+TEST_ATTICS = os.path.abspath(os.path.dirname(os.path.realpath(__file__)) + "/attics/")
+
+class XOSGeneratorTest(unittest.TestCase):
+ """
+ Testing the XOS Generative Toolchain
+ """
+
+ def setUp(self):
+ filesToRemove = [f for f in os.listdir(OUTPUT_DIR)]
+ for f in filesToRemove:
+ if not f.startswith('.'):
+ os.remove(OUTPUT_DIR + '/' + f)
+
+ def _test_generator_custom_target_from_file(self):
+ """
+ [XOS-GenX] Generate output from base.xproto
+ """
+ args = FakeArgs()
+ args.files = [TEST_XPROTO]
+ args.target = TEST_TARGET
+ output = XOSGenerator.generate(args)
+ self.assertEqual(output, TEST_EXPECTED_OUTPUT)
+
+ def _test_generator_custom_target_from_inputs(self):
+ """
+ [XOS-GenX] Generate output from base.xproto
+ """
+ args = FakeArgs()
+ args.inputs = open(TEST_XPROTO).read()
+ args.target = TEST_TARGET
+ output = XOSGenerator.generate(args)
+ self.assertEqual(output, TEST_EXPECTED_OUTPUT)
+
+ def _test_generator_tosca(self):
+ """
+ [XOS-GenX] Generate TOSCA from base.xproto, and write to file
+ """
+ args = FakeArgs()
+ args.files = [TEST_XPROTO, VROUTER_XPROTO]
+ args.target = 'tosca.xtarget'
+ args.output = OUTPUT_DIR
+ args.write_to_file = 'single'
+ args.dest_file = 'base.yaml'
+ XOSGenerator.generate(args)
+
+ dest_file = OUTPUT_DIR + '/' + args.dest_file
+ self.assertTrue(os.path.isfile(dest_file))
+
+ output = open(dest_file, "r").read()
+ self.assertIn("tosca.nodes.XOSModel", output)
+ self.assertIn("tosca.nodes.VRouterPort", output)
+
+ def _test_django_with_attic(self):
+ """
+ [XOS-GenX] Generate django output from test.xproto
+ """
+ args = FakeArgs()
+ args.files = [TEST_XPROTO, VROUTER_XPROTO]
+ args.target = 'django.xtarget'
+ args.attic = TEST_ATTICS
+ args.output = OUTPUT_DIR
+ args.dest_extension = 'py'
+ args.write_to_file = 'model'
+ output = XOSGenerator.generate(args)
+
+ # xosmodel has custom header attic
+ self.assertIn('from xosmodel_header import *', output['XOSModel'])
+ self.assertIn('class XOSModel(XOSBase):', output['XOSModel'])
+
+ # vrouter port use the default header
+ self.assertIn('header import *', output['VRouterPort'])
+ self.assertIn('class VRouterPort(XOSBase):', output['VRouterPort'])
+
+ #verify files
+ xosmodel = OUTPUT_DIR + '/xosmodel.py'
+ self.assertTrue(os.path.isfile(xosmodel))
+ xmf = open(xosmodel).read()
+ self.assertIn('from xosmodel_header import *', xmf)
+ self.assertIn('class XOSModel(XOSBase):', xmf)
+
+ vrouterport = OUTPUT_DIR + '/vrouterport.py'
+ self.assertTrue(os.path.isfile(vrouterport))
+ vrpf = open(vrouterport).read()
+ self.assertIn('header import *', vrpf)
+ self.assertIn('class VRouterPort(XOSBase):', vrpf)
+
+ def _test_django_with_base(self):
+ args = FakeArgs()
+ args.files = [TEST_XPROTO, BASE_XPROTO]
+ args.target = 'django.xtarget'
+ args.attic = TEST_ATTICS
+ args.output = OUTPUT_DIR
+ args.dest_extension = 'py'
+ args.write_to_file = 'model'
+ output = XOSGenerator.generate(args)
+
+ # verify files
+ xosmodel = OUTPUT_DIR + '/xosmodel.py'
+ self.assertTrue(os.path.isfile(xosmodel))
+ xmf = open(xosmodel).read()
+ self.assertIn('from xosmodel_header import *', xmf)
+ self.assertIn('class XOSModel(XOSBase):', xmf)
+
+ xosbase = OUTPUT_DIR + '/xosbase.py'
+ self.assertTrue(os.path.isfile(xosbase))
+ xbf = open(xosbase).read()
+ self.assertIn('header import *', xbf)
+ self.assertIn('class XOSBase(models.Model, PlModelMixIn):', xbf)
+
+ def _test_write_multiple_files(self):
+ """
+ [XOS-GenX] read multiple models as input, print one file per model
+ """
+ args = FakeArgs()
+ args.files = [TEST_XPROTO, VROUTER_XPROTO]
+ args.target = TEST_TARGET
+ args.output = OUTPUT_DIR
+ args.dest_extension = 'txt'
+ args.write_to_file = 'model'
+ XOSGenerator.generate(args)
+
+ generated_files = [f for f in os.listdir(OUTPUT_DIR) if not f.startswith('.')]
+ self.assertEqual(len(generated_files), 2)
+
+ xosmodel = open(os.path.join(OUTPUT_DIR, 'xosmodel.txt'), "r").read()
+ vrouterport = open(os.path.join(OUTPUT_DIR, 'vrouterport.txt'), "r").read()
+
+ self.assertIn("name: XOSModel", xosmodel)
+ self.assertIn("name: VRouterPort", vrouterport)
+
+ def test_write_multiple_files_from_xtarget(self):
+ """
+ [XOS-GenX] read multiple models as input, print separate files based on +++
+ """
+ args = FakeArgs()
+ args.files = [TEST_XPROTO, VROUTER_XPROTO]
+ args.target = SPLIT_TARGET
+ args.output = OUTPUT_DIR
+ args.write_to_file = 'target'
+ XOSGenerator.generate(args)
+
+ generated_files = [f for f in os.listdir(OUTPUT_DIR) if not f.startswith('.')]
+ self.assertEqual(len(generated_files), 2)
+
+ xosmodel = open(os.path.join(OUTPUT_DIR, 'xosmodel.txt'), "r").read()
+ vrouterport = open(os.path.join(OUTPUT_DIR, 'vrouterport.txt'), "r").read()
+
+ self.assertIn("name: XOSModel", xosmodel)
+ self.assertIn("name: VRouterPort", vrouterport)
+
+ def _test_skip_django(self):
+ args = FakeArgs()
+ args.files = [SKIP_DJANGO_XPROTO]
+ args.target = 'django.xtarget'
+ args.output = OUTPUT_DIR
+ args.dest_extension = 'py'
+ args.write_to_file = 'model'
+ output = XOSGenerator.generate(args)
+
+ # should not print a file if options.skip_django = True
+ file = OUTPUT_DIR + '/user.py'
+ self.assertFalse(os.path.isfile(file))
+
+ def test_service_order(self):
+ args = FakeArgs()
+ args.files = [BASE_XPROTO, TEST_XPROTO, VROUTER_XPROTO]
+ args.target = 'service.xtarget'
+ args.output = OUTPUT_DIR
+ args.write_to_file = 'target'
+ output = XOSGenerator.generate(args)
+
+ model = OUTPUT_DIR + '/models.py'
+ self.assertTrue(os.path.isfile(model))
+ line_num = 0
+
+ for line in open(model).readlines():
+ line_num += 1
+ if line.find('class XOSBase(models.Model, PlModelMixIn):') >= 0:
+ base_line = line_num
+ if line.find('XOSModel(XOSBase):') >= 0:
+ xosmodel_line = line_num
+ if line.find('class VRouterPort(XOSBase):') >= 0:
+ vrouter_line = line_num
+ self.assertLess(base_line, xosmodel_line)
+ self.assertLess(xosmodel_line, vrouter_line)
+
+
+if __name__ == '__main__':
+ unittest.main()
\ No newline at end of file