|
2 | 2 | # License, v. 2.0. If a copy of the MPL was not distributed with this |
3 | 3 | # file, You can obtain one at http://mozilla.org/MPL/2.0/. |
4 | 4 |
|
| 5 | +from django.conf import settings |
| 6 | + |
5 | 7 | from nose.tools import eq_ |
6 | | -from bedrock.mozorg.tests import TestCase |
| 8 | + |
7 | 9 | from bedrock.security.management.commands import update_security_advisories |
8 | 10 |
|
9 | 11 |
|
10 | | -class TestUpdateSecurityAdvisories(TestCase): |
11 | | - def test_fix_product_name(self): |
12 | | - """Should fix SeaMonkey and strip '.0' from names.""" |
13 | | - eq_(update_security_advisories.fix_product_name('Seamonkey 2.2'), |
14 | | - 'SeaMonkey 2.2') |
15 | | - eq_(update_security_advisories.fix_product_name('Firefox 2.2'), |
16 | | - 'Firefox 2.2') |
17 | | - eq_(update_security_advisories.fix_product_name('fredflintstone 2.2'), |
18 | | - 'fredflintstone 2.2') |
19 | | - eq_(update_security_advisories.fix_product_name('Firefox 32.0'), |
20 | | - 'Firefox 32') |
21 | | - eq_(update_security_advisories.fix_product_name('Firefox 32.0.1'), |
22 | | - 'Firefox 32.0.1') |
| 12 | +def test_fix_product_name(): |
| 13 | + """Should fix SeaMonkey and strip '.0' from names.""" |
| 14 | + eq_(update_security_advisories.fix_product_name('Seamonkey 2.2'), |
| 15 | + 'SeaMonkey 2.2') |
| 16 | + eq_(update_security_advisories.fix_product_name('Firefox 2.2'), |
| 17 | + 'Firefox 2.2') |
| 18 | + eq_(update_security_advisories.fix_product_name('fredflintstone 2.2'), |
| 19 | + 'fredflintstone 2.2') |
| 20 | + eq_(update_security_advisories.fix_product_name('Firefox 32.0'), |
| 21 | + 'Firefox 32') |
| 22 | + eq_(update_security_advisories.fix_product_name('Firefox 32.0.1'), |
| 23 | + 'Firefox 32.0.1') |
| 24 | + |
| 25 | + |
| 26 | +def test_filter_advisory_names(): |
| 27 | + filenames = [ |
| 28 | + 'README.md', |
| 29 | + 'LICENSE.txt', |
| 30 | + 'announce/2015/mfsa2015-01.md', |
| 31 | + 'stuff/whatnot.md', |
| 32 | + 'mfsa2015-02.md', |
| 33 | + ] |
| 34 | + good_filenames = [ |
| 35 | + settings.MOFO_SECURITY_ADVISORIES_PATH + '/announce/2015/mfsa2015-01.md', |
| 36 | + settings.MOFO_SECURITY_ADVISORIES_PATH + '/mfsa2015-02.md', |
| 37 | + ] |
| 38 | + eq_(update_security_advisories.filter_advisory_filenames(filenames), good_filenames) |
0 commit comments