#!/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()