//
//          Copyright (c) 2022, Gittyup Community
//
// This software is licensed under the MIT License. The LICENSE.md file
// describes the conditions under which this software may be distributed.
//
// Author: Martin Marmsoler
//

#define private public
#include "Test.h" // RepoView included here
#include "ui/RepoView.h" // To be able to access mDetail
#include "ui/DetailView.h" // To be able to access mContent
#include "ui/DoubleTreeWidget.h" // To be able to access the diffview
#include "ui/DiffView/DiffView.h" // To be able to access the filewidgets
#undef private

#include "ui/MainWindow.h"
#include "ui/DetailView.h"
#include "ui/DiffView/FileWidget.h"
#include "ui/DiffView/HunkWidget.h"

#include "git/Reference.h"
#include "git/Diff.h"
#include "git/Commit.h"
#include "git/Tree.h"
#include "git/Patch.h"
#include "git/Command.h"

#include "tools/MergeTool.h" // TODO: remove

#include "log/LogEntry.h"

#include <QStackedWidget>

// TODO: move this macro into Test.h is also used in TestEditorLines
#define INIT_REPO(repoPath, /* bool */ useTempDir)                             \
  QString path = Test::extractRepository(repoPath, useTempDir);                \
  QVERIFY(!path.isEmpty());                                                    \
  mRepo = git::Repository::open(path);                                         \
  QVERIFY(mRepo.isValid());                                                    \
  MainWindow window(mRepo);                                                    \
  window.show();                                                               \
  QVERIFY(QTest::qWaitForWindowExposed(&window));                              \
                                                                               \
  git::Reference head = mRepo.head();                                          \
  git::Commit commit = head.target();                                          \
  git::Diff stagedDiff = mRepo.diffTreeToIndex(commit.tree()); /* correct */   \
                                                                               \
  RepoView *repoView = window.currentView();                                   \
  DiffView diffView = DiffView(mRepo, repoView);                               \
  auto diff = mRepo.status(mRepo.index(), nullptr, false);

#ifndef GIT_EXECUTABLE
#error "To execute those tests it is neccessary to have git installed on your computer. Turn off tests or exclude this test to build the project"
#endif

#define EXECUTE_GIT_COMMAND(workdir, arguments) \
{ \
    QProcess p(this); \
    p.setWorkingDirectory(workdir); \
    QString bash = git::Command::bashPath(); \
    QVERIFY(!bash.isEmpty()); \
    QString command = QString(GIT_EXECUTABLE) + " " + arguments; \
    QStringList a = {"-c", command}; \
    p.start(bash, a); \
    p.waitForStarted(); \
    p.waitForFinished(); \
   /*  QCOMPARE(p.exitCode(), 1); TODO turn on */ \
}

using namespace git;

class TestRebase : public QObject {
  Q_OBJECT

private slots:
    void withoutConflicts();
    void conflictingRebase();
    void continueExternalStartedRebase(); // must have conflicts otherwise it is not possible to continue
    void startRebaseContinueInCLI();
    void startRebaseContinueInCLIContinueGUI(); // start rebase, continue in the cli and finish in the GUI
    void abortMR();
    void commitDuringRebase();
private:
    git::Repository mRepo;
};

//###################################################################################################
//###################################################################################################
//###################################################################################################
// Tests starting ######################################################################################
//###################################################################################################
//###################################################################################################
//###################################################################################################

