def test_OssecQueue_protected_send_ko(mock_send, mock_conn): """Tests OssecQueue._send function exceptions works""" queue = OssecQueue('test_path') with pytest.raises(WazuhException, match=".* 1011 .*"): queue._send('msg') mock_conn.assert_called_with('test_path')
def test_OssecQueue_protected_send(mock_conn, send_response, error): """Tests OssecQueue._send function works""" queue = OssecQueue('test_path') with patch('socket.socket.send', return_value=send_response): if error: with pytest.raises(WazuhException, match=".* 1011 .*"): queue._send('msg') else: queue._send('msg') mock_conn.assert_called_with('test_path')