rlogconnector_unittest.cpp 7.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264
  1. /* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
  2. /* This Source Code Form is subject to the terms of the Mozilla Public
  3. * License, v. 2.0. If a copy of the MPL was not distributed with this file,
  4. * You can obtain one at http://mozilla.org/MPL/2.0/. */
  5. /* Original author: bcampen@mozilla.com */
  6. #include "rlogconnector.h"
  7. extern "C" {
  8. #include "registry.h"
  9. #include "r_log.h"
  10. }
  11. #define GTEST_HAS_RTTI 0
  12. #include "gtest/gtest.h"
  13. #include "gtest_utils.h"
  14. #include <deque>
  15. #include <string>
  16. #include <vector>
  17. using mozilla::RLogConnector;
  18. int NR_LOG_TEST = 0;
  19. class RLogConnectorTest : public ::testing::Test {
  20. public:
  21. RLogConnectorTest() {
  22. Init();
  23. }
  24. ~RLogConnectorTest() {
  25. Free();
  26. }
  27. static void SetUpTestCase() {
  28. NR_reg_init(NR_REG_MODE_LOCAL);
  29. r_log_init();
  30. /* Would be nice to be able to unregister in the fixture */
  31. const char* facility = "rlogconnector_test";
  32. r_log_register(const_cast<char*>(facility), &NR_LOG_TEST);
  33. }
  34. void Init() {
  35. RLogConnector::CreateInstance();
  36. }
  37. void Free() {
  38. RLogConnector::DestroyInstance();
  39. }
  40. void ReInit() {
  41. Free();
  42. Init();
  43. }
  44. };
  45. TEST_F(RLogConnectorTest, TestGetFree) {
  46. RLogConnector* instance = RLogConnector::GetInstance();
  47. ASSERT_NE(nullptr, instance);
  48. }
  49. TEST_F(RLogConnectorTest, TestFilterEmpty) {
  50. std::deque<std::string> logs;
  51. RLogConnector::GetInstance()->GetAny(0, &logs);
  52. ASSERT_EQ(0U, logs.size());
  53. }
  54. TEST_F(RLogConnectorTest, TestBasicFilter) {
  55. r_log(NR_LOG_TEST, LOG_INFO, "Test");
  56. std::deque<std::string> logs;
  57. RLogConnector::GetInstance()->Filter("Test", 0, &logs);
  58. ASSERT_EQ(1U, logs.size());
  59. }
  60. TEST_F(RLogConnectorTest, TestBasicFilterContent) {
  61. r_log(NR_LOG_TEST, LOG_INFO, "Test");
  62. std::deque<std::string> logs;
  63. RLogConnector::GetInstance()->Filter("Test", 0, &logs);
  64. ASSERT_EQ("Test", logs.back());
  65. }
  66. TEST_F(RLogConnectorTest, TestFilterAnyFrontMatch) {
  67. r_log(NR_LOG_TEST, LOG_INFO, "Test");
  68. std::vector<std::string> substrings;
  69. substrings.push_back("foo");
  70. substrings.push_back("Test");
  71. std::deque<std::string> logs;
  72. RLogConnector::GetInstance()->FilterAny(substrings, 0, &logs);
  73. ASSERT_EQ("Test", logs.back());
  74. }
  75. TEST_F(RLogConnectorTest, TestFilterAnyBackMatch) {
  76. r_log(NR_LOG_TEST, LOG_INFO, "Test");
  77. std::vector<std::string> substrings;
  78. substrings.push_back("Test");
  79. substrings.push_back("foo");
  80. std::deque<std::string> logs;
  81. RLogConnector::GetInstance()->FilterAny(substrings, 0, &logs);
  82. ASSERT_EQ("Test", logs.back());
  83. }
  84. TEST_F(RLogConnectorTest, TestFilterAnyBothMatch) {
  85. r_log(NR_LOG_TEST, LOG_INFO, "Test");
  86. std::vector<std::string> substrings;
  87. substrings.push_back("Tes");
  88. substrings.push_back("est");
  89. std::deque<std::string> logs;
  90. RLogConnector::GetInstance()->FilterAny(substrings, 0, &logs);
  91. ASSERT_EQ("Test", logs.back());
  92. }
  93. TEST_F(RLogConnectorTest, TestFilterAnyNeitherMatch) {
  94. r_log(NR_LOG_TEST, LOG_INFO, "Test");
  95. std::vector<std::string> substrings;
  96. substrings.push_back("tes");
  97. substrings.push_back("esT");
  98. std::deque<std::string> logs;
  99. RLogConnector::GetInstance()->FilterAny(substrings, 0, &logs);
  100. ASSERT_EQ(0U, logs.size());
  101. }
  102. TEST_F(RLogConnectorTest, TestAllMatch) {
  103. r_log(NR_LOG_TEST, LOG_INFO, "Test1");
  104. r_log(NR_LOG_TEST, LOG_INFO, "Test2");
  105. std::deque<std::string> logs;
  106. RLogConnector::GetInstance()->GetAny(0, &logs);
  107. ASSERT_EQ(2U, logs.size());
  108. }
  109. TEST_F(RLogConnectorTest, TestOrder) {
  110. r_log(NR_LOG_TEST, LOG_INFO, "Test1");
  111. r_log(NR_LOG_TEST, LOG_INFO, "Test2");
  112. std::deque<std::string> logs;
  113. RLogConnector::GetInstance()->GetAny(0, &logs);
  114. ASSERT_EQ("Test2", logs.back());
  115. ASSERT_EQ("Test1", logs.front());
  116. }
  117. TEST_F(RLogConnectorTest, TestNoMatch) {
  118. r_log(NR_LOG_TEST, LOG_INFO, "Test1");
  119. r_log(NR_LOG_TEST, LOG_INFO, "Test2");
  120. std::deque<std::string> logs;
  121. RLogConnector::GetInstance()->Filter("foo", 0, &logs);
  122. ASSERT_EQ(0U, logs.size());
  123. }
  124. TEST_F(RLogConnectorTest, TestSubstringFilter) {
  125. r_log(NR_LOG_TEST, LOG_INFO, "Test1");
  126. r_log(NR_LOG_TEST, LOG_INFO, "Test2");
  127. std::deque<std::string> logs;
  128. RLogConnector::GetInstance()->Filter("t1", 0, &logs);
  129. ASSERT_EQ(1U, logs.size());
  130. ASSERT_EQ("Test1", logs.back());
  131. }
  132. TEST_F(RLogConnectorTest, TestFilterLimit) {
  133. r_log(NR_LOG_TEST, LOG_INFO, "Test1");
  134. r_log(NR_LOG_TEST, LOG_INFO, "Test2");
  135. r_log(NR_LOG_TEST, LOG_INFO, "Test3");
  136. r_log(NR_LOG_TEST, LOG_INFO, "Test4");
  137. r_log(NR_LOG_TEST, LOG_INFO, "Test5");
  138. r_log(NR_LOG_TEST, LOG_INFO, "Test6");
  139. std::deque<std::string> logs;
  140. RLogConnector::GetInstance()->Filter("Test", 2, &logs);
  141. ASSERT_EQ(2U, logs.size());
  142. ASSERT_EQ("Test6", logs.back());
  143. ASSERT_EQ("Test5", logs.front());
  144. }
  145. TEST_F(RLogConnectorTest, TestFilterAnyLimit) {
  146. r_log(NR_LOG_TEST, LOG_INFO, "TestOne");
  147. r_log(NR_LOG_TEST, LOG_INFO, "TestTwo");
  148. r_log(NR_LOG_TEST, LOG_INFO, "TestThree");
  149. r_log(NR_LOG_TEST, LOG_INFO, "TestFour");
  150. r_log(NR_LOG_TEST, LOG_INFO, "TestFive");
  151. r_log(NR_LOG_TEST, LOG_INFO, "TestSix");
  152. std::vector<std::string> substrings;
  153. // Matches Two, Three, Four, and Six
  154. substrings.push_back("tT");
  155. substrings.push_back("o");
  156. substrings.push_back("r");
  157. substrings.push_back("S");
  158. std::deque<std::string> logs;
  159. RLogConnector::GetInstance()->FilterAny(substrings, 2, &logs);
  160. ASSERT_EQ(2U, logs.size());
  161. ASSERT_EQ("TestSix", logs.back());
  162. ASSERT_EQ("TestFour", logs.front());
  163. }
  164. TEST_F(RLogConnectorTest, TestLimit) {
  165. RLogConnector::GetInstance()->SetLogLimit(3);
  166. r_log(NR_LOG_TEST, LOG_INFO, "Test1");
  167. r_log(NR_LOG_TEST, LOG_INFO, "Test2");
  168. r_log(NR_LOG_TEST, LOG_INFO, "Test3");
  169. r_log(NR_LOG_TEST, LOG_INFO, "Test4");
  170. r_log(NR_LOG_TEST, LOG_INFO, "Test5");
  171. r_log(NR_LOG_TEST, LOG_INFO, "Test6");
  172. std::deque<std::string> logs;
  173. RLogConnector::GetInstance()->GetAny(0, &logs);
  174. ASSERT_EQ(3U, logs.size());
  175. ASSERT_EQ("Test6", logs.back());
  176. ASSERT_EQ("Test4", logs.front());
  177. }
  178. TEST_F(RLogConnectorTest, TestLimitBulkDiscard) {
  179. r_log(NR_LOG_TEST, LOG_INFO, "Test1");
  180. r_log(NR_LOG_TEST, LOG_INFO, "Test2");
  181. r_log(NR_LOG_TEST, LOG_INFO, "Test3");
  182. r_log(NR_LOG_TEST, LOG_INFO, "Test4");
  183. r_log(NR_LOG_TEST, LOG_INFO, "Test5");
  184. r_log(NR_LOG_TEST, LOG_INFO, "Test6");
  185. RLogConnector::GetInstance()->SetLogLimit(3);
  186. std::deque<std::string> logs;
  187. RLogConnector::GetInstance()->GetAny(0, &logs);
  188. ASSERT_EQ(3U, logs.size());
  189. ASSERT_EQ("Test6", logs.back());
  190. ASSERT_EQ("Test4", logs.front());
  191. }
  192. TEST_F(RLogConnectorTest, TestIncreaseLimit) {
  193. RLogConnector::GetInstance()->SetLogLimit(3);
  194. r_log(NR_LOG_TEST, LOG_INFO, "Test1");
  195. r_log(NR_LOG_TEST, LOG_INFO, "Test2");
  196. r_log(NR_LOG_TEST, LOG_INFO, "Test3");
  197. r_log(NR_LOG_TEST, LOG_INFO, "Test4");
  198. r_log(NR_LOG_TEST, LOG_INFO, "Test5");
  199. r_log(NR_LOG_TEST, LOG_INFO, "Test6");
  200. RLogConnector::GetInstance()->SetLogLimit(300);
  201. std::deque<std::string> logs;
  202. RLogConnector::GetInstance()->GetAny(0, &logs);
  203. ASSERT_EQ(3U, logs.size());
  204. ASSERT_EQ("Test6", logs.back());
  205. ASSERT_EQ("Test4", logs.front());
  206. }
  207. TEST_F(RLogConnectorTest, TestClear) {
  208. r_log(NR_LOG_TEST, LOG_INFO, "Test1");
  209. r_log(NR_LOG_TEST, LOG_INFO, "Test2");
  210. r_log(NR_LOG_TEST, LOG_INFO, "Test3");
  211. r_log(NR_LOG_TEST, LOG_INFO, "Test4");
  212. r_log(NR_LOG_TEST, LOG_INFO, "Test5");
  213. r_log(NR_LOG_TEST, LOG_INFO, "Test6");
  214. RLogConnector::GetInstance()->SetLogLimit(0);
  215. RLogConnector::GetInstance()->SetLogLimit(4096);
  216. std::deque<std::string> logs;
  217. RLogConnector::GetInstance()->GetAny(0, &logs);
  218. ASSERT_EQ(0U, logs.size());
  219. }
  220. TEST_F(RLogConnectorTest, TestReInit) {
  221. r_log(NR_LOG_TEST, LOG_INFO, "Test1");
  222. r_log(NR_LOG_TEST, LOG_INFO, "Test2");
  223. r_log(NR_LOG_TEST, LOG_INFO, "Test3");
  224. r_log(NR_LOG_TEST, LOG_INFO, "Test4");
  225. r_log(NR_LOG_TEST, LOG_INFO, "Test5");
  226. r_log(NR_LOG_TEST, LOG_INFO, "Test6");
  227. ReInit();
  228. std::deque<std::string> logs;
  229. RLogConnector::GetInstance()->GetAny(0, &logs);
  230. ASSERT_EQ(0U, logs.size());
  231. }