void TestRebase::withoutConflicts() {
    INIT_REPO("rebaseConflicts.zip", false);

    int rebaseFinished = 0;
    int rebaseAboutToRebase = 0;
    int rebaseCommitSuccess = 0;
    int rebaseConflict = 0;

    connect(mRepo.notifier(), &git::RepositoryNotifier::rebaseInitError, [=](){QVERIFY(false);});  // Should not be called
    connect(mRepo.notifier(), &git::RepositoryNotifier::rebaseAboutToRebase, [=, &rebaseAboutToRebase](const Rebase rebase, const Commit before, int count, LogEntry* parent){
        QVERIFY(rebase.isValid());
        QCOMPARE(count, 1);
        QCOMPARE(before.message(), "File2.txt added");
        rebaseAboutToRebase++;
    });
    connect(mRepo.notifier(), &git::RepositoryNotifier::rebaseCommitInvalid, [=](){QVERIFY(false);}); // Should not be called
    connect(mRepo.notifier(), &git::RepositoryNotifier::rebaseFinished, [=, &rebaseFinished](){rebaseFinished++;});
    connect(mRepo.notifier(), &git::RepositoryNotifier::rebaseCommitSuccess, [=, &rebaseCommitSuccess](const Rebase rebase, const Commit before, const Commit after, int counter, LogEntry* parent) {
        QVERIFY(rebase.isValid());
        rebaseCommitSuccess++;
    });
    connect(mRepo.notifier(), &git::RepositoryNotifier::rebaseConflict, [=, &rebaseConflict](){
        rebaseConflict++;
    });

    const QString rebaseBranchName = "refs/heads/noConflict";

    git::Reference branch = mRepo.lookupRef(rebaseBranchName);
    QVERIFY(branch.isValid());
    auto c = branch.annotatedCommit().commit();

    LogEntry *entry = repoView->addLogEntry("Rebase", "Rebase", nullptr);

    // Checkout correct branch
    QCOMPARE(mRepo.checkout(c), true);
    //Test::refresh(repoView, false);

    // Rebase on main
    git::Reference mainBranch = mRepo.lookupRef(QString("refs/heads/main"));
    QVERIFY(mainBranch.isValid());
    auto ac = mainBranch.annotatedCommit();
    repoView->rebase(ac, entry);

    // Check that branch is based on "main" now
    branch = mRepo.lookupRef(rebaseBranchName);
    QVERIFY(branch.isValid());
    QList<Commit> parents = branch.annotatedCommit().commit().parents();
    QCOMPARE(parents.count(), 1);
    QCOMPARE(parents.at(0).id(), ac.commit().id());

    // Check that rebase was really finished
    QCOMPARE(mRepo.rebaseOngoing(), false);

    // Check call counters
    QCOMPARE(rebaseFinished, 1);
    QCOMPARE(rebaseAboutToRebase, 1);
    QCOMPARE(rebaseCommitSuccess, 1);
    QCOMPARE(rebaseConflict, 0);

    QCOMPARE(repoView->mDetails->isRebaseContinueVisible(), false);
    QCOMPARE(repoView->mDetails->isRebaseAbortVisible(), false);
}

