src/tint/utils/io/tmpfile_test.cc

// Copyright 2021 The Tint Authors.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

#include "src/tint/utils/io/tmpfile.h"

#include <fstream>

#include "gtest/gtest.h"

namespace tint::utils {
namespace {

TEST(TmpFileTest, WriteReadAppendDelete) {
  std::string path;
  {
    TmpFile tmp;
    if (!tmp) {
      GTEST_SKIP() << "Unable to create a temporary file";
    }

    path = tmp.Path();

    // Write a string to the temporary file
    tmp << "hello world\n";

    // Check the content of the file
    {
      std::ifstream file(path);
      ASSERT_TRUE(file);
      std::string line;
      EXPECT_TRUE(std::getline(file, line));
      EXPECT_EQ(line, "hello world");
      EXPECT_FALSE(std::getline(file, line));
    }

    // Write some more content to the file
    tmp << 42;

    // Check the content of the file again
    {
      std::ifstream file(path);
      ASSERT_TRUE(file);
      std::string line;
      EXPECT_TRUE(std::getline(file, line));
      EXPECT_EQ(line, "hello world");
      EXPECT_TRUE(std::getline(file, line));
      EXPECT_EQ(line, "42");
      EXPECT_FALSE(std::getline(file, line));
    }
  }

  // Check the file has been deleted when it fell out of scope
  std::ifstream file(path);
  ASSERT_FALSE(file);
}

TEST(TmpFileTest, FileExtension) {
  const std::string kExt = ".foo";
  std::string path;
  {
    TmpFile tmp(kExt);
    if (!tmp) {
      GTEST_SKIP() << "Unable create a temporary file";
    }
    path = tmp.Path();
  }

  ASSERT_GT(path.length(), kExt.length());
  EXPECT_EQ(kExt, path.substr(path.length() - kExt.length()));

  // Check the file has been deleted when it fell out of scope
  std::ifstream file(path);
  ASSERT_FALSE(file);
}

}  // namespace
}  // namespace tint::utils