165 lines
3.1 KiB
C
165 lines
3.1 KiB
C
#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
|