khanat-code-old/code/nel/tools/nel_unit_test/ut_misc_file.h

166 lines
3.1 KiB
C
Raw Normal View History

2010-05-06 00:08:41 +00:00
#ifndef UT_MISC_FILE
#define UT_MISC_FILE
#include <nel/misc/file.h>
#include <nel/misc/path.h>
// Test suite for CFile behavior
struct CUTMiscFile : public Test::Suite
{
CUTMiscFile()
{
TEST_ADD(CUTMiscFile::copyOneBigFile);
TEST_ADD(CUTMiscFile::copyDifferentFileSize);
TEST_ADD(CUTMiscFile::moveOneBigFile);
TEST_ADD(CUTMiscFile::moveDifferentFileSize);
// Add a line here when adding a new test METHOD
}
private:
string _SrcFile;
string _DstFile;
void setup()
{
_SrcFile = "__copy_file_src.foo";
_DstFile = "__copy_file_dst.foo";
}
void tear_down()
{
}
void copyFileSize(uint fileSize)
{
// create a source file (using standard c code)
FILE *fp = fopen(_SrcFile.c_str(), "wb");
nlverify(fp != NULL);
for (uint i=0; i<fileSize; ++i)
{
uint8 c = uint8(i & 0xff);
nlverify(fwrite(&c, 1, 1, fp) == 1);
}
fclose(fp);
fp = NULL;
NLMISC::CFile::copyFile(_DstFile.c_str(), _SrcFile.c_str(), false);
// verify the resulting file
fp = fopen(_DstFile.c_str(), "rb");
TEST_ASSERT(fp != NULL);
if (fp)
{
for (uint i=0; i<fileSize; ++i)
{
uint8 c;
size_t nbRead = fread(&c, 1,1, fp);
TEST_ASSERT(nbRead == 1);
if (nbRead != 1)
break;
TEST_ASSERT_MSG(c == uint8(i & 0xff), "File content changed during copy");
if (c != uint8(i & 0xff))
break;
}
fclose(fp);
}
}
void copyOneBigFile()
{
// check for a big file
copyFileSize(1024*1024);
}
void copyDifferentFileSize()
{
// check for a series of size
for (uint i=0; i<10; ++i)
{
copyFileSize(i);
}
srand(1234);
for (uint i=0; i<1024; ++i)
{
i += rand()%10;
copyFileSize(i);
}
}
void moveFileSize(size_t fileSize)
{
// remove the destination if any
FILE *fp = fopen(_DstFile.c_str(), "rb");
if (fp != NULL)
{
fclose(fp);
NLMISC::CFile::deleteFile(_DstFile);
}
// create a source file (using standard c code)
fp = fopen(_SrcFile.c_str(), "wb");
nlverify(fp != NULL);
for (uint i=0; i<fileSize; ++i)
{
uint8 c = uint8(i & 0xff);
nlverify(fwrite(&c, 1, 1, fp) == 1);
}
fclose(fp);
fp = NULL;
NLMISC::CFile::moveFile(_DstFile.c_str(), _SrcFile.c_str());
// verify the resulting file
fp = fopen(_SrcFile.c_str(), "rb");
TEST_ASSERT_MSG(fp == NULL, "The source file is not removed");
if (fp)
fclose(fp);
fp = fopen(_DstFile.c_str(), "rb");
TEST_ASSERT(fp != NULL);
if (fp)
{
for (uint i=0; i<fileSize; ++i)
{
uint8 c;
size_t nbRead = fread(&c, 1,1, fp);
TEST_ASSERT(nbRead == 1);
if (nbRead != 1)
break;
TEST_ASSERT_MSG(c == uint8(i & 0xff), "File content changed during move");
if (c != uint8(i & 0xff))
break;
}
fclose(fp);
}
}
void moveOneBigFile()
{
// check for a big file
moveFileSize(1024*1024);
}
void moveDifferentFileSize()
{
// check for a series of size
for (uint i=0; i<10; ++i)
{
moveFileSize(i);
}
srand(1234);
for (uint i=0; i<1024; ++i)
{
i += rand()%10;
moveFileSize(i);
}
}
};
#endif