void TestRebase::conflictingRebase() {
    INIT_REPO("rebaseConflicts.zip", false);

    QCOMPARE(repoView->mDetails->isRebaseContinueVisible(), false);
    QCOMPARE(repoView->mDetails->isRebaseAbortVisible(), false);

    int rebaseFinished = 0;
    int rebaseAboutToRebase = 0;
    int rebaseCommitSuccess = 0;
    int rebaseConflict = 0;
    int refreshTriggered = 0;

    connect(mRepo.notifier(), &git::RepositoryNotifier::rebaseInitError, [=](){QVERIFY(false);});  // Should not be called
    connect(mRepo.notifier(), &git::RepositoryNotifier::rebaseAboutToRebase, [=, &rebaseAboutToRebase](const Rebase rebase, const Commit before, int count, LogEntry* parent){
        QVERIFY(rebase.isValid());
        QCOMPARE(count, 1);
        QCOMPARE(before.message(), "File.txt changed by second branch\n");
        rebaseAboutToRebase++;
    });
    // TODO: = needed?
    connect(mRepo.notifier(), &git::RepositoryNotifier::rebaseCommitInvalid, [=](){QVERIFY(false);}); // Should not be called
    connect(mRepo.notifier(), &git::RepositoryNotifier::rebaseFinished, [=, &rebaseFinished](){rebaseFinished++;});
    connect(mRepo.notifier(), &git::RepositoryNotifier::rebaseCommitSuccess, [=, &rebaseCommitSuccess](const Rebase rebase, const Commit before, const Commit after, int counter, LogEntry* parent) {
        QVERIFY(rebase.isValid());
        rebaseCommitSuccess++;
    });
    connect(mRepo.notifier(), &git::RepositoryNotifier::rebaseConflict, [=, &rebaseConflict, &rebaseCommitSuccess](){
        QCOMPARE(rebaseCommitSuccess, 0); // was not called yet
        rebaseConflict++;
    });

    connect(mRepo.notifier(), &RepositoryNotifier::referenceUpdated, [this, &refreshTriggered](const Reference &ref) {
        // TODO: enable
        //QCOMPARE(ref, mRepo.head());
        refreshTriggered++;
    });

    const QString rebaseBranchName = "refs/heads/singleCommitConflict";

    git::Reference branch = mRepo.lookupRef(rebaseBranchName);
    QVERIFY(branch.isValid());
    auto c = branch.annotatedCommit().commit();

    // Checkout correct branch
    //QCOMPARE(mRepo.checkout(c), true);
    repoView->checkout(branch);
    //Test::refresh(repoView);

    // Rebase on main
    git::Reference mainBranch = mRepo.lookupRef(QString("refs/heads/main"));
    QVERIFY(mainBranch.isValid());
    auto ac = mainBranch.annotatedCommit();
    refreshTriggered = 0;
    LogEntry *entry = repoView->addLogEntry("Rebase", "Rebase", nullptr);
    repoView->rebase(ac, entry);
    QCOMPARE(refreshTriggered, 1); // Check that refresh was triggered

    QCOMPARE(mRepo.rebaseOngoing(), true);
    QCOMPARE(rebaseFinished, 0);
    QCOMPARE(rebaseConflict, 1);


    // Check that buttons are visible
    QTest::qWait(100);
    QCOMPARE(repoView->mDetails->isRebaseContinueVisible(), true);
    QCOMPARE(repoView->mDetails->isRebaseAbortVisible(), true);

    // Resolve conflicts
    diff = mRepo.status(mRepo.index(), nullptr, false);
    QCOMPARE(diff.count(), 1);
    QCOMPARE(diff.patch(0).isConflicted(), true);
    QFile f(mRepo.workdir().filePath(diff.patch(0).name()));
    QCOMPARE(f.open(QIODevice::WriteOnly), true);
    QVERIFY(f.write("Test123") != -1); // just write something to resolve the conflict
    f.close();

    Test::refresh(repoView);

    repoView->continueRebase(); // should fail
    QCOMPARE(rebaseConflict, 2); // User tries to continue without staging

    Test::refresh(repoView);

    // Staging the file
    auto cw = static_cast<ContentWidget*>(repoView->mDetails->mContent->currentWidget());
    auto dtw = dynamic_cast<DoubleTreeWidget*>(cw);
    QVERIFY(dtw);
    dtw->mDiffView->mFiles.at(0)->stageStateChanged(dtw->mDiffView->mFiles.at(0)->modelIndex(), git::Index::StagedState::Staged);

// Does not work, because for some reason the diffView is not updated as the dtw->DiffView
//    // stage file otherwise it is not possible to continue
//    {
//        auto stagedDiff = mRepo.diffTreeToIndex(commit.tree()); /* correct */
//        auto diff = mRepo.status(mRepo.index(), nullptr, false);
//        auto stagedPatch = stagedDiff.patch(0);
//        auto name = diff.patch(0).name();
//        bool submodule = mRepo.lookupSubmodule(name).isValid();


//        FileWidget fw(&diffView, diff, diff.patch(0), stagedPatch, QModelIndex(), name,
//                      path, submodule);
//        auto hunks = fw.hunks();
//        QVERIFY(hunks.count() == 1);
//        hunks[0]->load();

//        fw.stageHunks(nullptr, git::Index::StagedState::Staged, true, true);
//    }

    repoView->mDetails->setCommitMessage("Test message");

    refreshTriggered = 0;
    rebaseConflict = 0;
    repoView->continueRebase();
    QCOMPARE(refreshTriggered, 1);

    // Check that branch is based on "main" now
    branch = mRepo.lookupRef(rebaseBranchName);
    QVERIFY(branch.isValid());
    QList<Commit> parents = branch.annotatedCommit().commit().parents();
    QCOMPARE(parents.count(), 1);
    QCOMPARE(parents.at(0).id(), ac.commit().id());
    QCOMPARE(branch.annotatedCommit().commit().message(), "Test message"); // custom message was used instead of the original one

    // Check that rebase was really finished
    QCOMPARE(mRepo.rebaseOngoing(), false);

    // Check that buttons are visible
    QCOMPARE(repoView->mDetails->isRebaseContinueVisible(), false);
    QCOMPARE(repoView->mDetails->isRebaseAbortVisible(), false);

    // Check call counters
    QCOMPARE(rebaseFinished, 1);
    QCOMPARE(rebaseAboutToRebase, 1);
    QCOMPARE(rebaseCommitSuccess, 1);
    QCOMPARE(rebaseConflict, 0);
}

