Die build.py, Die: Part 1 (BranchUtility and SubversionFetcher)

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

Index: chrome/common/extensions/docs/server2/branch_utility_test.py
diff --git a/chrome/common/extensions/docs/server2/branch_utility_test.py b/chrome/common/extensions/docs/server2/branch_utility_test.py
new file mode 100755
index 0000000000000000000000000000000000000000..914ba8c219992dfdc70acf2388ab2b4860ca94bd
--- /dev/null
+++ b/chrome/common/extensions/docs/server2/branch_utility_test.py
@@ -0,0 +1,40 @@
+#!/usr/bin/env python
+# Copyright (c) 2012 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 branch_utility
+import unittest
+
+def ReadFile(filename):
+  with open(filename, 'r') as f:
+    return f.read()
+
+class MockBranchUtility(branch_utility.BranchUtility):
+  def _FetchURL(self, url):
+    return ReadFile('test_data/' + url)
+
+class BranchUtilityTest(unittest.TestCase):
+  def testGetChannelNameFromURL(self):
+    b_util = MockBranchUtility()
+    self.assertEquals('dev', b_util.GetChannelNameFromURL(
+        'dev/hello/stuff.html'))
+    self.assertEquals('beta', b_util.GetChannelNameFromURL(
+        'beta/hello/stuff.html'))
+    self.assertEquals('trunk', b_util.GetChannelNameFromURL(
+        'trunk/hello/stuff.html'))
+    self.assertEquals('stable', b_util.GetChannelNameFromURL(
+        'hello/stuff.html'))
+    self.assertEquals('stable', b_util.GetChannelNameFromURL(
+        'hello/dev/stuff.html'))
+
+  def testGetBranchNumberForURL(self):
+    b_util = MockBranchUtility()
+    b_util.SetURL('branch_number/first.json')
+    self.assertEquals('1132', b_util.GetBranchNumberForURL('dev/stuff.html'))
+    self.assertEquals('1084', b_util.GetBranchNumberForURL('beta/stuff.html'))
+    self.assertEquals('1234', b_util.GetBranchNumberForURL('stuff.html'))
+    self.assertEquals('trunk', b_util.GetBranchNumberForURL('trunk/stuff.html'))
+
+if __name__ == '__main__':
+  unittest.main()