#!/usr/bin/env python3
# Copyright 2012 The Chromium Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

from schema_util import JsFunctionNameToClassName
from schema_util import StripNamespace
import unittest


class SchemaUtilTest(unittest.TestCase):

  def testStripNamespace(self):
    self.assertEqual('Bar', StripNamespace('foo.Bar'))
    self.assertEqual('Baz', StripNamespace('Baz'))

  def testJsFunctionNameToClassName(self):
    self.assertEqual('FooBar', JsFunctionNameToClassName('foo', 'bar'))
    self.assertEqual('FooBar',
                     JsFunctionNameToClassName('experimental.foo', 'bar'))
    self.assertEqual('FooBarBaz', JsFunctionNameToClassName('foo.bar', 'baz'))
    self.assertEqual('FooBarBaz',
                     JsFunctionNameToClassName('experimental.foo.bar', 'baz'))


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