def test_create_context_fail(self, mock_redis): mock_redis.side_effect = RCE logging.Logger.error = MagicMock() with patch.object(time, 'sleep', return_value=None): with self.assertRaises(SystemExit): OpenvasDB.create_context() logging.Logger.error.assert_called_with( # pylint: disable=no-member 'Redis Error: Not possible to connect to the kb.')
def test_create_context_success(self, mock_redis): ctx = mock_redis.return_value ret = OpenvasDB.create_context() self.assertIs(ret, ctx)