mojo: Adding struct's fields for python bindings

mojo/python/tests/bindings_structs_unittest.py

+# Copyright 2014 The Chromium Authors. All rights reserved.
+# Use of this source code is governed by a BSD-style license that can be
+# found in the LICENSE file.
+
+import math
+import unittest
+
+# pylint: disable=F0401
+import mojo.system
+
+# Generated files
+# pylint: disable=F0401
+import sample_import_mojom
+import sample_import2_mojom
+import sample_service_mojom
+
+
+class StructBindingsTest(unittest.TestCase):
+
+  def testModule(self):
+    self.assertEquals(sample_service_mojom.DefaultsTest.__module__,
+                      'sample_service_mojom')
+
+  def testDefaultsTest(self):
+    defaults_test = sample_service_mojom.DefaultsTest()
+    self.assertEquals(defaults_test.a0, -12)
+    self.assertEquals(defaults_test.a1, 12)
+    self.assertEquals(defaults_test.a2, 1234)
+    self.assertEquals(defaults_test.a3, 34567)
+    self.assertEquals(defaults_test.a4, 123456)
+    self.assertEquals(defaults_test.a5, 3456789012)
+    self.assertEquals(defaults_test.a6, -111111111111)
+    self.assertEquals(defaults_test.a7, 9999999999999999999)
+    self.assertEquals(defaults_test.a8, 0x12345)
+    self.assertEquals(defaults_test.a9, -0x12345)
+    self.assertEquals(defaults_test.a10, 1234)
+    self.assertEquals(defaults_test.a11, True)
+    self.assertEquals(defaults_test.a12, False)
+    self.assertEquals(defaults_test.a13, 123.25)
+    self.assertEquals(defaults_test.a14, 1234567890.123)
+    self.assertEquals(defaults_test.a15, 1E10)
+    self.assertEquals(defaults_test.a16, -1.2E+20)
+    self.assertEquals(defaults_test.a17, 1.23E-20)
+    self.assertEquals(defaults_test.a18, None)
+    self.assertEquals(defaults_test.a19, None)
+    self.assertEquals(defaults_test.a20, sample_service_mojom.Bar.Type.BOTH)
+    self.assertEquals(defaults_test.a21, None)
+    self.assertTrue(isinstance(defaults_test.a22, sample_import2_mojom.Thing))
+    self.assertEquals(defaults_test.a23, 0xFFFFFFFFFFFFFFFF)
+    self.assertEquals(defaults_test.a24, 0x123456789)
+    self.assertEquals(defaults_test.a25, -0x123456789)
+    self.assertEquals(defaults_test.a26, float('inf'))
+    self.assertEquals(defaults_test.a27, float('-inf'))
+    self.assertTrue(math.isnan(defaults_test.a28))
+    self.assertEquals(defaults_test.a29, float('inf'))
+    self.assertEquals(defaults_test.a30, float('-inf'))
+    self.assertTrue(math.isnan(defaults_test.a31))
+
+  def testNoAliasing(self):
+    foo1 = sample_service_mojom.Foo()
+    foo2 = sample_service_mojom.Foo()
+    foo1.name = "foo1"
+    foo2.name = "foo2"
+    self.assertEquals(foo1.name, "foo1")
+    self.assertEquals(foo2.name, "foo2")
+
+    defaults_test1 = sample_service_mojom.DefaultsTest()
+    defaults_test2 = sample_service_mojom.DefaultsTest()
+    self.assertNotEquals(defaults_test1.a22, defaults_test2.a22)
+
+  def testImmutableAttributeSet(self):
+    foo_instance = sample_service_mojom.Foo()
+    with self.assertRaises(AttributeError):
+      foo_instance.new_attribute = None
+    with self.assertRaises(AttributeError):
+      del foo_instance.name
+
+  def _TestIntegerField(self, entity, field_name, bits, signed):
+    if signed:
+      min_value = -(1 << (bits - 1))
+      max_value = (1 << (bits - 1)) - 1
+    else:
+      min_value = 0
+      max_value = (1 << bits) - 1
+    entity.__setattr__(field_name, min_value)
+    entity.__setattr__(field_name, max_value)
+    with self.assertRaises(ValueError):
+      entity.__setattr__(field_name, None)
+    with self.assertRaises(ValueError):
+      entity.__setattr__(field_name, min_value - 1)
+    with self.assertRaises(ValueError):
+      entity.__setattr__(field_name, max_value + 1)
+    with self.assertRaises(ValueError):
+      entity.__setattr__(field_name, 'hello world')
+
+  def testTypes(self):
+    defaults_test = sample_service_mojom.DefaultsTest()
+    # Integer types
+    self._TestIntegerField(defaults_test, 'a0', 8, True)
+    self._TestIntegerField(defaults_test, 'a1', 8, False)
+    self._TestIntegerField(defaults_test, 'a2', 16, True)
+    self._TestIntegerField(defaults_test, 'a3', 16, False)
+    self._TestIntegerField(defaults_test, 'a4', 32, True)
+    self._TestIntegerField(defaults_test, 'a5', 32, False)
+    self._TestIntegerField(defaults_test, 'a6', 64, True)
+    self._TestIntegerField(defaults_test, 'a7', 64, False)
+
+    # Boolean types
+    defaults_test.a11 = False
+    self.assertEquals(defaults_test.a11, False)
+    defaults_test.a11 = None
+    self.assertEquals(defaults_test.a11, False)
+    defaults_test.a11 = []
+    self.assertEquals(defaults_test.a11, False)
+    defaults_test.a12 = True
+    self.assertEquals(defaults_test.a12, True)
+    defaults_test.a12 = 1
+    self.assertEquals(defaults_test.a12, True)
+    defaults_test.a12 = [[]]
+    self.assertEquals(defaults_test.a12, True)
+
+    # Floating point types
+    with self.assertRaises(ValueError):
+      defaults_test.a13 = 'hello'
+    with self.assertRaises(ValueError):
+      defaults_test.a14 = 'hello'
+
+    # Array type
+    defaults_test.a18 = None
+    defaults_test.a18 = []
+    defaults_test.a18 = [ 0 ]
+    defaults_test.a18 = [ 255 ]
+    defaults_test.a18 = [ 0, 255 ]
+    with self.assertRaises(ValueError):
+      defaults_test.a18 = [[]]
+    with self.assertRaises(ValueError):
+      defaults_test.a18 = [ -1 ]
+    with self.assertRaises(ValueError):
+      defaults_test.a18 = [ 256 ]
+
+    # String type
+    defaults_test.a19 = None
+    defaults_test.a19 = ''
+    defaults_test.a19 = 'hello world'
+    with self.assertRaises(ValueError):
+      defaults_test.a19 = [[]]
+    with self.assertRaises(ValueError):
+      defaults_test.a19 = [ -1 ]
+    with self.assertRaises(ValueError):
+      defaults_test.a19 = [ 256 ]
+
+    # Structs
+    defaults_test.a21 = None
+    defaults_test.a21 = sample_import_mojom.Point()
+    with self.assertRaises(ValueError):
+      defaults_test.a21 = 1
+    with self.assertRaises(ValueError):
+      defaults_test.a21 = sample_import2_mojom.Thing()
+
+    # Handles
+    foo_instance = sample_service_mojom.Foo()
+    foo_instance.source = None
+    foo_instance.source = mojo.system.Handle()
+    with self.assertRaises(ValueError):
+      foo_instance.source = 1
+    with self.assertRaises(ValueError):
+      foo_instance.source = object()
+
+  def testConstructor(self):
+    bar_instance = sample_service_mojom.Bar()
+    foo_instance = sample_service_mojom.Foo(name="Foo",
+                                            x=-1,
+                                            y=5,
+                                            a=False,
+                                            bar=bar_instance)
+    self.assertEquals(foo_instance.name, "Foo")
+    self.assertEquals(foo_instance.x, -1)
+    self.assertEquals(foo_instance.y, 5)
+    self.assertEquals(foo_instance.a, False)
+    self.assertEquals(foo_instance.bar, bar_instance)
+