open5gs/lib/core/test/testfilecopy.c

164 lines
4.6 KiB
C

#include "core_file.h"
#include "testutil.h"
#define DIRNAME "data"
#define FILENAME1 DIRNAME "/file_datafile.txt"
#define TESTSTR1 "This is the file data file."
#define FILENAME2 DIRNAME "/mmap_datafile.txt"
#define TESTSTR2 "This is the MMAP data file."
static void test_filecopy_init(abts_case *tc, void *data)
{
status_t rv;
file_info_t finfo;
size_t bytes;
file_t *filetest = NULL;
rv = dir_make("data", FILE_UREAD | FILE_UWRITE | FILE_UEXECUTE);
rv = file_stat(&finfo, "data", FILE_INFO_TYPE);
ABTS_INT_EQUAL(tc, CORE_OK, rv);
ABTS_INT_EQUAL(tc, FILE_DIR, finfo.filetype);
rv = file_open(&filetest, FILENAME1,
FILE_WRITE | FILE_CREATE,
FILE_UREAD | FILE_UWRITE | FILE_GREAD);
ABTS_INT_EQUAL(tc, CORE_OK, rv);
bytes = strlen(TESTSTR1);
rv = file_write(filetest, TESTSTR1, &bytes);
ABTS_INT_EQUAL(tc, CORE_OK, rv);
file_close(filetest);
rv = file_open(&filetest, FILENAME2,
FILE_WRITE | FILE_CREATE,
FILE_UREAD | FILE_UWRITE | FILE_GREAD);
ABTS_INT_EQUAL(tc, CORE_OK, rv);
bytes = strlen(TESTSTR2);
rv = file_write(filetest, TESTSTR2, &bytes);
ABTS_INT_EQUAL(tc, CORE_OK, rv);
file_close(filetest);
}
static void copy_helper(abts_case *tc, const char *from, const char * to,
file_perms_t perms, int append)
{
status_t rv;
status_t dest_rv;
file_info_t copy;
file_info_t orig;
file_info_t dest;
dest_rv = file_stat(&dest, to, FILE_INFO_SIZE);
if (!append) {
rv = file_copy(from, to, perms);
}
else {
rv = file_append(from, to, perms);
}
CORE_ASSERT_OK(tc, "Error copying file", rv);
rv = file_stat(&orig, from, FILE_INFO_SIZE);
CORE_ASSERT_OK(tc, "Couldn't stat original file", rv);
rv = file_stat(&copy, to, FILE_INFO_SIZE);
CORE_ASSERT_OK(tc, "Couldn't stat copy file", rv);
if (!append) {
ABTS_ASSERT(tc, "File size differs", orig.size == copy.size);
}
else {
ABTS_ASSERT(tc, "File size differs",
((dest_rv == CORE_OK)
? dest.size : 0) + orig.size == copy.size);
}
}
static void copy_short_file(abts_case *tc, void *data)
{
status_t rv;
/* make absolutely sure that the dest file doesn't exist. */
file_remove("data/file_copy.txt");
copy_helper(tc, "data/file_datafile.txt", "data/file_copy.txt",
FILE_SOURCE_PERMS, 0);
rv = file_remove("data/file_copy.txt");
CORE_ASSERT_OK(tc, "Couldn't remove copy file", rv);
}
static void copy_over_existing(abts_case *tc, void *data)
{
status_t rv;
/* make absolutely sure that the dest file doesn't exist. */
file_remove("data/file_copy.txt");
/* This is a cheat. I don't want to create a new file, so I just copy
* one file, then I copy another. If the second copy succeeds, then
* this works.
*/
copy_helper(tc, "data/file_datafile.txt", "data/file_copy.txt",
FILE_SOURCE_PERMS, 0);
copy_helper(tc, "data/mmap_datafile.txt", "data/file_copy.txt",
FILE_SOURCE_PERMS, 0);
rv = file_remove("data/file_copy.txt");
CORE_ASSERT_OK(tc, "Couldn't remove copy file", rv);
}
static void append_nonexist(abts_case *tc, void *data)
{
status_t rv;
/* make absolutely sure that the dest file doesn't exist. */
file_remove("data/file_copy.txt");
copy_helper(tc, "data/file_datafile.txt", "data/file_copy.txt",
FILE_SOURCE_PERMS, 0);
rv = file_remove("data/file_copy.txt");
CORE_ASSERT_OK(tc, "Couldn't remove copy file", rv);
}
static void append_exist(abts_case *tc, void *data)
{
status_t rv;
/* make absolutely sure that the dest file doesn't exist. */
file_remove("data/file_copy.txt");
/* This is a cheat. I don't want to create a new file, so I just copy
* one file, then I copy another. If the second copy succeeds, then
* this works.
*/
copy_helper(tc, "data/file_datafile.txt", "data/file_copy.txt",
FILE_SOURCE_PERMS, 0);
copy_helper(tc, "data/mmap_datafile.txt", "data/file_copy.txt",
FILE_SOURCE_PERMS, 1);
rv = file_remove("data/file_copy.txt");
CORE_ASSERT_OK(tc, "Couldn't remove copy file", rv);
}
abts_suite *testfilecopy(abts_suite *suite)
{
suite = ADD_SUITE(suite)
abts_run_test(suite, test_filecopy_init, NULL);
abts_run_test(suite, copy_short_file, NULL);
abts_run_test(suite, copy_over_existing, NULL);
abts_run_test(suite, append_nonexist, NULL);
abts_run_test(suite, append_exist, NULL);
return suite;
}