srsRAN_4G/lib/test/srslog/file_test_utils.h

/**
 * Copyright 2013-2021 Software Radio Systems Limited
 *
 * This file is part of srsRAN.
 *
 * srsRAN is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as
 * published by the Free Software Foundation, either version 3 of
 * the License, or (at your option) any later version.
 *
 * srsRAN is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Affero General Public License for more details.
 *
 * A copy of the GNU Affero General Public License can be found in
 * the LICENSE file in the top-level directory of this distribution
 * and at http://www.gnu.org/licenses/.
 *
 */

#ifndef SRSLOG_FILE_TEST_UTILS_H
#define SRSLOG_FILE_TEST_UTILS_H

#include <cstdio>
#include <fstream>
#include <string>
#include <unistd.h>
#include <vector>

namespace file_test_utils {

/// Removes a list of files from the file system on object destruction.
class scoped_file_deleter
{
  std::vector<std::string> paths;

public:
  explicit scoped_file_deleter(const std::string& path) : paths({path}) {}
  scoped_file_deleter(std::initializer_list<std::string> init) : paths(init) {}

  ~scoped_file_deleter()
  {
    for (const auto& path : paths) {
      ::remove(path.c_str());
    }
  }
};

/// Returns true if a file exists in the specified path, otherwise returns
/// false.
inline bool file_exists(const std::string& path)
{
  return ::access(path.c_str(), F_OK) == 0;
}

/// Reads the contents of the file specified in path and returns true if they
/// match with the data in entries, otherwise returns false.
inline bool compare_file_contents(const std::string& path, const std::vector<std::string>& entries)
{
  std::ifstream file(path, std::ios::binary);
  if (!file.is_open()) {
    return false;
  }

  std::vector<std::string> data;
  for (std::string line; std::getline(file, line);) {
    data.push_back(line + '\n');
  }

  return (data == entries);
}

} // namespace file_test_utils

#endif // SRSLOG_FILE_TEST_UTILS_H