Move fail tests into a separate class.

tests/patch_test.py

 #!/usr/bin/env python
 # Copyright (c) 2011 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.
 
 """Unit tests for patch.py."""
 
 import logging
 import os
 import sys
@@ skipping 152 matching lines @@
     if hasattr(p, 'patchlevel'):
       self.assertEquals(p.patchlevel, patchlevel)
     if diff:
       self.assertEquals(p.get(), diff)
     if hasattr(p, 'svn_properties'):
       self.assertEquals(p.svn_properties, svn_properties)
 
   def testFilePatchDelete(self):
     p = patch.FilePatchDelete('foo', False)
     self._check_patch(p, 'foo', None, is_delete=True)
-    try:
-      p.get()
-      self.fail()
-    except NotImplementedError:
-      pass
 
   def testFilePatchDeleteBin(self):
     p = patch.FilePatchDelete('foo', True)
     self._check_patch(p, 'foo', None, is_delete=True, is_binary=True)
-    try:
-      p.get()
-      self.fail()
-    except NotImplementedError:
-      pass
 
   def testFilePatchBinary(self):
     p = patch.FilePatchBinary('foo', 'data', [], is_new=False)
     self._check_patch(p, 'foo', 'data', is_binary=True)
 
   def testFilePatchBinaryNew(self):
     p = patch.FilePatchBinary('foo', 'data', [], is_new=True)
     self._check_patch(p, 'foo', 'data', is_binary=True, is_new=True)
 
   def testFilePatchDiff(self):
@@ skipping 25 matching lines @@
     # The code path is different for git and svn.
     p = patch.FilePatchDiff('foo', NEW, [])
     self._check_patch(p, 'foo', NEW, is_new=True)
 
   def testFilePatchDiffGitNew(self):
     # The code path is different for git and svn.
     p = patch.FilePatchDiff('foo', GIT_NEW, [])
     self._check_patch(
         p, 'foo', GIT_NEW, is_new=True, is_git_diff=True, patchlevel=1)
 
-  def testFilePatchDiffBad(self):
-    try:
-      patch.FilePatchDiff('foo', 'data', [])
-      self.fail()
-    except patch.UnsupportedPatchFormat:
-      pass
-
-  def testFilePatchDiffEmpty(self):
-    try:
-      patch.FilePatchDiff('foo', '', [])
-      self.fail()
-    except patch.UnsupportedPatchFormat:
-      pass
-
-  def testFilePatchDiffNone(self):
-    try:
-      patch.FilePatchDiff('foo', None, [])
-      self.fail()
-    except patch.UnsupportedPatchFormat:
-      pass
-
-  def testFilePatchBadDiffName(self):
-    try:
-      patch.FilePatchDiff('foo', SVN_PATCH, [])
-      self.fail()
-    except patch.UnsupportedPatchFormat, e:
-      self.assertEquals(
-          "Can't process patch for file foo.\nUnexpected diff: chrome/file.cc.",
-          str(e))
-
-  def testFilePatchDiffBadHeader(self):
-    try:
-      diff = (
-        '+++ b/foo\n'
-        '@@ -0,0 +1 @@\n'
-        '+bar\n')
-      patch.FilePatchDiff('foo', diff, [])
-      self.fail()
-    except patch.UnsupportedPatchFormat:
-      pass
-
-  def testFilePatchDiffBadGitHeader(self):
-    try:
-      diff = (
-        'diff --git a/foo b/foo\n'
-        '+++ b/foo\n'
-        '@@ -0,0 +1 @@\n'
-        '+bar\n')
-      patch.FilePatchDiff('foo', diff, [])
-      self.fail()
-    except patch.UnsupportedPatchFormat:
-      pass
-
-  def testFilePatchDiffBadHeaderReversed(self):
-    try:
-      diff = (
-        '+++ b/foo\n'
-        '--- b/foo\n'
-        '@@ -0,0 +1 @@\n'
-        '+bar\n')
-      patch.FilePatchDiff('foo', diff, [])
-      self.fail()
-    except patch.UnsupportedPatchFormat:
-      pass
-
-  def testFilePatchDiffGitBadHeaderReversed(self):
-    try:
-      diff = (
-        'diff --git a/foo b/foo\n'
-        '+++ b/foo\n'
-        '--- b/foo\n'
-        '@@ -0,0 +1 @@\n'
-        '+bar\n')
-      patch.FilePatchDiff('foo', diff, [])
-      self.fail()
-    except patch.UnsupportedPatchFormat:
-      pass
-
-  def testFilePatchDiffInvalidGit(self):
-    try:
-      patch.FilePatchDiff('svn_utils_test.txt', (
-        'diff --git a/tests/svn_utils_test_data/svn_utils_test.txt '
-        'b/tests/svn_utils_test_data/svn_utils_test.txt\n'
-        'index 0e4de76..8320059 100644\n'
-        '--- a/svn_utils_test.txt\n'
-        '+++ b/svn_utils_test.txt\n'
-        '@@ -3,6 +3,7 @@ bb\n'
-        'ccc\n'
-        'dd\n'
-        'e\n'
-        '+FOO!\n'
-        'ff\n'
-        'ggg\n'
-        'hh\n'),
-        [])
-      self.fail()
-    except patch.UnsupportedPatchFormat:
-      pass
-    try:
-      patch.FilePatchDiff('svn_utils_test2.txt', (
-        'diff --git a/svn_utils_test_data/svn_utils_test.txt '
-        'b/svn_utils_test.txt\n'
-        'index 0e4de76..8320059 100644\n'
-        '--- a/svn_utils_test.txt\n'
-        '+++ b/svn_utils_test.txt\n'
-        '@@ -3,6 +3,7 @@ bb\n'
-        'ccc\n'
-        'dd\n'
-        'e\n'
-        '+FOO!\n'
-        'ff\n'
-        'ggg\n'
-        'hh\n'),
-        [])
-      self.fail()
-    except patch.UnsupportedPatchFormat:
-      pass
-
   def testValidSvn(self):
     # pylint: disable=R0201
     # Method could be a function
     # Should not throw.
     p = patch.FilePatchDiff('chrome/file.cc', SVN_PATCH, [])
     lines = SVN_PATCH.splitlines(True)
     header = ''.join(lines[:4])
     hunks = ''.join(lines[4:])
     self.assertEquals(header, p.diff_header)
     self.assertEquals(hunks, p.diff_hunks)