void TestRebase::continueExternalStartedRebase() {
    INIT_REPO("rebaseConflicts.zip", false);

    QCOMPARE(repoView->mDetails->isRebaseContinueVisible(), false);
    QCOMPARE(repoView->mDetails->isRebaseAbortVisible(), false);

    int rebaseFinished = 0;
    int rebaseAboutToRebase = 0;
    int rebaseCommitSuccess = 0;
    int rebaseConflict = 0;
    int refreshTriggered = 0;
    int workdirChangeTriggered = 0;

    connect(mRepo.notifier(), &git::RepositoryNotifier::rebaseInitError, [=](){QVERIFY(false);});  // Should not be called
    connect(mRepo.notifier(), &git::RepositoryNotifier::rebaseAboutToRebase, [=, &rebaseAboutToRebase](const Rebase rebase, const Commit before, int count, LogEntry* parent){
        QVERIFY(rebase.isValid());
        QCOMPARE(count, 1);
        QCOMPARE(before.message(), "File.txt changed by second branch\n");
        rebaseAboutToRebase++;
    });
    // TODO: = needed?
    connect(mRepo.notifier(), &git::RepositoryNotifier::rebaseCommitInvalid, [=](){QVERIFY(false);}); // Should not be called
    connect(mRepo.notifier(), &git::RepositoryNotifier::rebaseFinished, [=, &rebaseFinished](){rebaseFinished++;});
    connect(mRepo.notifier(), &git::RepositoryNotifier::rebaseCommitSuccess, [=, &rebaseCommitSuccess](const Rebase rebase, const Commit before, const Commit after, int counter, LogEntry* parent) {
        QVERIFY(rebase.isValid());
        rebaseCommitSuccess++;
    });
    connect(mRepo.notifier(), &git::RepositoryNotifier::rebaseConflict, [=, &rebaseConflict, &rebaseCommitSuccess](){
        QCOMPARE(rebaseCommitSuccess, 0); // was not called yet
        rebaseConflict++;
    });

    connect(mRepo.notifier(), &RepositoryNotifier::referenceUpdated, [this, &refreshTriggered](const Reference &ref) {
        // TODO: enable
        //QCOMPARE(ref, mRepo.head());
        refreshTriggered++;
    });

    connect(mRepo.notifier(), &RepositoryNotifier::workdirChanged, [this, &workdirChangeTriggered]() {
        workdirChangeTriggered++;
    });

    EXECUTE_GIT_COMMAND(path, "checkout singleCommitConflict")
    EXECUTE_GIT_COMMAND(path, "rebase main")

    workdirChangeTriggered = 0;
    while (workdirChangeTriggered == 0)
        QTest::qWait(100);

    QCOMPARE(repoView->mDetails->isRebaseContinueVisible(), true);
    QCOMPARE(repoView->mDetails->isRebaseAbortVisible(), true);

    diff = mRepo.status(mRepo.index(), nullptr, false);
    QCOMPARE(diff.count(), 1);
    QCOMPARE(diff.patch(0).isConflicted(), true);
    QFile f(mRepo.workdir().filePath(diff.patch(0).name()));
    QCOMPARE(f.open(QIODevice::WriteOnly), true);
    QVERIFY(f.write("Test123") != -1); // just write something to resolve the conflict
    f.close();

    repoView->continueRebase();

    QCOMPARE(rebaseFinished, 1);
    QCOMPARE(rebaseAboutToRebase, 1);
    QCOMPARE(rebaseCommitSuccess, 1);
    QCOMPARE(rebaseConflict, 0);
}

