/*
    This file is part of the KDE libraries
    SPDX-FileCopyrightText: 2010 Bernhard Beschow <bbeschow@cs.tu-berlin.de>

    SPDX-License-Identifier: LGPL-2.0-or-later
*/

#include "searchbar_test.h"

#include "ui_searchbarincremental.h"
#include "ui_searchbarpower.h"

#include <KMessageBox>
#include <kateconfig.h>
#include <katedocument.h>
#include <kateglobal.h>
#include <katesearchbar.h>
#include <kateview.h>
#include <ktexteditor/movingrange.h>

#include <QStringListModel>
#include <QTest>

QTEST_MAIN(SearchBarTest)

#define testNewRow() (QTest::newRow(QStringLiteral("line %1").arg(__LINE__).toLatin1().data()))

using namespace KTextEditor;

SearchBarTest::SearchBarTest()
    : QObject()
{
}

SearchBarTest::~SearchBarTest()
{
}

void SearchBarTest::initTestCase()
{
    QStandardPaths::setTestModeEnabled(true);
    KMessageBox::saveDontShowAgainTwoActions(QLatin1String("DoNotShowAgainContinueSearchDialog"), KMessageBox::PrimaryAction);
}

void SearchBarTest::cleanupTestCase()
{
}

void SearchBarTest::testFindNextIncremental()
{
    KTextEditor::DocumentPrivate doc;
    doc.setText(QStringLiteral("a a a b b"));

    KTextEditor::ViewPrivate view(&doc, nullptr);
    KateViewConfig config(&view);

    KateSearchBar bar(false, &view, &config);

    bar.setSearchPattern(QStringLiteral("b"));

    QCOMPARE(view.selectionRange(), Range(0, 6, 0, 7));

    bar.findNext();

    QCOMPARE(view.selectionRange(), Range(0, 8, 0, 9));

    bar.setSearchPattern(QStringLiteral("a"));

    QCOMPARE(view.selectionRange(), Range(0, 0, 0, 1));

    bar.findNext();

    QCOMPARE(view.selectionRange(), Range(0, 2, 0, 3));

    bar.findNext();

    QCOMPARE(view.selectionRange(), Range(0, 4, 0, 5));

    bar.findNext();

    QCOMPARE(view.selectionRange(), Range(0, 0, 0, 1));
}

void SearchBarTest::testFindNextZeroLengthMatch()
{
    // Some regularexpression matches are zero length assertions
    // e.g. '$', '^', '\b', test that the cursor won't be stuck
    // on one match when using FindNext
    KTextEditor::DocumentPrivate doc;
    doc.setText(QStringLiteral("a\nb \nc\n\n"));

    KTextEditor::ViewPrivate view(&doc, nullptr);
    KateViewConfig config(&view);
    KateSearchBar bar(true, &view, &config);
    bar.setSearchMode(KateSearchBar::MODE_REGEX);

    bar.setSearchPattern(QStringLiteral("$"));

    QVERIFY(bar.isPower());

    bar.findNext();
    QCOMPARE(view.cursorPosition(), Cursor(0, 1));

    bar.findNext();
    QCOMPARE(view.cursorPosition(), Cursor(1, 2));

    bar.findNext();
    QCOMPARE(view.cursorPosition(), Cursor(2, 1));

    bar.findNext();
    QCOMPARE(view.cursorPosition(), Cursor(3, 0));

    // Test Unicode
    doc.setText(QStringLiteral("aéöz\n"));
    bar.setSearchPattern(QStringLiteral("\\w"));
    view.setCursorPosition(Cursor(0, 0));

    bar.findNext();
    QCOMPARE(view.cursorPosition(), Cursor(0, 1));

    bar.findNext();
    QCOMPARE(view.cursorPosition(), Cursor(0, 2));

    bar.findNext();
    QCOMPARE(view.cursorPosition(), Cursor(0, 3));

    bar.findNext();
    QCOMPARE(view.cursorPosition(), Cursor(0, 4));

    doc.setText(QStringLiteral("aé ö z\n"));
    bar.setSearchPattern(QStringLiteral("\\b"));
    view.setCursorPosition(Cursor(0, 0));

    bar.findNext();
    QCOMPARE(view.cursorPosition(), Cursor(0, 2));
    QCOMPARE(doc.text(Range(0, 1, 0, 2)), QStringLiteral("é"));

    bar.findNext();
    QCOMPARE(view.cursorPosition(), Cursor(0, 3));
    QCOMPARE(doc.text(Range(0, 3, 0, 4)), QStringLiteral("ö"));

    bar.findNext();
    QCOMPARE(view.cursorPosition(), Cursor(0, 4));
    QCOMPARE(doc.text(Range(0, 3, 0, 4)), QStringLiteral("ö"));

    bar.findNext();
    QCOMPARE(view.cursorPosition(), Cursor(0, 5));
    QCOMPARE(doc.text(Range(0, 5, 0, 6)), QStringLiteral("z"));

    bar.findNext();
    QCOMPARE(view.cursorPosition(), Cursor(0, 6));
    QCOMPARE(doc.text(Range(0, 5, 0, 6)), QStringLiteral("z"));

    bar.findNext();
    // Search wraps, back to before first char
    QCOMPARE(view.cursorPosition(), Cursor(0, 0));
    QCOMPARE(doc.text(Range(0, 0, 0, 1)), QStringLiteral("a"));
}

