def test_runner(self): ''' Test for execute a runner on the master and return the data from the runner function ''' with patch.dict(raet_publish.__opts__, {'id': 'id'}): with patch.object(salt.transport.Channel, 'factory', MagicMock()): self.assertTrue(raet_publish.runner('fun')) class MockFactory(object): ''' Mock factory class ''' load = '' def send(self, load): ''' mock send method ''' self.load = load raise SaltReqTimeoutError(load) with patch.dict(raet_publish.__opts__, {'id': 'id'}): with patch.object(salt.transport.Channel, 'factory', MagicMock(return_value=MockFactory())): self.assertEqual(raet_publish.runner(1), "'1' runner publish timed out")
def test_runner(self): ''' Test for execute a runner on the master and return the data from the runner function ''' with patch.dict(raet_publish.__opts__, {'id': 'id'}): with patch.object(salt.transport.Channel, 'factory', MagicMock()): self.assertTrue(raet_publish.runner('fun')) class MockFactory(object): ''' Mock factory class ''' load = '' def send(self, load): ''' mock send method ''' self.load = load raise SaltReqTimeoutError(load) with patch.dict(raet_publish.__opts__, {'id': 'id'}): with patch.object(salt.transport.Channel, 'factory', MagicMock(return_value=MockFactory())): self.assertEqual(raet_publish.runner(1), '\'1\' runner publish timed out')