void TestRebase::startRebaseContinueInCLI() {
    // Check that GUI is updated correctly

    INIT_REPO("rebaseConflicts.zip", false);

    int rebaseFinished = 0;
    int rebaseAboutToRebase = 0;
    int rebaseCommitSuccess = 0;
    int rebaseConflict = 0;
    int refreshTriggered = 0;

    connect(mRepo.notifier(), &git::RepositoryNotifier::rebaseInitError, [=](){QVERIFY(false);});  // Should not be called
    connect(mRepo.notifier(), &git::RepositoryNotifier::rebaseAboutToRebase, [=, &rebaseAboutToRebase](const Rebase rebase, const Commit before, int count, LogEntry* parent){
        QVERIFY(rebase.isValid());
        QCOMPARE(count, 1);
        QCOMPARE(before.message(), "File.txt changed by second branch\n");
        rebaseAboutToRebase++;
    });
    // TODO: = needed?
    connect(mRepo.notifier(), &git::RepositoryNotifier::rebaseCommitInvalid, [=](){QVERIFY(false);}); // Should not be called
    connect(mRepo.notifier(), &git::RepositoryNotifier::rebaseFinished, [=, &rebaseFinished](){rebaseFinished++;});
    connect(mRepo.notifier(), &git::RepositoryNotifier::rebaseCommitSuccess, [=, &rebaseCommitSuccess](const Rebase rebase, const Commit before, const Commit after, int counter, LogEntry* parent) {
        QVERIFY(rebase.isValid());
        rebaseCommitSuccess++;
    });
    connect(mRepo.notifier(), &git::RepositoryNotifier::rebaseConflict, [=, &rebaseConflict, &rebaseCommitSuccess](){
        QCOMPARE(rebaseCommitSuccess, 0); // was not called yet
        rebaseConflict++;
    });

    connect(mRepo.notifier(), &RepositoryNotifier::referenceUpdated, [this, &refreshTriggered](const Reference &ref) {
        // TODO: enable
        //QCOMPARE(ref, mRepo.head());
        refreshTriggered++;
    });

    const QString rebaseBranchName = "refs/heads/singleCommitConflict";

    git::Reference branch = mRepo.lookupRef(rebaseBranchName);
    QVERIFY(branch.isValid());
    auto c = branch.annotatedCommit().commit();

    // Checkout correct branch
    //QCOMPARE(mRepo.checkout(c), true);
    repoView->checkout(branch);
    //Test::refresh(repoView);

    // Rebase on main
    git::Reference mainBranch = mRepo.lookupRef(QString("refs/heads/main"));
    QVERIFY(mainBranch.isValid());
    auto ac = mainBranch.annotatedCommit();
    refreshTriggered = 0;
    LogEntry *entry = repoView->addLogEntry("Rebase", "Rebase", nullptr);
    repoView->rebase(ac, entry);
    QCOMPARE(refreshTriggered, 1); // Check that refresh was triggered

    QCOMPARE(mRepo.rebaseOngoing(), true);
    QCOMPARE(rebaseFinished, 0);
    QCOMPARE(rebaseConflict, 1);

    QCOMPARE(repoView->mDetails->isRebaseContinueVisible(), true);
    QCOMPARE(repoView->mDetails->isRebaseAbortVisible(), true);

    // Solve conflict
    QFile f(mRepo.workdir().filePath(diff.patch(0).name()));
    QCOMPARE(f.open(QIODevice::WriteOnly), true);
    QVERIFY(f.write("Test123") != -1); // just write something to resolve the conflict
    f.close();

    EXECUTE_GIT_COMMAND(path, "rebase --continue")

    // TODO: check that GUI was updated properly
    QCOMPARE(repoView->mDetails->isRebaseContinueVisible(), false);
    QCOMPARE(repoView->mDetails->isRebaseAbortVisible(), false);

    // Check that rebase was really finished
    QCOMPARE(mRepo.rebaseOngoing(), false);
}

