161 lines
3.9 KiB
C++
161 lines
3.9 KiB
C++
// 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)));
|
|
|
|
}
|
|
}
|