OLD | NEW |
---|---|
(Empty) | |
1 #!/usr/bin/env python | |
2 # coding: utf-8 | |
3 # Copyright 2016 The LUCI Authors. All rights reserved. | |
4 # Use of this source code is governed under the Apache License, Version 2.0 | |
5 # that can be found in the LICENSE file. | |
6 | |
7 import collections | |
8 import logging | |
9 import re | |
10 import sys | |
11 import unittest | |
12 | |
13 # Setups environment. | |
14 import test_env_handlers | |
15 | |
16 import cipd | |
17 import swarming_rpcs | |
18 | |
19 | |
20 class TestPinChecker(unittest.TestCase): | |
21 def setUp(self): | |
22 super(TestPinChecker, self).setUp() | |
23 self.pin_checker = cipd.PinChecker() | |
24 self.cp = collections.namedtuple('CipdPackage', 'path package_name version') | |
M-A Ruel
2016/08/30 02:18:07
I didn't know you like javascript. :P
iannucci
2016/08/30 09:13:29
:(
| |
25 | |
26 def test_correct_pins(self): | |
27 a = self.cp('path', 'package_name/${platform}-${os_ver}', 'latest') | |
28 b = self.cp('path', 'package_name/windows-amd64-something_10', 'deadbeef'*5) | |
29 | |
30 # will not raise | |
31 self.pin_checker.check(a, b) | |
32 | |
33 self.assertEqual(self.pin_checker.platform, re.escape('windows-amd64')) | |
34 self.assertEqual(self.pin_checker.os_ver, re.escape('something_10')) | |
35 | |
36 a = self.cp('path', 'other/${platform}-${os_ver}', 'latest') | |
37 b = self.cp('path', 'other/windows-amd64-something_10', 'deadbeef'*5) | |
38 | |
39 # will not raise | |
40 self.pin_checker.check(a, b) | |
41 | |
42 def test_mismatched_pins(self): | |
43 # if a is already a pin, b must match its version exactly | |
44 a = self.cp('path', 'package_name/${platform}-${os_ver}', 'deadbeef'*5) | |
45 b = self.cp('path', 'package_name/windows-amd64-something_10', 'badc0ffe'*5) | |
46 | |
47 with self.assertRaisesRegexp(ValueError, 'Mismatched pins'): | |
48 self.pin_checker.check(a, b) | |
49 | |
50 def test_mismatched_paths(self): | |
51 a = self.cp('path', 'package_name/${platform}-${os_ver}', 'latest') | |
52 b = self.cp('else', 'package_name/windows-amd64-something_10', 'deadbeef'*5) | |
53 | |
54 with self.assertRaisesRegexp(ValueError, 'Mismatched path'): | |
55 self.pin_checker.check(a, b) | |
56 | |
57 def test_mismatched_names(self): | |
58 a = self.cp('', 'package_name/${platform}-${os_ver}', 'latest') | |
59 b = self.cp('', 'else/windows-amd64-something_10', 'deadbeef'*5) | |
60 | |
61 with self.assertRaisesRegexp(ValueError, 'Mismatched package_name'): | |
62 self.pin_checker.check(a, b) | |
63 | |
64 self.assertIsNone(self.pin_checker.platform) | |
65 self.assertIsNone(self.pin_checker.os_ver) | |
66 | |
67 a = self.cp('', 'package_name/${platform}-${os_ver}', 'latest') | |
68 b = self.cp('', 'package_name/windows-amd64-something_10', 'deadbeef'*5) | |
69 # will not raise | |
70 self.pin_checker.check(a, b) | |
71 | |
72 self.assertEqual(self.pin_checker.platform, re.escape('windows-amd64')) | |
73 self.assertEqual(self.pin_checker.os_ver, re.escape('something_10')) | |
74 | |
75 # This doesn't match the previous knowledge of platform or os_ver, so it | |
76 # will not match. | |
77 a = self.cp('', 'package_name/${platform}-${os_ver}', 'latest') | |
78 b = self.cp('', 'package_name/linux-32-nerds', 'deadbeef'*5) | |
79 | |
80 with self.assertRaisesRegexp(ValueError, 'Mismatched package_name'): | |
81 self.pin_checker.check(a, b) | |
82 | |
83 | |
84 if __name__ == '__main__': | |
85 if '-v' in sys.argv: | |
86 unittest.TestCase.maxDiff = None | |
87 logging.basicConfig( | |
88 level=logging.DEBUG if '-v' in sys.argv else logging.CRITICAL, | |
89 format='%(levelname)-7s %(filename)s:%(lineno)3d %(message)s') | |
90 unittest.main() | |
OLD | NEW |