|
5 | 5 | from django.utils.encoding import smart_str |
6 | 6 | from django.test import TestCase |
7 | 7 | from concurrency.core import InconsistencyError |
| 8 | +from concurrency.exceptions import VersionError |
8 | 9 | from concurrency.forms import ConcurrentForm, VersionField, VersionFieldSigner, VersionWidget |
9 | 10 | from concurrency.tests import TestModel0, TestIssue3Model |
10 | 11 | from django.test.testcases import SimpleTestCase |
@@ -44,23 +45,17 @@ def test_with_dummy_signer(self): |
44 | 45 | self.assertEqual(1, f.clean('1')) |
45 | 46 | self.assertEqual(0, f.clean(None)) |
46 | 47 | self.assertEqual(0, f.clean('')) |
47 | | - self.assertRaisesMessage(SuspiciousOperation, |
48 | | - "Version number seems tampered", f.clean, 'aa:bb') |
49 | | - self.assertRaisesMessage(SuspiciousOperation, |
50 | | - "Version number seems tampered", f.clean, 1.5) |
| 48 | + self.assertRaises(VersionError, f.clean, 'aa:bb') |
| 49 | + self.assertRaises(VersionError, f.clean, 1.5) |
51 | 50 |
|
52 | 51 | def test(self): |
53 | 52 | f = VersionField() |
54 | 53 | self.assertEqual(1, f.clean(VersionFieldSigner().sign(1))) |
55 | 54 | self.assertEqual(1, f.clean(VersionFieldSigner().sign('1'))) |
56 | 55 | self.assertEqual(0, f.clean(None)) |
57 | 56 | self.assertEqual(0, f.clean('')) |
58 | | - self.assertRaisesMessage(SuspiciousOperation, |
59 | | - "Version number seems tampered", f.clean, '100') |
60 | | - self.assertRaisesMessage(SuspiciousOperation, |
61 | | - "Version number seems tampered", |
62 | | - f.clean, |
63 | | - VersionFieldSigner().sign(1.5)) |
| 57 | + self.assertRaises(VersionError, f.clean, '100') |
| 58 | + self.assertRaises(VersionError, f.clean, VersionFieldSigner().sign(1.5)) |
64 | 59 |
|
65 | 60 |
|
66 | 61 | class ConcurrentFormTest(TestCase): |
|
0 commit comments