Example #1
0
    def test_eval_args(self):
        with mock.patch('downstream_farmer.shell.check_connectivity') as check:
            check.side_effect = ConnectError('Oops')
            m = mock.Mock()
            with self.assertRaises(SystemExit):
                shell.eval_args(m)
            self.assertTrue(check.called)

        with mock.patch('downstream_farmer.shell.check_connectivity'):
            m.number = -1
            with self.assertRaises(SystemExit):
                shell.eval_args(m)
            with mock.patch('downstream_farmer.shell.run_prototype') as rp:
                m.number = 2
                shell.eval_args(m)
                self.assertTrue(rp.called)
Example #2
0
 def test_eval_args_catchall(self):
     with mock.patch('downstream_farmer.shell.Farmer') as farmer:
         farmer.side_effect = BaseException('error')
         with self.assertRaises(SystemExit):
             shell.eval_args(None)
Example #3
0
 def test_eval_args_downstream_error(self):
     with mock.patch('downstream_farmer.shell.Farmer') as farmer:
         farmer.side_effect = DownstreamError('error')
         with self.assertRaises(SystemExit):
             shell.eval_args(None)
Example #4
0
 def test_eval_args_run(self):
     with mock.patch('downstream_farmer.shell.Farmer') as farmer:
         shell.eval_args(mock.MagicMock())
     self.assertTrue(farmer.called)
     self.assertTrue(farmer.return_value.run.called)
Example #5
0
 def test_eval_args_catchall(self):
     with mock.patch('downstream_farmer.shell.Farmer') as farmer:
         farmer.side_effect = BaseException('error')
         with self.assertRaises(SystemExit):
             shell.eval_args(None)
Example #6
0
 def test_eval_args_downstream_error(self):
     with mock.patch('downstream_farmer.shell.Farmer') as farmer:
         farmer.side_effect = DownstreamError('error')
         with self.assertRaises(SystemExit):
             shell.eval_args(None)
Example #7
0
 def test_eval_args_run(self):
     with mock.patch('downstream_farmer.shell.Farmer') as farmer:
         shell.eval_args(mock.MagicMock())
     self.assertTrue(farmer.called)
     self.assertTrue(farmer.return_value.run.called)