[CORD-1440] Moving the generative toolchain in a library
Change-Id: Ifa8e8f930ac34e1f8952099b7e34842a52f4664d
diff --git a/lib/xos-genx/tests/django_generator_test.py b/lib/xos-genx/tests/django_generator_test.py
new file mode 100644
index 0000000..08997b7
--- /dev/null
+++ b/lib/xos-genx/tests/django_generator_test.py
@@ -0,0 +1,27 @@
+import unittest
+import os
+from xosgenx.generator import XOSGenerator
+from helpers import FakeArgs
+
+VROUTER_XPROTO = os.path.abspath(os.path.dirname(os.path.realpath(__file__)) + "/xproto/vrouterport.xproto")
+
+# Generate Protobuf from Xproto and then parse the resulting Protobuf
+class XProtoProtobufGeneratorTest(unittest.TestCase):
+ def test_proto_generator(self):
+ """
+ [XOS-GenX] Generate DJANGO models, verify Fields and Foreign Keys
+ """
+ args = FakeArgs()
+ args.files = [VROUTER_XPROTO]
+ args.target = 'django.xtarget'
+ output = XOSGenerator.generate(args)
+
+ fields = filter(lambda s:'Field(' in s, output.splitlines())
+ self.assertEqual(len(fields), 2)
+ links = filter(lambda s:'Key(' in s, output.splitlines())
+ self.assertEqual(len(links), 2)
+
+if __name__ == '__main__':
+ unittest.main()
+
+