Home | History | Annotate | Download | only in tests 
      1 # Copyright 2014 The Chromium Authors. All rights reserved.
      2 # Use of this source code is governed by a BSD-style license that can be
      3 # found in the LICENSE file.
      4 
      5 import math
      6 import unittest
      7 
      8 # pylint: disable=F0401
      9 import mojo.bindings.reflection as reflection
     10 import mojo.system
     11 
     12 # Generated files
     13 # pylint: disable=F0401
     14 import sample_import_mojom
     15 import sample_import2_mojom
     16 import sample_service_mojom
     17 
     18 
     19 def _NewHandle():
     20   return mojo.system.MessagePipe().handle0
     21 
     22 
     23 def _TestEquality(x, y):
     24   if x == y:
     25     return True
     26 
     27   if type(x) != type(y):
     28     print '\n%r != %r. Element are not of the same type.' % (x, y)
     29     return False
     30 
     31   if isinstance(x, float) and math.isnan(x) and math.isnan(y):
     32     return True
     33 
     34   if hasattr(x, '__len__'):
     35     if len(x) != len(y):
     36       print '\n%r != %r. Iterables are not of the same size.' % (x, y)
     37       return False
     38     for (x1, y1) in zip(x, y):
     39       if not _TestEquality(x1, y1):
     40         return False
     41     return True
     42 
     43   if (hasattr(x, '__metaclass__') and
     44       x.__metaclass__ == reflection.MojoStructType):
     45     properties = [p for p in dir(x) if not p.startswith('_')]
     46     for p in properties:
     47       p1 = getattr(x, p)
     48       p2 = getattr(y, p)
     49       if not hasattr(p1, '__call__') and not _TestEquality(p1, p2):
     50         print '\n%r != %r. Not equal for property %r.' % (x, y, p)
     51         return False
     52     return True
     53 
     54   return False
     55 
     56 
     57 def _NewBar():
     58   bar_instance = sample_service_mojom.Bar()
     59   bar_instance.alpha = 22
     60   bar_instance.beta = 87
     61   bar_instance.gamma = 122
     62   bar_instance.type = sample_service_mojom.Bar.Type.BOTH
     63   return bar_instance
     64 
     65 
     66 def _NewFoo():
     67   foo_instance = sample_service_mojom.Foo()
     68   foo_instance.name = "Foo.name"
     69   foo_instance.x = 23
     70   foo_instance.y = -23
     71   foo_instance.a = False
     72   foo_instance.b = True
     73   foo_instance.c = True
     74   foo_instance.bar = _NewBar()
     75   foo_instance.extra_bars = [
     76       _NewBar(),
     77       _NewBar(),
     78   ]
     79   foo_instance.data = 'Hello world'
     80   foo_instance.source = _NewHandle()
     81   foo_instance.input_streams = [ _NewHandle() ]
     82   foo_instance.output_streams = [ _NewHandle(), _NewHandle() ]
     83   foo_instance.array_of_array_of_bools = [ [ True, False ], [] ]
     84   foo_instance.multi_array_of_strings = [
     85       [
     86           [ "1", "2" ],
     87           [],
     88           [ "3", "4" ],
     89       ],
     90       [],
     91   ]
     92   foo_instance.array_of_bools = [ True, 0, 1, 2, 0, 0, 0, 0, 0, True ]
     93   return foo_instance
     94 
     95 
     96 class SerializationDeserializationTest(unittest.TestCase):
     97 
     98   def testTestEquality(self):
     99     self.assertFalse(_TestEquality(1, 2))
    100 
    101   def testFooSerialization(self):
    102     (data, _) = _NewFoo().Serialize()
    103     self.assertTrue(len(data))
    104     self.assertEquals(len(data) % 8, 0)
    105 
    106   def testFooDeserialization(self):
    107     (data, handles) = _NewFoo().Serialize()
    108     self.assertTrue(
    109         sample_service_mojom.Foo.Deserialize(data, handles))
    110 
    111   def testFooSerializationDeserialization(self):
    112     foo1 = _NewFoo()
    113     (data, handles) = foo1.Serialize()
    114     foo2 = sample_service_mojom.Foo.Deserialize(data, handles)
    115     self.assertTrue(_TestEquality(foo1, foo2))
    116 
    117   def testDefaultsTestSerializationDeserialization(self):
    118     v1 = sample_service_mojom.DefaultsTest()
    119     v1.a18 = []
    120     v1.a19 = ""
    121     v1.a21 = sample_import_mojom.Point()
    122     v1.a22.location = sample_import_mojom.Point()
    123     v1.a22.size = sample_import2_mojom.Size()
    124     (data, handles) = v1.Serialize()
    125     v2 = sample_service_mojom.DefaultsTest.Deserialize(data, handles)
    126     self.assertTrue(_TestEquality(v1, v2))
    127 
    128   def testFooDeserializationError(self):
    129     with self.assertRaises(Exception):
    130       sample_service_mojom.Foo.Deserialize("", [])
    131