65 lines
2.2 KiB
C++
65 lines
2.2 KiB
C++
// Copyright 2015 Google LLC. All rights reserved.
|
|
//
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file or at
|
|
// https://developers.google.com/open-source/licenses/bsd
|
|
|
|
#ifndef MEDIA_FILE_FILE_TEST_UTIL_H_
|
|
#define MEDIA_FILE_FILE_TEST_UTIL_H_
|
|
|
|
#include <gmock/gmock.h>
|
|
#include <gtest/gtest.h>
|
|
|
|
#include <iterator>
|
|
#include <string>
|
|
|
|
#include <packager/file/file.h>
|
|
|
|
namespace shaka {
|
|
|
|
#define ASSERT_FILE_EQ(file_name, array) \
|
|
do { \
|
|
std::string temp_data; \
|
|
ASSERT_TRUE(File::ReadFileToString((file_name), &temp_data)); \
|
|
const char* array_ptr = reinterpret_cast<const char*>(array); \
|
|
ASSERT_EQ(std::string(array_ptr, std::size(array)), temp_data); \
|
|
} while (false)
|
|
|
|
#define ASSERT_FILE_STREQ(file_name, str) \
|
|
do { \
|
|
std::string temp_data; \
|
|
ASSERT_TRUE(File::ReadFileToString((file_name), &temp_data)); \
|
|
ASSERT_EQ(str, temp_data); \
|
|
} while (false)
|
|
|
|
#define ASSERT_FILE_ENDS_WITH(file_name, array) \
|
|
do { \
|
|
std::string temp_data; \
|
|
ASSERT_TRUE(File::ReadFileToString((file_name), &temp_data)); \
|
|
EXPECT_THAT(temp_data, \
|
|
::testing::EndsWith(std::string( \
|
|
reinterpret_cast<const char*>(array), sizeof(array)))); \
|
|
} while (false)
|
|
|
|
// Generate a unique filename.
|
|
std::string generate_unique_temp_path();
|
|
void delete_file(const std::string& path);
|
|
|
|
// A temporary file that is removed from the filesystem when the object is
|
|
// destroyed. Useful in tests that use ASSERT to avoid leaving behind temp
|
|
// files.
|
|
class TempFile {
|
|
public:
|
|
TempFile();
|
|
~TempFile();
|
|
|
|
const std::string& path() const { return path_; }
|
|
|
|
private:
|
|
std::string path_;
|
|
};
|
|
|
|
} // namespace shaka
|
|
|
|
#endif // MEDIA_FILE_FILE_TEST_UTIL_H_
|