tests: Added tests for Helper::setElement()

cieslarmichal · Nov 21, 2023 · b5dd175 · b5dd175
1 parent 71d85f9
commit b5dd175
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/src/modules/helper/HelperTest.cpp b/src/modules/helper/HelperTest.cpp
@@ -4,6 +4,7 @@
 #include <regex>
 #include <stdexcept>
 #include <unordered_map>
+#include <vector>
 
 #include "gtest/gtest.h"
 
@@ -18,6 +19,24 @@ class HelperTest : public Test
 namespace
 {
 
+TEST_F(HelperTest, SetElement)
+{
+    std::set<char> chars{'a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i', 'j', 'l', 'm'};
+    std::vector<char> randomChars;
+    for (int i = 0; i < 30; ++i)
+    {
+        randomChars.push_back(Helper::setElement(chars));
+    }
+    for (auto character : randomChars)
+    {
+        ASSERT_TRUE(chars.find(character) != chars.end());
+    }
+}
+TEST_F(HelperTest, SetElementEmptyData)
+{
+    std::set<char> chars{};
+    ASSERT_THROW(Helper::setElement<char>(chars), std::invalid_argument);
+}
 TEST_F(HelperTest, ReplaceSymbolWithNumber)
 {
     std::string input = "123#456!";