1 files changed, 182 insertions, 0 deletions
diff --git a/tests/unit/codemodelbackend/unittest/translationunitstest.cpp b/tests/unit/codemodelbackend/unittest/translationunitstest.cpp
new file mode 100644
index 0000000000..533b422c3d
--- /dev/null
+++ b/tests/unit/codemodelbackend/unittest/translationunitstest.cpp
@@ -0,0 +1,182 @@
+/****************************************************************************
+**
+** Copyright (C) 2015 Digia Plc and/or its subsidiary(-ies).
+** Contact: http://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 Digia.  For licensing terms and
+** conditions see http://www.qt.io/licensing.  For further information
+** use the contact form at http://www.qt.io/contact-us.
+**
+** GNU Lesser General Public License Usage
+** Alternatively, this file may be used under the terms of the GNU Lesser
+** General Public License version 2.1 or version 3 as published by the Free
+** Software Foundation and appearing in the file LICENSE.LGPLv21 and
+** LICENSE.LGPLv3 included in the packaging of this file.  Please review the
+** following information to ensure the GNU Lesser General Public License
+** requirements will be met: https://www.gnu.org/licenses/lgpl.html and
+** http://www.gnu.org/licenses/old-licenses/lgpl-2.1.html.
+**
+** In addition, as a special exception, Digia gives you certain additional
+** rights.  These rights are described in the Digia Qt LGPL Exception
+** version 1.1, included in the file LGPL_EXCEPTION.txt in this package.
+**
+****************************************************************************/
+
+#include "gtest/gtest.h"
+#include "gmock/gmock-matchers.h"
+#include "gmock/gmock-generated-matchers.h"
+#include "gtest-qt-printing.h"
+
+#include <clang-c/Index.h>
+
+#include <translationunit.h>
+#include <unsavedfiles.h>
+#include <utf8string.h>
+#include <projectpart.h>
+#include <translationunits.h>
+#include <filecontainer.h>
+#include <projectpartcontainer.h>
+#include <projects.h>
+#include <translationunitdoesnotexistexception.h>
+#include <translationunitisnullexception.h>
+#include <translationunitfilenotexitexception.h>
+#include <projectpartsdonotexistexception.h>
+
+using CodeModelBackEnd::TranslationUnit;
+using CodeModelBackEnd::UnsavedFiles;
+using CodeModelBackEnd::ProjectPart;
+
+using testing::IsNull;
+using testing::NotNull;
+using testing::Gt;
+using testing::Not;
+using testing::Contains;
+
+namespace {
+
+using ::testing::PrintToString;
+
+MATCHER_P2(IsTranslationUnit, filePath, projectPartId,
+           std::string(negation ? "isn't" : "is") + " translation unit with file path "
+           + PrintToString(filePath) + " and project " + PrintToString(projectPartId)
+           )
+{
+    if (arg.filePath() != filePath) {
+        *result_listener << "file path is " + PrintToString(arg.filePath()) + " and not " +  PrintToString(filePath);
+        return false;
+    }
+
+    if (arg.projectPartId() != projectPartId) {
+        *result_listener << "file path is " + PrintToString(arg.projectPartId()) + " and not " +  PrintToString(projectPartId);
+        return false;
+    }
+
+    return true;
+}
+
+class TranslationUnits : public ::testing::Test
+{
+protected:
+    void SetUp() override;
+
+    CodeModelBackEnd::ProjectParts projects;
+    CodeModelBackEnd::UnsavedFiles unsavedFiles;
+    CodeModelBackEnd::TranslationUnits translationUnits = CodeModelBackEnd::TranslationUnits(projects, unsavedFiles);
+    const Utf8String filePath = Utf8StringLiteral(TESTDATA_DIR"/complete_testfile_1.cpp");
+    const Utf8String projectPartId = Utf8StringLiteral("/path/to/projectfile");
+
+};
+
+void TranslationUnits::SetUp()
+{
+    projects.createOrUpdate({CodeModelBackEnd::ProjectPartContainer(projectPartId)});
+}
+
+
+TEST_F(TranslationUnits, ThrowForGettingWithWrongFilePath)
+{
+    ASSERT_THROW(translationUnits.translationUnit(Utf8StringLiteral("foo.cpp"), projectPartId),
+                 CodeModelBackEnd::TranslationUnitDoesNotExistException);
+
+}
+
+TEST_F(TranslationUnits, ThrowForGettingWithWrongProjectPartFilePath)
+{
+    ASSERT_THROW(translationUnits.translationUnit(filePath, Utf8StringLiteral("foo.pro")),
+                 CodeModelBackEnd::ProjectPartDoNotExistException);
+
+}
+
+TEST_F(TranslationUnits, ThrowForAddingNonExistingFile)
+{
+    CodeModelBackEnd::FileContainer fileContainer(Utf8StringLiteral("foo.cpp"), projectPartId);
+
+    ASSERT_THROW(translationUnits.createOrUpdate({fileContainer}),
+                 CodeModelBackEnd::TranslationUnitFileNotExitsException);
+}
+
+TEST_F(TranslationUnits, DoNotThrowForAddingNonExistingFileWithUnsavedContent)
+{
+    CodeModelBackEnd::FileContainer fileContainer(Utf8StringLiteral("foo.cpp"), projectPartId, Utf8String(), true);
+
+    ASSERT_NO_THROW(translationUnits.createOrUpdate({fileContainer}));
+}
+
+TEST_F(TranslationUnits, Add)
+{
+    CodeModelBackEnd::FileContainer fileContainer(filePath, projectPartId);
+
+    translationUnits.createOrUpdate({fileContainer});
+
+    ASSERT_THAT(translationUnits.translationUnit(filePath, projectPartId),
+                IsTranslationUnit(filePath, projectPartId));
+}
+
+TEST_F(TranslationUnits, ThrowForRemovingWithWrongFilePath)
+{
+    CodeModelBackEnd::FileContainer fileContainer(Utf8StringLiteral("foo.cpp"), projectPartId);
+
+    ASSERT_THROW(translationUnits.remove({fileContainer}),
+                 CodeModelBackEnd::TranslationUnitDoesNotExistException);
+}
+
+TEST_F(TranslationUnits, ThrowForRemovingWithWrongProjectPartFilePath)
+{
+    CodeModelBackEnd::FileContainer fileContainer(filePath, Utf8StringLiteral("foo.pro"));
+
+    ASSERT_THROW(translationUnits.remove({fileContainer}),
+                 CodeModelBackEnd::ProjectPartDoNotExistException);
+}
+
+TEST_F(TranslationUnits, Remove)
+{
+    CodeModelBackEnd::FileContainer fileContainer(filePath, projectPartId);
+    translationUnits.createOrUpdate({fileContainer});
+
+    translationUnits.remove({fileContainer});
+
+    ASSERT_THROW(translationUnits.translationUnit(filePath, projectPartId),
+                 CodeModelBackEnd::TranslationUnitDoesNotExistException);
+}
+
+TEST_F(TranslationUnits, RemoveAllValidIfExceptionIsThrown)
+{
+    CodeModelBackEnd::FileContainer fileContainer(filePath, projectPartId);
+    translationUnits.createOrUpdate({fileContainer});
+
+    ASSERT_THROW(translationUnits.remove({CodeModelBackEnd::FileContainer(Utf8StringLiteral("dontextist.pro"), projectPartId), fileContainer}),
+                 CodeModelBackEnd::TranslationUnitDoesNotExistException);
+
+    ASSERT_THAT(translationUnits.translationUnits(),
+                Not(Contains(TranslationUnit(filePath, unsavedFiles, projects.project(projectPartId)))));
+}
+
+}
+
+