void SearchBarTest::testFindNextNoNewLineAtEnd()
{
    // Testing with a document that doesn't end with a new line
    KTextEditor::DocumentPrivate doc;
    doc.setText(QStringLiteral(" \n \n "));

    KTextEditor::ViewPrivate view(&doc, nullptr);
    KateViewConfig config(&view);
    KateSearchBar bar(true, &view, &config);
    QVERIFY(bar.isPower());
    bar.setSearchMode(KateSearchBar::MODE_REGEX);
    bar.setSearchPattern(QStringLiteral("^ *\\n"));

    bar.findNext();
    QCOMPARE(view.selectionRange(), Range(0, 0, 1, 0));

    bar.findNext();
    QCOMPARE(view.selectionRange(), Range(1, 0, 2, 0));

    bar.findNext();
    // Search wraps
    QCOMPARE(view.selectionRange(), Range(0, 0, 1, 0));
}

void SearchBarTest::testSetMatchCaseIncremental()
{
    KTextEditor::DocumentPrivate doc;
    KTextEditor::ViewPrivate view(&doc, nullptr);
    KateViewConfig config(&view);

    doc.setText(QStringLiteral("a A a"));
    KateSearchBar bar(false, &view, &config);

    QVERIFY(!bar.isPower());
    QVERIFY(!view.selection());

    bar.setMatchCase(false);
    bar.setSearchPattern(QStringLiteral("A"));

    QVERIFY(!bar.matchCase());
    QCOMPARE(view.selectionRange(), Range(0, 0, 0, 1));

    bar.setMatchCase(true);

    QVERIFY(bar.matchCase());
    QCOMPARE(view.selectionRange(), Range(0, 2, 0, 3));

    bar.setMatchCase(false);

    QVERIFY(!bar.matchCase());
    QCOMPARE(view.selectionRange(), Range(0, 0, 0, 1));

    bar.setMatchCase(true);

    QVERIFY(bar.matchCase());
    QCOMPARE(view.selectionRange(), Range(0, 2, 0, 3));
}

void SearchBarTest::testSetMatchCasePower()
{
    KTextEditor::DocumentPrivate doc;
    KTextEditor::ViewPrivate view(&doc, nullptr);
    KateViewConfig config(&view);

    doc.setText(QStringLiteral("a A a"));
    view.setCursorPosition(Cursor(0, 0));

    KateSearchBar bar(true, &view, &config);

    QVERIFY(bar.isPower());
    QVERIFY(!view.selection());

    bar.setMatchCase(false);
    bar.setSearchPattern(QStringLiteral("A"));
    bar.findNext();

    QCOMPARE(bar.searchPattern(), QStringLiteral("A"));
    QVERIFY(!bar.matchCase());
    QCOMPARE(view.selectionRange(), Range(0, 0, 0, 1));

    bar.setMatchCase(true);

    QCOMPARE(view.selectionRange(), Range(0, 0, 0, 1));

    bar.findNext();

    QVERIFY(bar.matchCase());
    QCOMPARE(view.selectionRange(), Range(0, 2, 0, 3));

    bar.setMatchCase(false);

    QVERIFY(!bar.matchCase());
    QCOMPARE(view.selectionRange(), Range(0, 2, 0, 3));

    bar.findNext();

    QCOMPARE(view.selectionRange(), Range(0, 4, 0, 5));
}