@@ skipping 45 matching lines @@
       if line.startswith('@@'):
         break
       if line[:3] == '---':
         line = line.replace(orig_source_name, new_source_name)
       if line[:3] == '+++':
         line = line.replace(orig_name, new_name)
       header.append(line)
     header = ''.join(header)
     self.assertEquals(header, patches.patches[0].diff_header)
 
-  def testRelPathBad(self):
-    patches = patch.PatchSet([
-        patch.FilePatchDiff('chrome\\file.cc', SVN_PATCH, []),
-        patch.FilePatchDelete('other\\place\\foo', True),
-    ])
-    try:
-      patches.set_relpath('..')
-      self.fail()
-    except patch.UnsupportedPatchFormat:
-      pass
-
   def testRelPathEmpty(self):
     patches = patch.PatchSet([
         patch.FilePatchDiff('chrome\\file.cc', SVN_PATCH, []),
         patch.FilePatchDelete('other\\place\\foo', True),
     ])
     patches.set_relpath('')
     self.assertEquals(
         ['chrome/file.cc', 'other/place/foo'],
         [f.filename for f in patches])
     self.assertEquals([None, None], [f.source_filename for f in patches])
@@ skipping 41 matching lines @@
   def testOnlyHeader(self):
     diff = '--- file_a\n+++ file_a\n'
     p = patch.FilePatchDiff('file_a', diff, [])
     self._check_patch(p, 'file_a', diff)
 
   def testSmallest(self):
     diff = '--- file_a\n+++ file_a\n@@ -0,0 +1 @@\n+foo\n'
     p = patch.FilePatchDiff('file_a', diff, [])
     self._check_patch(p, 'file_a', diff)
 
-  def testInverted(self):
-    try:
-      patch.FilePatchDiff(
-        'file_a', '+++ file_a\n--- file_a\n@@ -0,0 +1 @@\n+foo\n', [])
-      self.fail()
-    except patch.UnsupportedPatchFormat:
-      pass
-
-  def testInvertedOnlyHeader(self):
-    try:
-      patch.FilePatchDiff('file_a', '+++ file_a\n--- file_a\n', [])
-      self.fail()
-    except patch.UnsupportedPatchFormat:
-      pass
-
   def testRenameOnlyHeader(self):
     diff = '--- file_a\n+++ file_b\n'
     p = patch.FilePatchDiff('file_b', diff, [])
     # Should it be marked as new?
     self._check_patch(p, 'file_b', diff, source_filename='file_a', is_new=True)
 
   def testGitCopyPartial(self):
     diff = (
         'diff --git a/wtf b/wtf2\n'
         'similarity index 98%\n'
@@ skipping 33 matching lines @@
         'new file mode 100644\n'
         '--- /dev/null\n'
         '+++ b/natsort_test.py\n'
         '@@ -0,0 +1,1 @@\n'
         '+#!/usr/bin/env python\n')
     p = patch.FilePatchDiff('natsort_test.py', diff, [])
     self._check_patch(
         p, 'natsort_test.py', diff, is_new=True, is_git_diff=True, patchlevel=1)
 
 
