From 1b302b750cca4b6f88cb4794b97739acc140480c Mon Sep 17 00:00:00 2001
From: scrawl <scrawl@baseoftrash.de>
Date: Fri, 9 Jan 2015 20:52:31 +0100
Subject: [PATCH] UnitTests: add tests for KeywordSearch conflict resolution

---
 apps/openmw_test_suite/CMakeLists.txt         |  1 +
 .../mwdialogue/test_keywordsearch.cpp         | 48 +++++++++++++++++++
 2 files changed, 49 insertions(+)
 create mode 100644 apps/openmw_test_suite/mwdialogue/test_keywordsearch.cpp

diff --git a/apps/openmw_test_suite/CMakeLists.txt b/apps/openmw_test_suite/CMakeLists.txt
index 44354eac8..2ffb7ffa0 100644
--- a/apps/openmw_test_suite/CMakeLists.txt
+++ b/apps/openmw_test_suite/CMakeLists.txt
@@ -5,6 +5,7 @@ if (GTEST_FOUND)
 
     file(GLOB UNITTEST_SRC_FILES
         components/misc/test_*.cpp
+        mwdialogue/test_*.cpp
     )
 
     source_group(apps\\openmw_test_suite FILES openmw_test_suite.cpp ${UNITTEST_SRC_FILES})
diff --git a/apps/openmw_test_suite/mwdialogue/test_keywordsearch.cpp b/apps/openmw_test_suite/mwdialogue/test_keywordsearch.cpp
new file mode 100644
index 000000000..c8ea9d01c
--- /dev/null
+++ b/apps/openmw_test_suite/mwdialogue/test_keywordsearch.cpp
@@ -0,0 +1,48 @@
+#include <gtest/gtest.h>
+#include "apps/openmw/mwdialogue/keywordsearch.hpp"
+
+struct KeywordSearchTest : public ::testing::Test
+{
+  protected:
+    virtual void SetUp()
+    {
+    }
+
+    virtual void TearDown()
+    {
+    }
+};
+
+TEST_F(KeywordSearchTest, keyword_test_conflict_resolution)
+{
+    // test to make sure the longest keyword in a chain of conflicting keywords gets chosen
+    MWDialogue::KeywordSearch<std::string, int> search;
+    search.seed("foo bar", 0);
+    search.seed("bar lock", 0);
+    search.seed("lock switch", 0);
+
+    std::string text = "foo bar lock switch";
+
+    std::vector<MWDialogue::KeywordSearch<std::string, int>::Match> matches;
+    search.highlightKeywords(text.begin(), text.end(), matches);
+
+    // Should contain: "foo bar", "lock switch"
+    ASSERT_TRUE (matches.size() == 2);
+    ASSERT_TRUE (std::string(matches.front().mBeg, matches.front().mEnd) == "foo bar");
+    ASSERT_TRUE (std::string(matches.rbegin()->mBeg, matches.rbegin()->mEnd) == "lock switch");
+}
+
+TEST_F(KeywordSearchTest, keyword_test_conflict_resolution2)
+{
+    MWDialogue::KeywordSearch<std::string, int> search;
+    search.seed("the dwemer", 0);
+    search.seed("dwemer language", 0);
+
+    std::string text = "the dwemer language";
+
+    std::vector<MWDialogue::KeywordSearch<std::string, int>::Match> matches;
+    search.highlightKeywords(text.begin(), text.end(), matches);
+
+    ASSERT_TRUE (matches.size() == 1);
+    ASSERT_TRUE (std::string(matches.front().mBeg, matches.front().mEnd) == "dwemer language");
+}