| Index: recipe_engine/unittests/fetch_test.py
|
| diff --git a/recipe_engine/unittests/fetch_test.py b/recipe_engine/unittests/fetch_test.py
|
| index 81bbd2339da490fac2861e4bc6f695f08272e97f..94b307cf8be07c22717752cf5004bce3fb92cb4a 100755
|
| --- a/recipe_engine/unittests/fetch_test.py
|
| +++ b/recipe_engine/unittests/fetch_test.py
|
| @@ -8,11 +8,16 @@ import itertools
|
| import json
|
| import unittest
|
|
|
| +from cStringIO import StringIO
|
| +
|
| import test_env
|
|
|
| +import argparse # this is vendored
|
| +
|
| import mock
|
| import subprocess42
|
|
|
| +from recipe_engine import common_args
|
| from recipe_engine import fetch
|
| from recipe_engine import package_pb2
|
| from recipe_engine import requests_ssl
|
| @@ -451,5 +456,21 @@ class TestGitiles(unittest.TestCase):
|
| self.assertEqual(result, self.a_meta)
|
| self.assertMultiDone(requests_get)
|
|
|
| +
|
| +class TestArgs(unittest.TestCase):
|
| + def setUp(self):
|
| + self.p = argparse.ArgumentParser()
|
| + self.followup = common_args.add_common_args(self.p)
|
| + subp = self.p.add_subparsers()
|
| + fetch.add_subparser(subp)
|
| +
|
| + @mock.patch('argparse._sys.stderr', new_callable=StringIO)
|
| + def test_no_fetch(self, stderr):
|
| + with self.assertRaises(SystemExit):
|
| + args = self.p.parse_args(['--no-fetch', 'fetch'])
|
| + args.postprocess_func(self.p, args)
|
| + self.assertIn('--no-fetch does not make sense', stderr.getvalue())
|
| +
|
| +
|
| if __name__ == '__main__':
|
| unittest.main()
|
|
|