void TestRebase::startRebaseContinueInCLIContinueGUI() {
    // Check that GUI is updated correctly

    INIT_REPO("rebaseConflicts.zip", false);

    QCOMPARE(repoView->mDetails->isRebaseContinueVisible(), false);
    QCOMPARE(repoView->mDetails->isRebaseAbortVisible(), false);

    int rebaseFinished = 0;
    int rebaseAboutToRebase = 0;
    int rebaseCommitSuccess = 0;
    int rebaseConflict = 0;
    int refreshTriggered = 0;
    int workdirChangeTriggered = 0;

    connect(mRepo.notifier(), &git::RepositoryNotifier::rebaseInitError, [=](){QVERIFY(false);});  // Should not be called
    connect(mRepo.notifier(), &git::RepositoryNotifier::rebaseAboutToRebase, [=, &rebaseAboutToRebase](const Rebase rebase, const Commit before, int count, LogEntry* parent){
        QVERIFY(rebase.isValid());
        QCOMPARE(count, 1);
        QCOMPARE(before.message(), "File.txt changed by second branch\n");
        rebaseAboutToRebase++;
    });
    // TODO: = needed?
    connect(mRepo.notifier(), &git::RepositoryNotifier::rebaseCommitInvalid, [=](){QVERIFY(false);}); // Should not be called
    connect(mRepo.notifier(), &git::RepositoryNotifier::rebaseFinished, [=, &rebaseFinished](){rebaseFinished++;});
    connect(mRepo.notifier(), &git::RepositoryNotifier::rebaseCommitSuccess, [=, &rebaseCommitSuccess](const Rebase rebase, const Commit before, const Commit after, int counter, LogEntry* parent) {
        QVERIFY(rebase.isValid());
        rebaseCommitSuccess++;
    });
    connect(mRepo.notifier(), &git::RepositoryNotifier::rebaseConflict, [=, &rebaseConflict, &rebaseCommitSuccess](){
        QCOMPARE(rebaseCommitSuccess, 0); // was not called yet
        rebaseConflict++;
    });

    connect(mRepo.notifier(), &RepositoryNotifier::referenceUpdated, [this, &refreshTriggered](const Reference &ref) {
        // TODO: enable
        //QCOMPARE(ref, mRepo.head());
        refreshTriggered++;
    });

    connect(mRepo.notifier(), &RepositoryNotifier::workdirChanged, [this, &workdirChangeTriggered]() {
        workdirChangeTriggered++;
    });

    const QString rebaseBranchName = "refs/heads/singleCommitConflict";

    git::Reference branch = mRepo.lookupRef(rebaseBranchName);
    QVERIFY(branch.isValid());
    auto c = branch.annotatedCommit().commit();

    // Checkout correct branch
    //QCOMPARE(mRepo.checkout(c), true);
    repoView->checkout(branch);
    //Test::refresh(repoView);

    // Rebase on main
    git::Reference mainBranch = mRepo.lookupRef(QString("refs/heads/main"));
    QVERIFY(mainBranch.isValid());
    auto ac = mainBranch.annotatedCommit();
    refreshTriggered = 0;
    LogEntry *entry = repoView->addLogEntry("Rebase", "Rebase", nullptr);
    repoView->rebase(ac, entry);
    QCOMPARE(refreshTriggered, 1); // Check that refresh was triggered

    QCOMPARE(mRepo.rebaseOngoing(), true);
    QCOMPARE(rebaseFinished, 0);
    QCOMPARE(rebaseConflict, 1);

    // Gui updated
    QCOMPARE(repoView->mDetails->isRebaseContinueVisible(), true);
    QCOMPARE(repoView->mDetails->isRebaseAbortVisible(), true);

    // Solve conflict
    QFile f(mRepo.workdir().filePath(diff.patch(0).name()));
    QCOMPARE(f.open(QIODevice::WriteOnly), true);
    QVERIFY(f.write("Test123") != -1); // just write something to resolve the conflict
    f.close();

    EXECUTE_GIT_COMMAND(path, "rebase --continue")

    workdirChangeTriggered = 0;
    while (workdirChangeTriggered == 0)
        QTest::qWait(100);

    // TODO: another condition is needed, because there is no change in those visible so
    // it is not enough
    QCOMPARE(repoView->mDetails->isRebaseContinueVisible(), true);
    QCOMPARE(repoView->mDetails->isRebaseAbortVisible(), true);
}

