logchecker.h

Go to the documentation of this file.

#ifndef ION_BASE_LOGCHECKER_H_
#define ION_BASE_LOGCHECKER_H_

#include <sstream>
#include <string>
#include <vector>

#include "ion/base/logging.h"

namespace ion {
namespace base {

class ION_API LogChecker : public port::LogEntryWriter {
 public:
  LogChecker();

  ~LogChecker() override;

  void Write(port::LogSeverity severity, const std::string& message)
      override;

  bool HasMessage(const std::string& severity, const std::string& substring);

  bool HasNoMessage(const std::string& severity, const std::string& substring);

  bool HasAnyMessages() const {
    return !GetLogString().empty();
  }

  const std::vector<std::string> GetAllMessages();

  void ClearLog() {
    stream_.str("");
  }

  const std::string GetLogString() const {
    return stream_.str();
  }

 private:
  port::LogEntryWriter* previous_writer_;
  std::ostringstream stream_;
};

}  // namespace base
}  // namespace ion

#endif  // ION_BASE_LOGCHECKER_H_