コード例 #1
0
ファイル: test_db.py プロジェクト: befoulad/ospd-openvas
    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.')
コード例 #2
0
ファイル: test_db.py プロジェクト: befoulad/ospd-openvas
 def test_create_context_success(self, mock_redis):
     ctx = mock_redis.return_value
     ret = OpenvasDB.create_context()
     self.assertIs(ret, ctx)