void TestRebase::abortMR() {
    INIT_REPO("rebaseConflicts.zip", false);

    QCOMPARE(repoView->mDetails->isRebaseContinueVisible(), false);
    QCOMPARE(repoView->mDetails->isRebaseAbortVisible(), false);

    int rebaseFinished = 0;
    int rebaseAboutToRebase = 0;
    int rebaseCommitSuccess = 0;
    int rebaseConflict = 0;
    int refreshTriggered = 0;

    connect(mRepo.notifier(), &git::RepositoryNotifier::rebaseInitError, [=](){QVERIFY(false);});  // Should not be called
    connect(mRepo.notifier(), &git::RepositoryNotifier::rebaseAboutToRebase, [=, &rebaseAboutToRebase](const Rebase rebase, const Commit before, int count, LogEntry* parent){
        QVERIFY(rebase.isValid());
        QCOMPARE(count, 1);
        QCOMPARE(before.message(), "File.txt changed by second branch\n");
        rebaseAboutToRebase++;
    });
    // TODO: = needed?
    connect(mRepo.notifier(), &git::RepositoryNotifier::rebaseCommitInvalid, [=](){QVERIFY(false);}); // Should not be called
    connect(mRepo.notifier(), &git::RepositoryNotifier::rebaseFinished, [=, &rebaseFinished](){rebaseFinished++;});
    connect(mRepo.notifier(), &git::RepositoryNotifier::rebaseCommitSuccess, [=, &rebaseCommitSuccess](const Rebase rebase, const Commit before, const Commit after, int counter, LogEntry* parent) {
        QVERIFY(rebase.isValid());
        rebaseCommitSuccess++;
    });
    connect(mRepo.notifier(), &git::RepositoryNotifier::rebaseConflict, [=, &rebaseConflict, &rebaseCommitSuccess](){
        QCOMPARE(rebaseCommitSuccess, 0); // was not called yet
        rebaseConflict++;
    });

    connect(mRepo.notifier(), &RepositoryNotifier::referenceUpdated, [this, &refreshTriggered](const Reference &ref) {
        // TODO: enable
        //QCOMPARE(ref, mRepo.head());
        refreshTriggered++;
    });

    const QString rebaseBranchName = "refs/heads/singleCommitConflict";

    git::Reference branch = mRepo.lookupRef(rebaseBranchName);
    QVERIFY(branch.isValid());
    auto c = branch.annotatedCommit().commit();

    // Checkout correct branch
    //QCOMPARE(mRepo.checkout(c), true);
    repoView->checkout(branch);
    //Test::refresh(repoView);

    // Rebase on main
    git::Reference mainBranch = mRepo.lookupRef(QString("refs/heads/main"));
    QVERIFY(mainBranch.isValid());
    auto ac = mainBranch.annotatedCommit();
    refreshTriggered = 0;
    LogEntry *entry = repoView->addLogEntry("Rebase", "Rebase", nullptr);
    repoView->rebase(ac, entry);
    QCOMPARE(refreshTriggered, 1); // Check that refresh was triggered

    QCOMPARE(mRepo.rebaseOngoing(), true);
    QCOMPARE(rebaseFinished, 0);
    QCOMPARE(rebaseConflict, 1);


    // Check that buttons are visible
    QTest::qWait(100);
    QCOMPARE(repoView->mDetails->isRebaseContinueVisible(), true);
    QCOMPARE(repoView->mDetails->isRebaseAbortVisible(), true);

    // Resolve conflicts
    diff = mRepo.status(mRepo.index(), nullptr, false);
    QCOMPARE(diff.count(), 1);
    QCOMPARE(diff.patch(0).isConflicted(), true);
    QFile f(mRepo.workdir().filePath(diff.patch(0).name()));
    QCOMPARE(f.open(QIODevice::WriteOnly), true);
    QVERIFY(f.write("Test123") != -1); // just write something to resolve the conflict
    f.close();

    Test::refresh(repoView); // TODO: check if they are not triggered automatically

    // Staging the file
    auto cw = static_cast<ContentWidget*>(repoView->mDetails->mContent->currentWidget());
    auto dtw = dynamic_cast<DoubleTreeWidget*>(cw);
    QVERIFY(dtw);
    dtw->mDiffView->mFiles.at(0)->stageStateChanged(dtw->mDiffView->mFiles.at(0)->modelIndex(), git::Index::StagedState::Staged);

    repoView->mDetails->setCommitMessage("Test message");

    refreshTriggered = 0;
    rebaseConflict = 0;
    repoView->abortRebase();
    QCOMPARE(refreshTriggered, 1);

    // Check that rebase was really finished
    QCOMPARE(mRepo.rebaseOngoing(), false);

    // Check that buttons are visible
    QCOMPARE(repoView->mDetails->isRebaseContinueVisible(), false);
    QCOMPARE(repoView->mDetails->isRebaseAbortVisible(), false);

    // Check call counters
    QCOMPARE(rebaseFinished, 1);
    QCOMPARE(rebaseAboutToRebase, 1);
    QCOMPARE(rebaseCommitSuccess, 1);
    QCOMPARE(rebaseConflict, 0);
}

