| // Licensed to the Apache Software Foundation (ASF) under one |
| // or more contributor license agreements. See the NOTICE file |
| // distributed with this work for additional information |
| // regarding copyright ownership. The ASF licenses this file |
| // to you 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 <gtest/gtest.h> |
| #include <errno.h> // errno |
| #include "butil/files/temp_file.h" |
| |
| namespace { |
| |
| class TempFileTest : public ::testing::Test{ |
| protected: |
| TempFileTest(){}; |
| virtual ~TempFileTest(){}; |
| virtual void SetUp() { |
| }; |
| virtual void TearDown() { |
| }; |
| }; |
| |
| TEST_F(TempFileTest, should_create_tmp_file) |
| { |
| butil::TempFile tmp; |
| struct stat st; |
| //check if existed |
| ASSERT_EQ(0, stat(tmp.fname(), &st)); |
| } |
| |
| TEST_F(TempFileTest, should_write_string) |
| { |
| butil::TempFile tmp; |
| const char *exp = "a test file"; |
| ASSERT_EQ(0, tmp.save(exp)); |
| |
| FILE *fp = fopen(tmp.fname(), "r"); |
| ASSERT_NE((void*)0, fp); |
| |
| char buf[1024]; |
| char *act = fgets(buf, 1024, fp); |
| fclose(fp); |
| |
| EXPECT_STREQ(exp, act); |
| } |
| |
| TEST_F(TempFileTest, temp_with_specific_ext) |
| { |
| butil::TempFile tmp("blah"); |
| const char *exp = "a test file"; |
| ASSERT_EQ(0, tmp.save(exp)); |
| struct stat st; |
| ASSERT_EQ(0, stat(tmp.fname(), &st)); |
| |
| ASSERT_STREQ(".blah", strrchr(tmp.fname(), '.')); |
| FILE *fp = fopen(tmp.fname(), "r"); |
| ASSERT_NE((void*)0, fp) << strerror(errno); |
| |
| char buf[1024]; |
| char *act = fgets(buf, 1024, fp); |
| fclose(fp); |
| |
| EXPECT_STREQ(exp, act); |
| } |
| |
| TEST_F(TempFileTest, should_delete_when_exit) |
| { |
| std::string fname; |
| struct stat st; |
| { |
| butil::TempFile tmp; |
| //check if existed |
| ASSERT_EQ(0, stat(tmp.fname(), &st)); |
| fname = tmp.fname(); |
| } |
| |
| //check if existed |
| ASSERT_EQ(-1, stat(fname.c_str(), &st)); |
| ASSERT_EQ(ENOENT, errno); |
| } |
| |
| TEST_F(TempFileTest, should_save_with_format) |
| { |
| butil::TempFile tmp; |
| tmp.save_format("%s%d%ld%s", "justmp", 1, 98L, "hello world"); |
| |
| FILE *fp = fopen(tmp.fname(), "r"); |
| ASSERT_NE((void*)0, fp); |
| |
| char buf[1024]; |
| char *act = fgets(buf, 1024, fp); |
| fclose(fp); |
| |
| EXPECT_STREQ("justmp198hello world", act); |
| } |
| |
| TEST_F(TempFileTest, should_save_with_format_in_long_string) |
| { |
| char buf[2048]; |
| memset(buf, 'a', sizeof(buf)); |
| buf[2047] = '\0'; |
| |
| butil::TempFile tmp; |
| tmp.save_format("%s", buf); |
| |
| FILE *fp = fopen(tmp.fname(), "r"); |
| ASSERT_NE((void*)0, fp); |
| |
| char buf2[2048]; |
| char *act = fgets(buf2, 2048, fp); |
| fclose(fp); |
| EXPECT_STREQ(buf, act); |
| } |
| |
| struct test_t { |
| int a; |
| int b; |
| char c[4]; |
| }; |
| |
| TEST_F(TempFileTest, save_binary_twice) |
| { |
| test_t data = {12, -34, {'B', 'E', 'E', 'F'}}; |
| butil::TempFile tmp; |
| ASSERT_EQ(0, tmp.save_bin(&data, sizeof(data))); |
| |
| FILE *fp = fopen(tmp.fname(), "r"); |
| ASSERT_NE((void*)0, fp); |
| test_t act_data; |
| bzero(&act_data, sizeof(act_data)); |
| ASSERT_EQ((size_t)1, fread(&act_data, sizeof(act_data), 1, fp)); |
| fclose(fp); |
| |
| ASSERT_EQ(0, memcmp(&data, &act_data, sizeof(data))); |
| |
| // save twice |
| test_t data2 = { 89, 1000, {'E', 'C', 'A', 'Z'}}; |
| ASSERT_EQ(0, tmp.save_bin(&data2, sizeof(data2))); |
| |
| fp = fopen(tmp.fname(), "r"); |
| ASSERT_NE((void*)0, fp); |
| bzero(&act_data, sizeof(act_data)); |
| ASSERT_EQ((size_t)1, fread(&act_data, sizeof(act_data), 1, fp)); |
| fclose(fp); |
| |
| ASSERT_EQ(0, memcmp(&data2, &act_data, sizeof(data2))); |
| |
| } |
| } |