void SearchBarTest::testSetSelectionOnlyPower()
{
    KTextEditor::DocumentPrivate doc;
    KTextEditor::ViewPrivate view(&doc, nullptr);
    KateViewConfig config(&view);

    doc.setText(QStringLiteral("a a a a"));
    KateSearchBar bar(true, &view, &config);

    bar.setSearchPattern(QStringLiteral("a"));

    QVERIFY(bar.isPower());
    QVERIFY(!view.selection());

    bar.setSelectionOnly(false);
    bar.findNext();

    QVERIFY(!bar.selectionOnly());
    QCOMPARE(view.selectionRange(), Range(0, 0, 0, 1));

    view.setSelection(Range(0, 2, 0, 5));
    bar.setSelectionOnly(true);

    QVERIFY(bar.selectionOnly());

    bar.findNext();

    QCOMPARE(view.selectionRange(), Range(0, 2, 0, 3));
    QVERIFY(bar.selectionOnly());

    bar.findNext();

    QCOMPARE(view.selectionRange(), Range(0, 4, 0, 5));
    QVERIFY(bar.selectionOnly());

    // Test Search wrap for selection only
    bar.findNext();

    QCOMPARE(view.selectionRange(), Range(0, 2, 0, 3));
    QVERIFY(bar.selectionOnly());

    bar.findPrevious();

    QCOMPARE(view.selectionRange(), Range(0, 4, 0, 5));
    QVERIFY(bar.selectionOnly());

    bar.setSelectionOnly(false);
    bar.findNext();

    QCOMPARE(view.selectionRange(), Range(0, 6, 0, 7));
    QVERIFY(!bar.selectionOnly());
}

void SearchBarTest::testSetSearchPattern_data()
{
    QTest::addColumn<bool>("power");
    QTest::addColumn<int>("numMatches2");

    testNewRow() << false << 0;
    testNewRow() << true << 3;
}

void SearchBarTest::testSetSearchPattern()
{
    QFETCH(bool, power);
    QFETCH(int, numMatches2);

    KTextEditor::DocumentPrivate doc;
    KTextEditor::ViewPrivate view(&doc, nullptr);
    KateViewConfig config(&view);

    doc.setText(QStringLiteral("a a a"));

    KateSearchBar bar(power, &view, &config);

    bar.setSearchPattern(QStringLiteral("a"));
    bar.findAll();

    QCOMPARE(bar.m_hlRanges.size(), 3);

    bar.setSearchPattern(QStringLiteral("a "));

    QCOMPARE(bar.m_hlRanges.size(), numMatches2);

    bar.findAll();

    QCOMPARE(bar.m_hlRanges.size(), 2);
}

void SearchBarTest::testSetSelectionOnly()
{
    KTextEditor::DocumentPrivate doc;
    KTextEditor::ViewPrivate view(&doc, nullptr);
    KateViewConfig config(&view);

    doc.setText(QStringLiteral("a a a"));
    view.setSelection(Range(0, 0, 0, 3));

    KateSearchBar bar(false, &view, &config);

    bar.setSelectionOnly(false);
    bar.setSearchPattern(QStringLiteral("a"));
    bar.findAll();

    QCOMPARE(bar.m_hlRanges.size(), 3);

    bar.setSelectionOnly(true);

    QCOMPARE(bar.m_hlRanges.size(), 3);
}

void SearchBarTest::testFindAll_data()
{
    QTest::addColumn<bool>("power");
    QTest::addColumn<int>("numMatches2");
    QTest::addColumn<int>("numMatches4");

    testNewRow() << false << 0 << 0;
    testNewRow() << true << 3 << 2;
}

void SearchBarTest::testFindAll()
{
    QFETCH(bool, power);
    QFETCH(int, numMatches2);
    QFETCH(int, numMatches4);

    KTextEditor::DocumentPrivate doc;
    KTextEditor::ViewPrivate view(&doc, nullptr);
    KateViewConfig config(&view);

    doc.setText(QStringLiteral("a a a"));
    KateSearchBar bar(power, &view, &config);

    QCOMPARE(bar.isPower(), power);

    bar.setSearchPattern(QStringLiteral("a"));
    bar.findAll();

    QCOMPARE(bar.m_hlRanges.size(), 3);
    QCOMPARE(bar.m_hlRanges.at(0)->toRange(), Range(0, 0, 0, 1));
    QCOMPARE(bar.m_hlRanges.at(1)->toRange(), Range(0, 2, 0, 3));
    QCOMPARE(bar.m_hlRanges.at(2)->toRange(), Range(0, 4, 0, 5));

    bar.setSearchPattern(QStringLiteral("a "));

    QCOMPARE(bar.m_hlRanges.size(), numMatches2);

    bar.findAll();

    QCOMPARE(bar.m_hlRanges.size(), 2);

    bar.setSearchPattern(QStringLiteral("a  "));

    QCOMPARE(bar.m_hlRanges.size(), numMatches4);

    bar.findAll();

    QCOMPARE(bar.m_hlRanges.size(), 0);
}

void SearchBarTest::testReplaceInSelectionOnly()
{
    KTextEditor::DocumentPrivate doc;
    KTextEditor::ViewPrivate view(&doc, nullptr);
    KateViewConfig config(&view);

    doc.setText(QStringLiteral("a\na\na\na\na"));
    KateSearchBar bar(true, &view, &config);

    bar.setSearchPattern(QStringLiteral("a\n"));

    view.setSelection(Range(1, 0, 4, 0));
    bar.setSelectionOnly(true);

    QVERIFY(bar.selectionOnly());

    bar.replaceNext();

    QCOMPARE(view.selectionRange(), Range(1, 0, 2, 0));
    QVERIFY(bar.selectionOnly());

    bar.replaceNext();

    QCOMPARE(view.selectionRange(), Range(1, 0, 2, 0));
    QVERIFY(bar.selectionOnly());

    bar.replaceNext();

    QCOMPARE(view.selectionRange(), Range(1, 0, 2, 0));
    QVERIFY(bar.selectionOnly());

    bar.replaceNext();

    QCOMPARE(view.selectionRange(), Range(1, 0, 1, 0));
    QCOMPARE(doc.text(), QStringLiteral("a\na"));
    QVERIFY(bar.selectionOnly());

    // Test undo (search selection range should still be preserved)
    doc.undo();
    doc.undo();
    doc.undo();

    QCOMPARE(view.selectionRange(), Range(1, 0, 2, 0));
    QVERIFY(bar.selectionOnly());

    bar.findPrevious();

    QCOMPARE(view.selectionRange(), Range(3, 0, 4, 0));
    QVERIFY(bar.selectionOnly());

    // TODO: Make sure deleted parts of the selection range are added back on undo (currently, the MovingRange just moves forward and does not add back the
    // deleted range) bar.findNext();
    //
    // QCOMPARE(view.selectionRange(), Range(1, 0, 2, 0));
    // QVERIFY(bar.selectionOnly());
}

void SearchBarTest::testReplaceAll()
{
    KTextEditor::DocumentPrivate doc;
    KTextEditor::ViewPrivate view(&doc, nullptr);
    KateViewConfig config(&view);

    doc.setText(QStringLiteral("a a a"));
    KateSearchBar bar(true, &view, &config);

    bar.setSearchPattern(QStringLiteral("a"));
    bar.setReplacementPattern(QString());
    bar.replaceAll();

    QCOMPARE(bar.m_hlRanges.size(), 3);
    QCOMPARE(bar.m_hlRanges.at(0)->toRange(), Range(0, 0, 0, 0));
    QCOMPARE(bar.m_hlRanges.at(1)->toRange(), Range(0, 1, 0, 1));
    QCOMPARE(bar.m_hlRanges.at(2)->toRange(), Range(0, 2, 0, 2));

    bar.setSearchPattern(QStringLiteral(" "));
    bar.setReplacementPattern(QStringLiteral("b"));
    bar.replaceAll();

    QCOMPARE(bar.m_hlRanges.size(), 2);
    QCOMPARE(bar.m_hlRanges.at(0)->toRange(), Range(0, 0, 0, 1));
    QCOMPARE(bar.m_hlRanges.at(1)->toRange(), Range(0, 1, 0, 2));
}

void SearchBarTest::testFindSelectionForward_data()
{
    QTest::addColumn<QString>("text");
    QTest::addColumn<bool>("selectionOnly");
    QTest::addColumn<Range>("selectionRange");
    QTest::addColumn<Range>("match");

    testNewRow() << QStringLiteral("a a a") << false << Range(0, 0, 0, 1) << Range(0, 0, 0, 2);
    testNewRow() << QStringLiteral("a a a") << true << Range(0, 0, 0, 1) << Range(0, 0, 0, 1);

    testNewRow() << QStringLiteral("a a a") << false << Range(0, 0, 0, 2) << Range(0, 2, 0, 4);
    testNewRow() << QStringLiteral("a a a") << true << Range(0, 0, 0, 2) << Range(0, 0, 0, 2);

    testNewRow() << QStringLiteral("a a a") << false << Range(0, 0, 0, 3) << Range(0, 0, 0, 2);
    testNewRow() << QStringLiteral("a a a") << true << Range(0, 0, 0, 3) << Range(0, 0, 0, 2);

    testNewRow() << QStringLiteral("a a a") << false << Range(0, 2, 0, 4) << Range(0, 0, 0, 2);
    testNewRow() << QStringLiteral("a a a") << true << Range(0, 2, 0, 4) << Range(0, 2, 0, 4);

    testNewRow() << QStringLiteral("a a a") << false << Range(0, 3, 0, 4) << Range(0, 0, 0, 2);
    testNewRow() << QStringLiteral("a a a") << true << Range(0, 3, 0, 4) << Range(0, 3, 0, 4);
}

void SearchBarTest::testFindSelectionForward()
{
    QFETCH(QString, text);
    QFETCH(bool, selectionOnly);
    QFETCH(Range, selectionRange);
    QFETCH(Range, match);

    KTextEditor::DocumentPrivate doc;
    KTextEditor::ViewPrivate view(&doc, nullptr);
    KateViewConfig config(&view);

    doc.setText(text);

    view.setSelection(Range(0, 0, 0, 2));

    KateSearchBar bar(true, &view, &config);
    QVERIFY(bar.searchPattern() == QStringLiteral("a "));

    view.setSelection(selectionRange);
    QCOMPARE(view.selectionRange(), selectionRange);
    bar.setSelectionOnly(selectionOnly);

    bar.findNext();

    QCOMPARE(view.selectionRange(), match);
}

void SearchBarTest::testRemoveWithSelectionForward_data()
{
    QTest::addColumn<Range>("selectionRange");
    QTest::addColumn<Range>("match");

    testNewRow() << Range(0, 0, 0, 1) << Range(0, 0, 0, 2);
    testNewRow() << Range(0, 0, 0, 2) << Range(0, 0, 0, 2);
    testNewRow() << Range(0, 0, 0, 3) << Range(0, 0, 0, 2);
    testNewRow() << Range(0, 2, 0, 4) << Range(0, 0, 0, 2);
    testNewRow() << Range(0, 3, 0, 4) << Range(0, 0, 0, 2);
}

void SearchBarTest::testRemoveWithSelectionForward()
{
    QFETCH(Range, selectionRange);
    QFETCH(Range, match);

    KTextEditor::DocumentPrivate doc;
    KTextEditor::ViewPrivate view(&doc, nullptr);
    KateViewConfig config(&view);

    doc.setText(QStringLiteral("a a a"));
    view.setSelection(selectionRange);

    KateSearchBar bar(true, &view, &config);
    bar.setSearchPattern(QStringLiteral("a "));
    bar.setSelectionOnly(false);

    bar.replaceNext();

    QCOMPARE(view.selectionRange(), match);
}

void SearchBarTest::testRemoveInSelectionForward_data()
{
    QTest::addColumn<Range>("selectionRange");
    QTest::addColumn<Range>("match");

    testNewRow() << Range(0, 0, 0, 1) << Range(0, 0, 0, 1);
    testNewRow() << Range(0, 0, 0, 2) << Range(0, 0, 0, 0);
    testNewRow() << Range(0, 0, 0, 3) << Range(0, 0, 0, 2);
    testNewRow() << Range(0, 0, 0, 4) << Range(0, 0, 0, 2);
    testNewRow() << Range(0, 2, 0, 4) << Range(0, 2, 0, 2);
    testNewRow() << Range(0, 3, 0, 4) << Range(0, 3, 0, 4);
}

void SearchBarTest::testRemoveInSelectionForward()
{
    QFETCH(Range, selectionRange);
    QFETCH(Range, match);

    KTextEditor::DocumentPrivate doc;
    KTextEditor::ViewPrivate view(&doc, nullptr);
    KateViewConfig config(&view);

    doc.setText(QStringLiteral("a a a"));
    view.setSelection(selectionRange);

    KateSearchBar bar(true, &view, &config);
    bar.setSearchPattern(QStringLiteral("a "));
    bar.setSelectionOnly(true);

    QVERIFY(bar.replacementPattern().isEmpty());

    bar.replaceNext();

    QCOMPARE(view.selectionRange(), match);
}

void SearchBarTest::testReplaceWithDoubleSelecion_data()
{
    QTest::addColumn<QString>("text");
    QTest::addColumn<Range>("selectionRange");
    QTest::addColumn<QString>("result");
    QTest::addColumn<Range>("match");

    //  testNewRow() << "a"  << Range(0, 0, 0, 1) << "aa" << Range(?, ?, ?, ?);
    testNewRow() << QStringLiteral("aa") << Range(0, 1, 0, 2) << QStringLiteral("aaa") << Range(0, 0, 0, 1);
    testNewRow() << QStringLiteral("aa") << Range(0, 0, 0, 1) << QStringLiteral("aaa") << Range(0, 2, 0, 3);

    //  testNewRow() << "ab"  << Range(0, 0, 0, 1) << "aab"  << Range(?, ?, ?, ?);
    testNewRow() << QStringLiteral("aab") << Range(0, 0, 0, 1) << QStringLiteral("aaab") << Range(0, 2, 0, 3);
    testNewRow() << QStringLiteral("aba") << Range(0, 0, 0, 1) << QStringLiteral("aaba") << Range(0, 3, 0, 4);

    //  testNewRow() << "ab"   << Range(0, 0, 0, 2) << "abab"   << Range(?, ?, ?, ?);
    testNewRow() << QStringLiteral("abab") << Range(0, 0, 0, 2) << QStringLiteral("ababab") << Range(0, 4, 0, 6);
    testNewRow() << QStringLiteral("abab") << Range(0, 2, 0, 4) << QStringLiteral("ababab") << Range(0, 0, 0, 2);
}

void SearchBarTest::testReplaceWithDoubleSelecion()
{
    QFETCH(QString, text);
    QFETCH(Range, selectionRange);
    QFETCH(QString, result);
    QFETCH(Range, match);

    KTextEditor::DocumentPrivate doc;
    KTextEditor::ViewPrivate view(&doc, nullptr);
    KateViewConfig config(&view);

    doc.setText(text);
    view.setSelection(selectionRange);

    KateSearchBar bar(true, &view, &config);

    bar.setSelectionOnly(false);
    bar.setReplacementPattern(bar.searchPattern() + bar.searchPattern());
    bar.replaceNext();

    QCOMPARE(doc.text(), result);
    QCOMPARE(view.selectionRange(), match);
}

void SearchBarTest::testReplaceDollar()
{
    KTextEditor::DocumentPrivate doc;
    KTextEditor::ViewPrivate view(&doc, nullptr);
    KateViewConfig config(&view);

    doc.setText(QStringLiteral("aaa\nbbb\nccc\n\n\naaa\nbbb\nccc\nddd\n"));

    KateSearchBar bar(true, &view, &config);

    bar.setSearchPattern(QStringLiteral("$"));
    bar.setSearchMode(KateSearchBar::MODE_REGEX);
    bar.setReplacementPattern(QStringLiteral("D"));
    bar.replaceAll();
    QCOMPARE(doc.text(), QStringLiteral("aaaD\nbbbD\ncccD\nD\nD\naaaD\nbbbD\ncccD\ndddD\n"));
}

void SearchBarTest::testSearchHistoryIncremental()
{
    KTextEditor::DocumentPrivate doc;
    KTextEditor::ViewPrivate view(&doc, nullptr);
    KateViewConfig *const config = view.config();
    KTextEditor::EditorPrivate::self()->searchHistoryModel()->setStringList(QStringList());

    doc.setText(QStringLiteral("foo bar"));

    KateSearchBar bar(false, &view, config);

    bar.setSearchPattern(QStringLiteral("foo"));
    bar.findNext();

    QCOMPARE(bar.m_incUi->pattern->findText(QStringLiteral("foo")), 0);

    bar.setSearchPattern(QStringLiteral("bar"));
    bar.findNext();

    QCOMPARE(bar.m_incUi->pattern->findText(QStringLiteral("bar")), 0);
    QCOMPARE(bar.m_incUi->pattern->findText(QStringLiteral("foo")), 1);

    KTextEditor::DocumentPrivate doc2;
    KTextEditor::ViewPrivate view2(&doc2, nullptr);
    KateViewConfig *const config2 = view2.config();
    KateSearchBar bar2(false, &view2, config2);

    QCOMPARE(bar2.m_incUi->pattern->findText(QStringLiteral("bar")), 0);
    QCOMPARE(bar2.m_incUi->pattern->findText(QStringLiteral("foo")), 1);

    // testcase for https://bugs.kde.org/show_bug.cgi?id=248305
    bar2.m_incUi->pattern->setCurrentIndex(1);
    QCOMPARE(bar2.searchPattern(), QLatin1String("foo"));
    bar2.findNext();
    QCOMPARE(bar2.searchPattern(), QLatin1String("foo"));
}

void SearchBarTest::testSearchHistoryPower()
{
    KTextEditor::DocumentPrivate doc;
    KTextEditor::ViewPrivate view(&doc, nullptr);
    KateViewConfig *const config = view.config();
    KTextEditor::EditorPrivate::self()->searchHistoryModel()->setStringList(QStringList());

    doc.setText(QStringLiteral("foo bar"));

    KateSearchBar bar(true, &view, config);

    QCOMPARE(bar.m_powerUi->pattern->count(), 0);

    bar.setSearchPattern(QStringLiteral("foo"));
    bar.findNext();

    QCOMPARE(bar.m_powerUi->pattern->findText(QStringLiteral("foo")), 0);

    bar.findNext();

    QCOMPARE(bar.m_powerUi->pattern->findText(QStringLiteral("foo")), 0);
    QCOMPARE(bar.m_powerUi->pattern->count(), 1);

    bar.setSearchPattern(QStringLiteral("bar"));
    bar.findNext();

    QCOMPARE(bar.m_powerUi->pattern->findText(QStringLiteral("bar")), 0);
    QCOMPARE(bar.m_powerUi->pattern->findText(QStringLiteral("foo")), 1);
    QCOMPARE(bar.m_powerUi->pattern->count(), 2);

    KTextEditor::DocumentPrivate doc2;
    KTextEditor::ViewPrivate view2(&doc2, nullptr);
    KateViewConfig *const config2 = view2.config();
    KateSearchBar bar2(true, &view2, config2);

    QCOMPARE(bar2.m_powerUi->pattern->findText(QStringLiteral("bar")), 0);
    QCOMPARE(bar2.m_powerUi->pattern->findText(QStringLiteral("foo")), 1);
}

// Make sure Kate doesn't replace anything outside selection in block mode (see bug 253191)
void SearchBarTest::testReplaceInBlockMode()
{
    KTextEditor::DocumentPrivate doc;
    KTextEditor::ViewPrivate view(&doc, nullptr);
    view.setInputMode(View::NormalInputMode);
    KateViewConfig config(&view);

    doc.setText(QStringLiteral("111\n111"));
    view.setBlockSelection(true);
    view.setSelection(KTextEditor::Range(0, 1, 1, 2));

    KateSearchBar bar(true, &view, &config);

    bar.setSearchPattern(QStringLiteral("1"));
    bar.setReplacementPattern(QStringLiteral("2"));
    bar.replaceAll();

    QCOMPARE(doc.text(), QStringLiteral("121\n121"));
}

// Bug 456367 happens when all these conditions are met :
// - block selection mode is checked
// - option "search in the selection only" is not checked
// - some text is selected
//
// In this case it should find/replace all occurrences in the whole document.
void SearchBarTest::testNonEmptyBlockSelectionAndSearchInSelectionOnlyDisabled()
{
    KTextEditor::DocumentPrivate doc;
    KTextEditor::ViewPrivate view(&doc, nullptr);
    KateViewConfig config(&view);

    // Last line is required to reproduce the bug :
    doc.setText(QStringLiteral("111\n111\n"));
    view.setBlockSelection(true);
    view.setSelection(KTextEditor::Range(0, 1, 1, 2));

    KateSearchBar bar(true, &view, &config);

    bar.setSearchPattern(QStringLiteral("1"));
    bar.setReplacementPattern(QStringLiteral("2"));
    bar.setSelectionOnly(false);
    bar.replaceAll();

    // Should replace all occurrences in the whole document.
    QCOMPARE(doc.text(), QStringLiteral("222\n222\n"));
}

void SearchBarTest::testReplaceManyCapturesBug365124()
{
    KTextEditor::DocumentPrivate doc;
    KTextEditor::ViewPrivate view(&doc, nullptr);
    KateViewConfig config(&view);

    doc.setText(QStringLiteral("one two three four five six seven eight nine ten eleven twelve thirteen\n"));

    KateSearchBar bar(true, &view, &config);

    bar.setSearchPattern(QStringLiteral("^(.*) (.*) (.*) (.*) (.*) (.*) (.*) (.*) (.*) (.*) (.*) (.*) (.*)$"));
    bar.setSearchMode(KateSearchBar::MODE_REGEX);
    bar.setReplacementPattern(QStringLiteral("\\{1}::\\2::\\3::\\4::\\5::\\6::\\7::\\8::\\9::\\{10}::\\{11}::\\{12}::\\{13}"));

    bar.replaceAll();

    QCOMPARE(doc.text(), QStringLiteral("one::two::three::four::five::six::seven::eight::nine::ten::eleven::twelve::thirteen\n"));
}

void SearchBarTest::testReplaceEscapeSequence_data()
{
    QTest::addColumn<QString>("textBefore");
    QTest::addColumn<QString>("textAfter");
    QTest::addColumn<Cursor>("cursorBefore");
    QTest::addColumn<Cursor>("cursorAfter");

    testNewRow() << QStringLiteral("a\n") << QStringLiteral("a ") << Cursor(1, 0) << Cursor(0, 2);
    testNewRow() << QStringLiteral("a\nb\n") << QStringLiteral("a b ") << Cursor(2, 0) << Cursor(0, 4);
    testNewRow() << QStringLiteral("\n\n\n") << QStringLiteral("   ") << Cursor(3, 0) << Cursor(0, 3);
}

void SearchBarTest::testReplaceEscapeSequence()
{
    QFETCH(QString, textBefore);
    QFETCH(QString, textAfter);
    QFETCH(Cursor, cursorBefore);
    QFETCH(Cursor, cursorAfter);

    // testcase for https://bugs.kde.org/show_bug.cgi?id=381080
    KTextEditor::DocumentPrivate doc;
    KTextEditor::ViewPrivate view(&doc, nullptr);
    KateViewConfig config(&view);

    doc.setText(textBefore);
    view.setCursorPosition(cursorBefore);
    QCOMPARE(view.cursorPosition(), cursorBefore);

    KateSearchBar bar(true, &view, &config);

    bar.setSearchMode(KateSearchBar::MODE_ESCAPE_SEQUENCES);
    bar.setSearchPattern(QStringLiteral("\\n"));
    bar.setReplacementPattern(QStringLiteral(" "));

    bar.replaceAll();

    QCOMPARE(doc.text(), textAfter);
    QCOMPARE(view.cursorPosition(), cursorAfter);
}

#include "moc_searchbar_test.cpp"