Docserver: separate models for apps and extensions

chrome/common/extensions/docs/server2/platform_util_test.py

+#!/usr/bin/env python
+# 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 unittest
+
+from platform_util import (GetPlatforms,
+                           GetExtensionTypes,
+                           ExtractPlatformFromURL,
+                           PluralToSingular,
+                           PlatformToExtensionType)
+
+
+class PlatformBundleUtilityTest(unittest.TestCase):
+  def testGetPlatforms(self):
+    self.assertEqual(('apps', 'extensions'), GetPlatforms())
+
+  def testGetExtensionTypes(self):
+    self.assertEqual(('platform_app', 'extension'), GetExtensionTypes())
+
+  def testExtractPlatformFromURL(self):
+    self.assertEqual('apps', ExtractPlatformFromURL('apps/something'))
+    self.assertEqual('apps', ExtractPlatformFromURL('apps'))
+    self.assertEqual('extensions', ExtractPlatformFromURL('extensions/a/b'))
+    self.assertTrue(ExtractPlatformFromURL('a/b') is None)
+    self.assertTrue(ExtractPlatformFromURL('app') is None)
+
+  def testPluralToSingular(self):
+    self.assertEqual('app', PluralToSingular('apps'))
+    self.assertEqual('extension', PluralToSingular('extensions'))
+    self.assertRaises(AssertionError, PluralToSingular, 'ab')
+
+  def testPlatformToExtensionType(self):
+    self.assertEqual('platform_app', PlatformToExtensionType('apps'))
+    self.assertEqual('extension', PlatformToExtensionType('extensions'))
+    self.assertRaises(AssertionError, PlatformToExtensionType, 'ab')
+
+if __name__ == '__main__':
+  unittest.main()