media/mtransport/test/rlogringbuffer_unittest.cpp

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