[CORD-1338] Adding onetomany relations to modeldefs
Change-Id: I213ebf807984adcc69edee91252ccd53f6ab1c25
diff --git a/lib/xos-genx/tests/optimize_test.py b/lib/xos-genx/tests/optimize_test.py
index af40575..0d2c1b1 100644
--- a/lib/xos-genx/tests/optimize_test.py
+++ b/lib/xos-genx/tests/optimize_test.py
@@ -1,6 +1,5 @@
import unittest
from xosgenx.jinja2_extensions.fol2 import FOL2Python
-import vimpdb
class XProtoOptimizeTest(unittest.TestCase):
def setUp(self):
diff --git a/lib/xos-genx/tests/translator_test.py b/lib/xos-genx/tests/translator_test.py
index e82af6d..fad4dba 100644
--- a/lib/xos-genx/tests/translator_test.py
+++ b/lib/xos-genx/tests/translator_test.py
@@ -228,6 +228,33 @@
output = XOSGenerator.generate(args)
self.assertNotIn('default:', output)
+ def test_one_to_many_in_modeldef(self):
+ xproto = \
+"""
+option app_label = "test";
+
+message ServiceDependency {
+ required manytoone provider_service->Service:provided_dependencies = 1;
+ required manytoone subscriber_service->Service:subscribed_dependencies = 2;
+}
+
+message Service {
+ required string name = 1;
+}
+"""
+
+ args = FakeArgs()
+ args.inputs = xproto
+ args.target = 'modeldefs.xtarget'
+ output = XOSGenerator.generate(args)
+ # Service deps model
+ self.assertIn('{model: Service, type: manytoone, on_field: provider_service}', output)
+ self.assertIn('{model: Service, type: manytoone, on_field: provider_service}', output)
+
+ # Service model
+ self.assertIn('{model: ServiceDependency, type: onetomany, on_field: provider_service}', output)
+ self.assertIn('{model: ServiceDependency, type: onetomany, on_field: provider_service}', output)
+
if __name__ == '__main__':
unittest.main()