mojo/public/tools/bindings/pylib/mojom_tests/parse/translate_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 imp
import os.path
import sys
import unittest

def _GetDirAbove(dirname):
  """Returns the directory "above" this file containing |dirname| (which must
  also be "above" this file)."""
  path = os.path.abspath(__file__)
  while True:
    path, tail = os.path.split(path)
    assert tail
    if tail == dirname:
      return path

try:
  imp.find_module("mojom")
except ImportError:
  sys.path.append(os.path.join(_GetDirAbove("pylib"), "pylib"))
from mojom.parse import ast
from mojom.parse import translate


class TranslateTest(unittest.TestCase):
  """Tests |parser.Parse()|."""

  def testSimpleArray(self):
    """Tests a simple int32[]."""
    # pylint: disable=W0212
    self.assertEquals(translate._MapKind("int32[]"), "a:i32")

  def testAssociativeArray(self):
    """Tests a simple uint8{string}."""
    # pylint: disable=W0212
    self.assertEquals(translate._MapKind("uint8{string}"), "m[s][u8]")

  def testLeftToRightAssociativeArray(self):
    """Makes sure that parsing is done from right to left on the internal kinds
       in the presence of an associative array."""
    # pylint: disable=W0212
    self.assertEquals(translate._MapKind("uint8[]{string}"), "m[s][a:u8]")

  def testTranslateSimpleUnions(self):
    """Makes sure that a simple union is translated correctly."""
    tree = ast.Mojom(
        None,
        ast.ImportList(),
        [ast.Union("SomeUnion", None, ast.UnionBody(
          [ast.UnionField("a", None, None, "int32"),
           ast.UnionField("b", None, None, "string")]))])
    expected = [{
      "name": "SomeUnion",
      "fields": [{"kind": "i32", "name": "a"},
                 {"kind": "s", "name": "b"}]}]
    actual = translate.Translate(tree, "mojom_tree")
    self.assertEquals(actual["unions"], expected)

  def testMapTreeForTypeRaisesWithDuplicate(self):
    """Verifies _MapTreeForType() raises when passed two values with the same
       name."""
    methods = [ast.Method('dup', None, None, ast.ParameterList(), None),
               ast.Method('dup', None, None, ast.ParameterList(), None)]
    self.assertRaises(Exception, translate._MapTreeForType,
                      (lambda x: x, methods, '', 'scope'))

  def testAssociatedKinds(self):
    """Tests type spec translation of associated interfaces and requests."""
    # pylint: disable=W0212
    self.assertEquals(translate._MapKind("asso<SomeInterface>?"),
                      "?asso:x:SomeInterface")
    self.assertEquals(translate._MapKind("asso<SomeInterface&>?"),
                      "?asso:r:x:SomeInterface")


if __name__ == "__main__":
  unittest.main()