michael@0: /* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
michael@0: /* vim: set ts=2 et sw=2 tw=80: */
michael@0: /* This Source Code Form is subject to the terms of the Mozilla Public
michael@0:  * License, v. 2.0. If a copy of the MPL was not distributed with this file,
michael@0:  * You can obtain one at http://mozilla.org/MPL/2.0/. */
michael@0: 
michael@0: /* Original author: bcampen@mozilla.com */
michael@0: 
michael@0: #include "rlogringbuffer.h"
michael@0: 
michael@0: extern "C" {
michael@0: #include "registry.h"
michael@0: #include "r_log.h"
michael@0: }
michael@0: 
michael@0: #define GTEST_HAS_RTTI 0
michael@0: #include "gtest/gtest.h"
michael@0: #include "gtest_utils.h"
michael@0: 
michael@0: #include <deque>
michael@0: #include <string>
michael@0: #include <vector>
michael@0: 
michael@0: using mozilla::RLogRingBuffer;
michael@0: 
michael@0: int NR_LOG_TEST = 0;
michael@0: 
michael@0: class RLogRingBufferTest : public ::testing::Test {
michael@0:   public:
michael@0:     RLogRingBufferTest() {
michael@0:       Init();
michael@0:     }
michael@0: 
michael@0:     ~RLogRingBufferTest() {
michael@0:       Free();
michael@0:     }
michael@0: 
michael@0:     void Init() {
michael@0:       RLogRingBuffer::CreateInstance();
michael@0:     }
michael@0: 
michael@0:     void Free() {
michael@0:       RLogRingBuffer::DestroyInstance();
michael@0:     }
michael@0: 
michael@0:     void ReInit() {
michael@0:       Free();
michael@0:       Init();
michael@0:     }
michael@0: };
michael@0: 
michael@0: TEST_F(RLogRingBufferTest, TestGetFree) {
michael@0:   RLogRingBuffer* instance = RLogRingBuffer::GetInstance();
michael@0:   ASSERT_NE(nullptr, instance);
michael@0: }
michael@0: 
michael@0: TEST_F(RLogRingBufferTest, TestFilterEmpty) {
michael@0:   std::deque<std::string> logs;
michael@0:   RLogRingBuffer::GetInstance()->GetAny(0, &logs);
michael@0:   ASSERT_EQ(0U, logs.size());
michael@0: }
michael@0: 
michael@0: TEST_F(RLogRingBufferTest, TestBasicFilter) {
michael@0:   r_log(NR_LOG_TEST, LOG_INFO, "Test");
michael@0:   std::deque<std::string> logs;
michael@0:   RLogRingBuffer::GetInstance()->Filter("Test", 0, &logs);
michael@0:   ASSERT_EQ(1U, logs.size());
michael@0: }
michael@0: 
michael@0: TEST_F(RLogRingBufferTest, TestBasicFilterContent) {
michael@0:   r_log(NR_LOG_TEST, LOG_INFO, "Test");
michael@0:   std::deque<std::string> logs;
michael@0:   RLogRingBuffer::GetInstance()->Filter("Test", 0, &logs);
michael@0:   ASSERT_EQ("Test", logs.back());
michael@0: }
michael@0: 
michael@0: TEST_F(RLogRingBufferTest, TestFilterAnyFrontMatch) {
michael@0:   r_log(NR_LOG_TEST, LOG_INFO, "Test");
michael@0:   std::vector<std::string> substrings;
michael@0:   substrings.push_back("foo");
michael@0:   substrings.push_back("Test");
michael@0:   std::deque<std::string> logs;
michael@0:   RLogRingBuffer::GetInstance()->FilterAny(substrings, 0, &logs);
michael@0:   ASSERT_EQ("Test", logs.back());
michael@0: }
michael@0: 
michael@0: TEST_F(RLogRingBufferTest, TestFilterAnyBackMatch) {
michael@0:   r_log(NR_LOG_TEST, LOG_INFO, "Test");
michael@0:   std::vector<std::string> substrings;
michael@0:   substrings.push_back("Test");
michael@0:   substrings.push_back("foo");
michael@0:   std::deque<std::string> logs;
michael@0:   RLogRingBuffer::GetInstance()->FilterAny(substrings, 0, &logs);
michael@0:   ASSERT_EQ("Test", logs.back());
michael@0: }
michael@0: 
michael@0: TEST_F(RLogRingBufferTest, TestFilterAnyBothMatch) {
michael@0:   r_log(NR_LOG_TEST, LOG_INFO, "Test");
michael@0:   std::vector<std::string> substrings;
michael@0:   substrings.push_back("Tes");
michael@0:   substrings.push_back("est");
michael@0:   std::deque<std::string> logs;
michael@0:   RLogRingBuffer::GetInstance()->FilterAny(substrings, 0, &logs);
michael@0:   ASSERT_EQ("Test", logs.back());
michael@0: }
michael@0: 
michael@0: TEST_F(RLogRingBufferTest, TestFilterAnyNeitherMatch) {
michael@0:   r_log(NR_LOG_TEST, LOG_INFO, "Test");
michael@0:   std::vector<std::string> substrings;
michael@0:   substrings.push_back("tes");
michael@0:   substrings.push_back("esT");
michael@0:   std::deque<std::string> logs;
michael@0:   RLogRingBuffer::GetInstance()->FilterAny(substrings, 0, &logs);
michael@0:   ASSERT_EQ(0U, logs.size());
michael@0: }
michael@0: 
michael@0: TEST_F(RLogRingBufferTest, TestAllMatch) {
michael@0:   r_log(NR_LOG_TEST, LOG_INFO, "Test1");
michael@0:   r_log(NR_LOG_TEST, LOG_INFO, "Test2");
michael@0:   std::deque<std::string> logs;
michael@0:   RLogRingBuffer::GetInstance()->GetAny(0, &logs);
michael@0:   ASSERT_EQ(2U, logs.size());
michael@0: }
michael@0: 
michael@0: TEST_F(RLogRingBufferTest, TestOrder) {
michael@0:   r_log(NR_LOG_TEST, LOG_INFO, "Test1");
michael@0:   r_log(NR_LOG_TEST, LOG_INFO, "Test2");
michael@0:   std::deque<std::string> logs;
michael@0:   RLogRingBuffer::GetInstance()->GetAny(0, &logs);
michael@0:   ASSERT_EQ("Test2", logs.back());
michael@0:   ASSERT_EQ("Test1", logs.front());
michael@0: }
michael@0: 
michael@0: TEST_F(RLogRingBufferTest, TestNoMatch) {
michael@0:   r_log(NR_LOG_TEST, LOG_INFO, "Test1");
michael@0:   r_log(NR_LOG_TEST, LOG_INFO, "Test2");
michael@0:   std::deque<std::string> logs;
michael@0:   RLogRingBuffer::GetInstance()->Filter("foo", 0, &logs);
michael@0:   ASSERT_EQ(0U, logs.size());
michael@0: }
michael@0: 
michael@0: TEST_F(RLogRingBufferTest, TestSubstringFilter) {
michael@0:   r_log(NR_LOG_TEST, LOG_INFO, "Test1");
michael@0:   r_log(NR_LOG_TEST, LOG_INFO, "Test2");
michael@0:   std::deque<std::string> logs;
michael@0:   RLogRingBuffer::GetInstance()->Filter("t1", 0, &logs);
michael@0:   ASSERT_EQ(1U, logs.size());
michael@0:   ASSERT_EQ("Test1", logs.back());
michael@0: }
michael@0: 
michael@0: TEST_F(RLogRingBufferTest, TestFilterLimit) {
michael@0:   r_log(NR_LOG_TEST, LOG_INFO, "Test1");
michael@0:   r_log(NR_LOG_TEST, LOG_INFO, "Test2");
michael@0:   r_log(NR_LOG_TEST, LOG_INFO, "Test3");
michael@0:   r_log(NR_LOG_TEST, LOG_INFO, "Test4");
michael@0:   r_log(NR_LOG_TEST, LOG_INFO, "Test5");
michael@0:   r_log(NR_LOG_TEST, LOG_INFO, "Test6");
michael@0:   std::deque<std::string> logs;
michael@0:   RLogRingBuffer::GetInstance()->Filter("Test", 2, &logs);
michael@0:   ASSERT_EQ(2U, logs.size());
michael@0:   ASSERT_EQ("Test6", logs.back());
michael@0:   ASSERT_EQ("Test5", logs.front());
michael@0: }
michael@0: 
michael@0: TEST_F(RLogRingBufferTest, TestFilterAnyLimit) {
michael@0:   r_log(NR_LOG_TEST, LOG_INFO, "TestOne");
michael@0:   r_log(NR_LOG_TEST, LOG_INFO, "TestTwo");
michael@0:   r_log(NR_LOG_TEST, LOG_INFO, "TestThree");
michael@0:   r_log(NR_LOG_TEST, LOG_INFO, "TestFour");
michael@0:   r_log(NR_LOG_TEST, LOG_INFO, "TestFive");
michael@0:   r_log(NR_LOG_TEST, LOG_INFO, "TestSix");
michael@0:   std::vector<std::string> substrings;
michael@0:   // Matches Two, Three, Four, and Six
michael@0:   substrings.push_back("tT");
michael@0:   substrings.push_back("o");
michael@0:   substrings.push_back("r");
michael@0:   substrings.push_back("S");
michael@0:   std::deque<std::string> logs;
michael@0:   RLogRingBuffer::GetInstance()->FilterAny(substrings, 2, &logs);
michael@0:   ASSERT_EQ(2U, logs.size());
michael@0:   ASSERT_EQ("TestSix", logs.back());
michael@0:   ASSERT_EQ("TestFour", logs.front());
michael@0: }
michael@0: 
michael@0: TEST_F(RLogRingBufferTest, TestLimit) {
michael@0:   RLogRingBuffer::GetInstance()->SetLogLimit(3);
michael@0:   r_log(NR_LOG_TEST, LOG_INFO, "Test1");
michael@0:   r_log(NR_LOG_TEST, LOG_INFO, "Test2");
michael@0:   r_log(NR_LOG_TEST, LOG_INFO, "Test3");
michael@0:   r_log(NR_LOG_TEST, LOG_INFO, "Test4");
michael@0:   r_log(NR_LOG_TEST, LOG_INFO, "Test5");
michael@0:   r_log(NR_LOG_TEST, LOG_INFO, "Test6");
michael@0:   std::deque<std::string> logs;
michael@0:   RLogRingBuffer::GetInstance()->GetAny(0, &logs);
michael@0:   ASSERT_EQ(3U, logs.size());
michael@0:   ASSERT_EQ("Test6", logs.back());
michael@0:   ASSERT_EQ("Test4", logs.front());
michael@0: }
michael@0: 
michael@0: TEST_F(RLogRingBufferTest, TestLimitBulkDiscard) {
michael@0:   r_log(NR_LOG_TEST, LOG_INFO, "Test1");
michael@0:   r_log(NR_LOG_TEST, LOG_INFO, "Test2");
michael@0:   r_log(NR_LOG_TEST, LOG_INFO, "Test3");
michael@0:   r_log(NR_LOG_TEST, LOG_INFO, "Test4");
michael@0:   r_log(NR_LOG_TEST, LOG_INFO, "Test5");
michael@0:   r_log(NR_LOG_TEST, LOG_INFO, "Test6");
michael@0:   RLogRingBuffer::GetInstance()->SetLogLimit(3);
michael@0:   std::deque<std::string> logs;
michael@0:   RLogRingBuffer::GetInstance()->GetAny(0, &logs);
michael@0:   ASSERT_EQ(3U, logs.size());
michael@0:   ASSERT_EQ("Test6", logs.back());
michael@0:   ASSERT_EQ("Test4", logs.front());
michael@0: }
michael@0: 
michael@0: TEST_F(RLogRingBufferTest, TestIncreaseLimit) {
michael@0:   RLogRingBuffer::GetInstance()->SetLogLimit(3);
michael@0:   r_log(NR_LOG_TEST, LOG_INFO, "Test1");
michael@0:   r_log(NR_LOG_TEST, LOG_INFO, "Test2");
michael@0:   r_log(NR_LOG_TEST, LOG_INFO, "Test3");
michael@0:   r_log(NR_LOG_TEST, LOG_INFO, "Test4");
michael@0:   r_log(NR_LOG_TEST, LOG_INFO, "Test5");
michael@0:   r_log(NR_LOG_TEST, LOG_INFO, "Test6");
michael@0:   RLogRingBuffer::GetInstance()->SetLogLimit(300);
michael@0:   std::deque<std::string> logs;
michael@0:   RLogRingBuffer::GetInstance()->GetAny(0, &logs);
michael@0:   ASSERT_EQ(3U, logs.size());
michael@0:   ASSERT_EQ("Test6", logs.back());
michael@0:   ASSERT_EQ("Test4", logs.front());
michael@0: }
michael@0: 
michael@0: TEST_F(RLogRingBufferTest, TestClear) {
michael@0:   r_log(NR_LOG_TEST, LOG_INFO, "Test1");
michael@0:   r_log(NR_LOG_TEST, LOG_INFO, "Test2");
michael@0:   r_log(NR_LOG_TEST, LOG_INFO, "Test3");
michael@0:   r_log(NR_LOG_TEST, LOG_INFO, "Test4");
michael@0:   r_log(NR_LOG_TEST, LOG_INFO, "Test5");
michael@0:   r_log(NR_LOG_TEST, LOG_INFO, "Test6");
michael@0:   RLogRingBuffer::GetInstance()->SetLogLimit(0);
michael@0:   RLogRingBuffer::GetInstance()->SetLogLimit(4096);
michael@0:   std::deque<std::string> logs;
michael@0:   RLogRingBuffer::GetInstance()->GetAny(0, &logs);
michael@0:   ASSERT_EQ(0U, logs.size());
michael@0: }
michael@0: 
michael@0: TEST_F(RLogRingBufferTest, TestReInit) {
michael@0:   r_log(NR_LOG_TEST, LOG_INFO, "Test1");
michael@0:   r_log(NR_LOG_TEST, LOG_INFO, "Test2");
michael@0:   r_log(NR_LOG_TEST, LOG_INFO, "Test3");
michael@0:   r_log(NR_LOG_TEST, LOG_INFO, "Test4");
michael@0:   r_log(NR_LOG_TEST, LOG_INFO, "Test5");
michael@0:   r_log(NR_LOG_TEST, LOG_INFO, "Test6");
michael@0:   ReInit();
michael@0:   std::deque<std::string> logs;
michael@0:   RLogRingBuffer::GetInstance()->GetAny(0, &logs);
michael@0:   ASSERT_EQ(0U, logs.size());
michael@0: }
michael@0: 
michael@0: int main(int argc, char** argv) {
michael@0:   NR_reg_init(NR_REG_MODE_LOCAL);
michael@0:   r_log_init();
michael@0:   /* Would be nice to be able to register/unregister in the fixture */
michael@0:   const char* facility = "rlogringbuffer_test";
michael@0:   r_log_register(const_cast<char*>(facility), &NR_LOG_TEST);
michael@0:   ::testing::InitGoogleTest(&argc, argv);
michael@0: 
michael@0:   int rv = RUN_ALL_TESTS();
michael@0:   return rv;
michael@0: }
michael@0: