/****************************************************************************
**
** Copyright (C) 2018 The Qt Company Ltd.
** Contact: https://www.qt.io/licensing/
**
** This file is part of Qt Creator.
**
** Commercial License Usage
** Licensees holding valid commercial Qt licenses may use this file in
** accordance with the commercial license agreement provided with the
** Software or, alternatively, in accordance with the terms contained in
** a written agreement between you and The Qt Company. For licensing terms
** and conditions see https://www.qt.io/terms-conditions. For further
** information use the contact form at https://www.qt.io/contact-us.
**
** GNU General Public License Usage
** Alternatively, this file may be used under the terms of the GNU
** General Public License version 3 as published by the Free Software
** Foundation with exceptions as appearing in the file LICENSE.GPL3-EXCEPT
** included in the packaging of this file. Please review the following
** information to ensure the GNU General Public License requirements will
** be met: https://www.gnu.org/licenses/gpl-3.0.html.
**
****************************************************************************/

#include "googletest.h"

#include <symbolindexertaskqueue.h>

namespace {

using ClangBackEnd::SymbolIndexerTask;
using ClangBackEnd::FilePathId;

MATCHER_P2(IsTask, filePathId, projectPartId,
          std::string(negation ? "is't" : "is")
          + PrintToString(SymbolIndexerTask(filePathId, projectPartId, []{})))
{
    const SymbolIndexerTask &task = arg;

    return task.filePathId == filePathId && task.projectPartId == projectPartId;
}

class SymbolIndexerTaskQueue : public testing::Test
{
protected:
    std::size_t projectPartId(const Utils::SmallString &projectPartId)
    {
        return queue.projectPartNumberId(projectPartId);
    }
protected:
    ClangBackEnd::SymbolIndexerTaskQueue queue;
};

TEST_F(SymbolIndexerTaskQueue, AddTasks)
{
    queue.addOrUpdateTasks({{{1, 2}, projectPartId("foo"), [] {}},
                            {{1, 4}, projectPartId("foo"), [] {}}});

    queue.addOrUpdateTasks({{{1, 1}, projectPartId("foo"), [] {}},
                            {{1, 3}, projectPartId("foo"), [] {}},
                            {{1, 5}, projectPartId("foo"), [] {}}});

    ASSERT_THAT(queue.tasks(),
                ElementsAre(IsTask(FilePathId{1, 1}, projectPartId("foo")),
                            IsTask(FilePathId{1, 2}, projectPartId("foo")),
                            IsTask(FilePathId{1, 3}, projectPartId("foo")),
                            IsTask(FilePathId{1, 4}, projectPartId("foo")),
                            IsTask(FilePathId{1, 5}, projectPartId("foo"))));
}

TEST_F(SymbolIndexerTaskQueue, ReplaceTask)
{
    queue.addOrUpdateTasks({{{1, 1}, projectPartId("foo"), [] {}},
                            {{1, 3}, projectPartId("foo"), [] {}},
                            {{1, 5}, projectPartId("foo"), [] {}}});

    queue.addOrUpdateTasks({{{1, 2}, projectPartId("foo"), [] {}},
                            {{1, 3}, projectPartId("foo"), [] {}}});

    ASSERT_THAT(queue.tasks(),
                ElementsAre(IsTask(FilePathId{1, 1}, projectPartId("foo")),
                            IsTask(FilePathId{1, 2}, projectPartId("foo")),
                            IsTask(FilePathId{1, 3}, projectPartId("foo")),
                            IsTask(FilePathId{1, 5}, projectPartId("foo"))));
}

TEST_F(SymbolIndexerTaskQueue, AddTaskWithDifferentProjectId)
{
    queue.addOrUpdateTasks({{{1, 1}, projectPartId("foo"), [] {}},
                            {{1, 3}, projectPartId("foo"), [] {}},
                            {{1, 5}, projectPartId("foo"), [] {}}});

    queue.addOrUpdateTasks({{{1, 2}, projectPartId("bar"), [] {}},
                            {{1, 3}, projectPartId("bar"), [] {}}});

    ASSERT_THAT(queue.tasks(),
                ElementsAre(IsTask(FilePathId{1, 1}, projectPartId("foo")),
                            IsTask(FilePathId{1, 2}, projectPartId("bar")),
                            IsTask(FilePathId{1, 3}, projectPartId("foo")),
                            IsTask(FilePathId{1, 3}, projectPartId("bar")),
                            IsTask(FilePathId{1, 5}, projectPartId("foo"))));
}

TEST_F(SymbolIndexerTaskQueue, RemoveTaskByProjectParts)
{
    queue.addOrUpdateTasks({{{1, 1}, projectPartId("yi"), [] {}},
                            {{1, 3}, projectPartId("yi"), [] {}},
                            {{1, 5}, projectPartId("yi"), [] {}}});
    queue.addOrUpdateTasks({{{1, 2}, projectPartId("er"), [] {}},
                            {{1, 3}, projectPartId("er"), [] {}}});
    queue.addOrUpdateTasks({{{1, 2}, projectPartId("san"), [] {}},
                            {{1, 3}, projectPartId("san"), [] {}}});
    queue.addOrUpdateTasks({{{1, 2}, projectPartId("se"), [] {}},
                            {{1, 3}, projectPartId("se"), [] {}}});

    queue.removeTasks({"er", "san"});

    ASSERT_THAT(queue.tasks(),
                ElementsAre(IsTask(FilePathId{1, 1}, projectPartId("yi")),
                            IsTask(FilePathId{1, 2}, projectPartId("se")),
                            IsTask(FilePathId{1, 3}, projectPartId("yi")),
                            IsTask(FilePathId{1, 3}, projectPartId("se")),
                            IsTask(FilePathId{1, 5}, projectPartId("yi"))));
}

TEST_F(SymbolIndexerTaskQueue, GetProjectPartIdIfEmpty)
{
    auto id = queue.projectPartNumberId("foo");

    ASSERT_THAT(id , 0);
}

TEST_F(SymbolIndexerTaskQueue, GetProjectPartIdIfNotExists)
{
    queue.projectPartNumberId("foo");

    auto id = queue.projectPartNumberId("bar");

    ASSERT_THAT(id , 1);
}

TEST_F(SymbolIndexerTaskQueue, GetProjectPartIdIfExists)
{
    queue.projectPartNumberId("foo");
    queue.projectPartNumberId("bar");

    auto id = queue.projectPartNumberId("foo");

    ASSERT_THAT(id , 0);
}

TEST_F(SymbolIndexerTaskQueue, GetProjectPartIds)
{
    queue.projectPartNumberIds({"yi", "er", "san"});

    auto ids = queue.projectPartNumberIds({"yi", "se", "san"});

    ASSERT_THAT(ids , ElementsAre(0, 2, 3));
}
}