void TestRebase::commitDuringRebase() {
    // TODO: implement
    /*
     * Start rebasing
     * Solve conflicts
     * Commit instead of continue rebase
     * Commit something else too
     * Continue rebase */

    INIT_REPO("rebaseConflicts.zip", false);

    QCOMPARE(repoView->mDetails->isRebaseContinueVisible(), false);
    QCOMPARE(repoView->mDetails->isRebaseAbortVisible(), false);

    int rebaseFinished = 0;
    int rebaseAboutToRebase = 0;
    int rebaseCommitSuccess = 0;
    int rebaseConflict = 0;
    int refreshTriggered = 0;

    connect(mRepo.notifier(), &git::RepositoryNotifier::rebaseInitError, [=](){QVERIFY(false);});  // Should not be called
    connect(mRepo.notifier(), &git::RepositoryNotifier::rebaseAboutToRebase, [=, &rebaseAboutToRebase](const Rebase rebase, const Commit before, int count, LogEntry* parent){
        QVERIFY(rebase.isValid());
        QCOMPARE(count, 1);
        QCOMPARE(before.message(), "File.txt changed by second branch\n");
        rebaseAboutToRebase++;
    });
    // TODO: = needed?
    connect(mRepo.notifier(), &git::RepositoryNotifier::rebaseCommitInvalid, [=](){QVERIFY(false);}); // Should not be called
    connect(mRepo.notifier(), &git::RepositoryNotifier::rebaseFinished, [=, &rebaseFinished](){rebaseFinished++;});
    connect(mRepo.notifier(), &git::RepositoryNotifier::rebaseCommitSuccess, [=, &rebaseCommitSuccess](const Rebase rebase, const Commit before, const Commit after, int counter, LogEntry* parent) {
        QVERIFY(rebase.isValid());
        rebaseCommitSuccess++;
    });
    connect(mRepo.notifier(), &git::RepositoryNotifier::rebaseConflict, [=, &rebaseConflict, &rebaseCommitSuccess](){
        QCOMPARE(rebaseCommitSuccess, 0); // was not called yet
        rebaseConflict++;
    });

    connect(mRepo.notifier(), &RepositoryNotifier::referenceUpdated, [this, &refreshTriggered](const Reference &ref) {
        // TODO: enable
        //QCOMPARE(ref, mRepo.head());
        refreshTriggered++;
    });

    const QString rebaseBranchName = "refs/heads/singleCommitConflict";

    git::Reference branch = mRepo.lookupRef(rebaseBranchName);
    QVERIFY(branch.isValid());
    auto c = branch.annotatedCommit().commit();

    // Checkout correct branch
    //QCOMPARE(mRepo.checkout(c), true);
    repoView->checkout(branch);
    //Test::refresh(repoView);

    // Rebase on main
    git::Reference mainBranch = mRepo.lookupRef(QString("refs/heads/main"));
    QVERIFY(mainBranch.isValid());
    auto ac = mainBranch.annotatedCommit();
    refreshTriggered = 0;
    LogEntry *entry = repoView->addLogEntry("Rebase", "Rebase", nullptr);
    repoView->rebase(ac, entry);
    QCOMPARE(refreshTriggered, 1); // Check that refresh was triggered

    QCOMPARE(mRepo.rebaseOngoing(), true);
    QCOMPARE(rebaseFinished, 0);
    QCOMPARE(rebaseConflict, 1);


    // Check that buttons are visible
    QTest::qWait(100);
    QCOMPARE(repoView->mDetails->isRebaseContinueVisible(), true);
    QCOMPARE(repoView->mDetails->isRebaseAbortVisible(), true);

    // Resolve conflicts
    diff = mRepo.status(mRepo.index(), nullptr, false);
    QCOMPARE(diff.count(), 1);
    QCOMPARE(diff.patch(0).isConflicted(), true);
    QFile f(mRepo.workdir().filePath(diff.patch(0).name()));
    QCOMPARE(f.open(QIODevice::WriteOnly), true);
    QVERIFY(f.write("Test123") != -1); // just write something to resolve the conflict
    f.close();

    Test::refresh(repoView);

    // Staging the file
    auto cw = static_cast<ContentWidget*>(repoView->mDetails->mContent->currentWidget());
    auto dtw = dynamic_cast<DoubleTreeWidget*>(cw);
    QVERIFY(dtw);
    dtw->mDiffView->mFiles.at(0)->stageStateChanged(dtw->mDiffView->mFiles.at(0)->modelIndex(), git::Index::StagedState::Staged);


    repoView->mDetails->setCommitMessage("Test message");

    // Do commit before going on
    // So the user can commit between the rebase to split up the changes
    repoView->commit();

    refreshTriggered = 0;
    rebaseConflict = 0;
    repoView->continueRebase();
    QCOMPARE(refreshTriggered, 1);

    // Check that branch is based on "main" now
    branch = mRepo.lookupRef(rebaseBranchName);
    QVERIFY(branch.isValid());
    QList<Commit> parents = branch.annotatedCommit().commit().parents();
    QCOMPARE(parents.count(), 1);
    QCOMPARE(parents.at(0).id(), ac.commit().id());
    QCOMPARE(branch.annotatedCommit().commit().message(), "Test message"); // custom message was used instead of the original one

    // Check that rebase was really finished
    QCOMPARE(mRepo.rebaseOngoing(), false);

    // Check that buttons are visible
    QCOMPARE(repoView->mDetails->isRebaseContinueVisible(), false);
    QCOMPARE(repoView->mDetails->isRebaseAbortVisible(), false);

    // Check call counters
    QCOMPARE(rebaseFinished, 1);
    QCOMPARE(rebaseAboutToRebase, 1);
    QCOMPARE(rebaseCommitSuccess, 1);
    QCOMPARE(rebaseConflict, 0);
}

TEST_MAIN(TestRebase)

#include "rebase.moc"