+class PatchTestFail(unittest.TestCase):
+  # All patches that should throw.
+  def testFilePatchDelete(self):
+    p = patch.FilePatchDelete('foo', False)
+    try:
+      p.get()
+      self.fail()
+    except NotImplementedError:
+      pass
+
+  def testFilePatchDeleteBin(self):
+    p = patch.FilePatchDelete('foo', True)
+    try:
+      p.get()
+      self.fail()
+    except NotImplementedError:
+      pass
+
+  def testFilePatchDiffBad(self):
+    try:
+      patch.FilePatchDiff('foo', 'data', [])
+      self.fail()
+    except patch.UnsupportedPatchFormat:
+      pass
+
+  def testFilePatchDiffEmpty(self):
+    try:
+      patch.FilePatchDiff('foo', '', [])
+      self.fail()
+    except patch.UnsupportedPatchFormat:
+      pass
+
+  def testFilePatchDiffNone(self):
+    try:
+      patch.FilePatchDiff('foo', None, [])
+      self.fail()
+    except patch.UnsupportedPatchFormat:
+      pass
+
+  def testFilePatchBadDiffName(self):
+    try:
+      patch.FilePatchDiff('foo', SVN_PATCH, [])
+      self.fail()
+    except patch.UnsupportedPatchFormat, e:
+      self.assertEquals(
+          "Can't process patch for file foo.\nUnexpected diff: chrome/file.cc.",
+          str(e))
+
+  def testFilePatchDiffBadHeader(self):
+    try:
+      diff = (
+        '+++ b/foo\n'
+        '@@ -0,0 +1 @@\n'
+        '+bar\n')
+      patch.FilePatchDiff('foo', diff, [])
+      self.fail()
+    except patch.UnsupportedPatchFormat:
+      pass
+
+  def testFilePatchDiffBadGitHeader(self):
+    try:
+      diff = (
+        'diff --git a/foo b/foo\n'
+        '+++ b/foo\n'
+        '@@ -0,0 +1 @@\n'
+        '+bar\n')
+      patch.FilePatchDiff('foo', diff, [])
+      self.fail()
+    except patch.UnsupportedPatchFormat:
+      pass
+
+  def testFilePatchDiffBadHeaderReversed(self):
+    try:
+      diff = (
+        '+++ b/foo\n'
+        '--- b/foo\n'
+        '@@ -0,0 +1 @@\n'
+        '+bar\n')
+      patch.FilePatchDiff('foo', diff, [])
+      self.fail()
+    except patch.UnsupportedPatchFormat:
+      pass
+
+  def testFilePatchDiffGitBadHeaderReversed(self):
+    try:
+      diff = (
+        'diff --git a/foo b/foo\n'
+        '+++ b/foo\n'
+        '--- b/foo\n'
+        '@@ -0,0 +1 @@\n'
+        '+bar\n')
+      patch.FilePatchDiff('foo', diff, [])
+      self.fail()
+    except patch.UnsupportedPatchFormat:
+      pass
+
+  def testFilePatchDiffInvalidGit(self):
+    try:
+      patch.FilePatchDiff('svn_utils_test.txt', (
+        'diff --git a/tests/svn_utils_test_data/svn_utils_test.txt '
+        'b/tests/svn_utils_test_data/svn_utils_test.txt\n'
+        'index 0e4de76..8320059 100644\n'
+        '--- a/svn_utils_test.txt\n'
+        '+++ b/svn_utils_test.txt\n'
+        '@@ -3,6 +3,7 @@ bb\n'
+        'ccc\n'
+        'dd\n'
+        'e\n'
+        '+FOO!\n'
+        'ff\n'
+        'ggg\n'
+        'hh\n'),
+        [])
+      self.fail()
+    except patch.UnsupportedPatchFormat:
+      pass
+    try:
+      patch.FilePatchDiff('svn_utils_test2.txt', (
+        'diff --git a/svn_utils_test_data/svn_utils_test.txt '
+        'b/svn_utils_test.txt\n'
+        'index 0e4de76..8320059 100644\n'
+        '--- a/svn_utils_test.txt\n'
+        '+++ b/svn_utils_test.txt\n'
+        '@@ -3,6 +3,7 @@ bb\n'
+        'ccc\n'
+        'dd\n'
+        'e\n'
+        '+FOO!\n'
+        'ff\n'
+        'ggg\n'
+        'hh\n'),
+        [])
+      self.fail()
+    except patch.UnsupportedPatchFormat:
+      pass
+
+  def testRelPathBad(self):
+    patches = patch.PatchSet([
+        patch.FilePatchDiff('chrome\\file.cc', SVN_PATCH, []),
+        patch.FilePatchDelete('other\\place\\foo', True),
+    ])
+    try:
+      patches.set_relpath('..')
+      self.fail()
+    except patch.UnsupportedPatchFormat:
+      pass
+
+  def testInverted(self):
+    try:
+      patch.FilePatchDiff(
+        'file_a', '+++ file_a\n--- file_a\n@@ -0,0 +1 @@\n+foo\n', [])
+      self.fail()
+    except patch.UnsupportedPatchFormat:
+      pass
+
+  def testInvertedOnlyHeader(self):
+    try:
+      patch.FilePatchDiff('file_a', '+++ file_a\n--- file_a\n', [])
+      self.fail()
+    except patch.UnsupportedPatchFormat:
+      pass
+
+
 if __name__ == '__main__':
   logging.basicConfig(level=
       [logging.WARNING, logging.INFO, logging.DEBUG][
         min(2, sys.argv.count('-v'))])
   unittest.main()