Python mainの例

コード例 #1

 def test_no_cmd_provided(self):
     with self.assertOutputMatches(exit_status='2', stderr='stor: error:.*arguments'):
         with mock.patch.object(sys, 'argv', ['stor']):
             cli.main()

コード例 #2

 def parse_args(self, args):
     with mock.patch.object(sys, 'argv', args.split()):
         cli.main()

コード例 #3

ファイル: test_cli.py プロジェクト: jdavisp3/stor

 def test_no_cmd_provided(self):
     with mock.patch.object(sys, 'argv', ['stor']):
         with self.assertRaisesRegexp(SystemExit, '2'):
             cli.main()
     self.assertIn('stor: error:', sys.stderr.getvalue())
     self.assertIn('arguments', sys.stderr.getvalue())