test_ncbifile.cpp

ncbi源码

// Work with directories
//

#if defined(NCBI_OS_MAC)
#   define REL ":"
#   define SEP ":"
#   define CWD ":"
#else
#   define REL ""
#   define SEP "/"
#   define CWD "."
#endif

static void s_TEST_Dir(void)
{
    // Delete the directory if it exists before we start testing
    CDirEntry("dir_1").Remove();
    
    // Create directory
    assert( CDir("dir_1").Create() );
    assert( !CDir("dir_1").Create() );

    // Create file in the directory
    FILE* file = fopen(REL "dir_1" SEP "file_1", "w+");
    assert( file );
    fclose(file);

    // Check the directory exists
    assert( CDir("dir_1").Exists() );
    assert( !CDir("dir_2").Exists() );
    assert( CDirEntry("dir_1").Exists() );
    assert( CDirEntry(REL"dir_1" SEP "file_1").Exists() );

    // Check entry type
    assert( CDirEntry("dir_1").IsDir() );
    assert( !CDirEntry("dir_1").IsFile() );
    assert( CDirEntry(REL "dir_1" SEP "file_1").IsFile() );
    assert( !CDirEntry(REL "dir_1" SEP "file_1").IsDir() );
    
    // Rename the directory
    assert( CDir("dir_1").Rename("dir_2") );

    // Remove the directory
    assert( !CDirEntry("dir_2").Remove(CDirEntry::eOnlyEmpty) );
    assert( CFile(REL "dir_2" SEP "file_1").Remove() );
    assert( CDir("dir_2").Remove(CDir::eOnlyEmpty) );

    // Check the directory exists
    assert( !CDir("dir_1").Exists() );
    assert( !CDir("dir_2").Exists() );

    // Current directory list
    CDir dir(CWD);

    cout << endl;
    CDir::TEntries contents = dir.GetEntries("*", CDir::eIgnoreRecursive);
    ITERATE(CDir::TEntries, i, contents) {
        string entry = (*i)->GetPath();
        cout << entry << endl;
    }
    cout << endl;


    vector<string> masks;
    masks.push_back("*");
    CDir::TEntries contents2 = dir.GetEntries(masks, CDir::eIgnoreRecursive);
    assert(contents.size() == contents2.size());

    vector<string> files;
    vector<string> paths;
    paths.push_back(".");

    FindFiles(files, paths.begin(), paths.end(), 
                     masks.begin(), masks.end());


    for (unsigned int i = 0; i < contents.size(); ++i) {
        const CDirEntry& entry1 = *contents[i];
        const CDirEntry& entry2 = *contents2[i];
        string ep1 = entry1.GetPath();
        string ep2 = entry2.GetPath();
        const string& f = files[i];
        assert(ep1 == ep2);
        assert(ep1 == f);
    }


    cout << "Recursive content" << endl;
    files.clear();
    FindFiles(files, paths.begin(), paths.end(), 
                     masks.begin(), masks.end(), 
                     fFF_File | fFF_Dir | fFF_Recursive);
    ITERATE(vector<string>, fit, files) {
        cout << *fit << endl;
    }
    cout << "-----" << endl;


    // Create dir structure for deletion
    assert( CDir("dir_3").Create() );
    assert( CDir(REL "dir_3" SEP "subdir_1").Create() );
    assert( CDir(REL "dir_3" SEP "subdir_2").Create() );
    
    file = fopen(REL "dir_3" SEP "file", "w+");
    assert( file );
    fclose(file);
    file = fopen(REL "dir_3" SEP "subdir_1" SEP "file", "w+");
    assert( file );
    fclose(file);

    // Delete dir
    dir.Reset("dir_3");
    assert( !dir.Remove(CDir::eOnlyEmpty) );
    assert( dir.Exists() );

    assert( !dir.Remove(CDir::eNonRecursive) );
    assert( dir.Exists() );
    assert( CDir(REL "dir_3" SEP "subdir_1").Exists() );
    assert( CFile(REL "dir_3" SEP "subdir_1" SEP "file").Exists() );
    assert( !CFile(REL "dir_3" SEP "file").Exists() );

    assert( dir.Remove(CDir::eRecursive) );
    assert( !dir.Exists() );
    
    // Home dir
    string homedir = CDir::GetHome();
    assert ( !homedir.empty() );
    cout << homedir << endl;

    // Creation of relative path from 2 absolute pathes:
    string rel_path;

#if defined(NCBI_OS_MSWIN)
    assert( CDirEntry::CreateRelativePath(
            "C:\\x\\y\\z\\",
            "C:\\x\\a\\b\\") == "..\\..\\a\\b\\" );

    assert( CDirEntry::CreateRelativePath(
            "C:\\x\\y\\z\\",
            "C:\\x\\a\\b\\file.txt") == "..\\..\\a\\b\\file.txt" );
 
    assert( CDirEntry::CreateRelativePath(
            "C:\\x\\y\\z\\", 
            "C:\\x\\y\\z\\").empty() );

    assert( CDirEntry::CreateRelativePath(
            "C:\\x\\y\\z\\", 
            "C:\\x\\y\\a\\") == "..\\a\\" );

    assert( CDirEntry::CreateRelativePath(
            "\\x\\y\\z\\", 
            "\\x\\y\\") == "..\\" );

#elif defined(NCBI_OS_UNIX )
    assert( CDirEntry::CreateRelativePath(
            "/usr/bin/", "/usr/" ) == "../" );
    assert( CDirEntry::CreateRelativePath(
            "/usr/bin/", "/etc/" ) == "../../etc/" );

#elif defined(NCBI_OS_MAC)
    // NOT implemented!
#endif
}


/////////////////////////////////
// Memory mapping
//

static void s_TEST_MemoryFile(void)
{
    static const char   s_FileName[] = "test.tmp";
    static const char   s_Data[]     = "test data";
    static const size_t s_DataLen    = sizeof(s_Data) - 1;

    // Create test file 
    FILE* file = fopen(s_FileName, "w+");
    assert( file );
    fputs( s_Data, file);
    fclose(file);
    
    // Check if the file exists now
    CFile f(s_FileName);
    assert( f.Exists() );
    assert( f.GetLength() == s_DataLen );

    // eMMP_Read [+ eMMS_Private]
    
    {{
        // Map file into memory (for reading only by default)
        CMemoryFile m1(s_FileName);
        assert( m1.GetSize() == s_DataLen );
        assert( memcmp(m1.GetPtr(), s_Data, s_DataLen) == 0 );

        // Map second copy of the file into the memory
        CMemoryFile m2(s_FileName);
        assert( m1.GetSize() == m2.GetSize() );
        assert( memcmp(m1.GetPtr(), m2.GetPtr(), (size_t)m2.GetSize()) == 0 );

        // Unmap second copy
        assert( m2.Unmap() );
        assert( m2.GetSize() == -1 );
        assert( m2.GetPtr()  ==  0 );
        assert( m2.Unmap() );
    }}

    // eMMP_ReadWrite + eMMS_Shared
    
    {{
        // Map file into memory
        CMemoryFile m1(s_FileName, CMemoryFile::eMMP_ReadWrite,
                       CMemoryFile::eMMS_Shared);
        assert( m1.GetSize() == s_DataLen );
        assert( memcmp(m1.GetPtr(), s_Data, s_DataLen) == 0 );
        char* ptr = (char*)m1.GetPtr();
        *ptr = '@';
        assert( m1.Flush() );

        // Map second copy of the file into the memory
        CMemoryFile m2(s_FileName, CMemoryFile::eMMP_Read,
                       CMemoryFile::eMMS_Shared);
        assert( m1.GetSize() == m2.GetSize() );
        // m2 must contain changes made via m1 
        assert( memcmp(m2.GetPtr(), ptr, s_DataLen) == 0 );

        // Restore previous data
        memcpy(ptr, s_Data, s_DataLen);
        // Flushing data to disk at memory unmapping in the destructor
    }}
    {{
        // Checking the previous flush
        CMemoryFile m1(s_FileName);
        assert( m1.GetSize() == s_DataLen );
        assert( memcmp(m1.GetPtr(), s_Data, s_DataLen) == 0 );
    }}

    // eMMP_ReadWrite + eMMS_Private

    {{
        // Map file into memory
        CMemoryFile m1(s_FileName, CMemoryFile::eMMP_ReadWrite,
                       CMemoryFile::eMMS_Private);
        assert( m1.GetSize() == s_DataLen );
        assert( memcmp(m1.GetPtr(), s_Data, s_DataLen) == 0 );
        char* ptr = (char*)m1.GetPtr();
        *ptr = '@';
        assert( m1.Flush() );

        // Map second copy of the file into the memory
        CMemoryFile m2(s_FileName, CMemoryFile::eMMP_Read,
                       CMemoryFile::eMMS_Shared);
        assert( m1.GetSize() == m2.GetSize() );
        // m2 must contain the original data
        assert( memcmp(m2.GetPtr(), s_Data, s_DataLen) == 0 );
    }}

    // Remove the file
    assert( f.Remove() );
}



////////////////////////////////
// Test application
//

class CTest : public CNcbiApplication
{
public:
    void Init(void);
    int Run(void);
};


void CTest::Init(void)
{
    SetDiagPostLevel(eDiag_Warning);
    auto_ptr<CArgDescriptions> d(new CArgDescriptions);
    d->SetUsageContext("test_files",
                       "test file's accessory functions");
    SetupArgDescriptions(d.release());
}


int CTest::Run(void)
{
    cout << "Run test" << endl << endl;

    // CDirEntry
    s_TEST_SplitPath();
    s_TEST_CheckPath();
    s_TEST_MatchesMask();

    // CFile
    s_TEST_File();
    // CDir
    s_TEST_Dir();

    // CMemoryFile
    s_TEST_MemoryFile();

    cout << endl;
    cout << "TEST execution completed successfully!" << endl << endl;
    return 0;
}


///////////////////////////////////
// APPLICATION OBJECT  and  MAIN
//

int main(int argc, const char* argv[])
{
    // Execute main application function
    return CTest().AppMain(argc, argv, 0, eDS_Default, 0);
}


/*
 * ===========================================================================
 * $Log: test_ncbifile.cpp,v $
 * Revision 1000.5  2004/06/01 19:10:13  gouriano
 * PRODUCTION: UPGRADED [GCC34_MSVC7] Dev-tree R1.38
 *
 * Revision 1.38  2004/05/14 13:59:51  gorelenk
 * Added include of ncbi_pch.hpp
 *
 * Revision 1.37  2004/04/29 15:15:00  kuznets
 * Modifications of FindFile test
 *
 * Revision 1.36  2004/03/17 15:41:28  ivanov
 * Expanded temporary file name generation test
 *
 * Revision 1.35  2004/01/05 22:10:33  gorelenk
 * += UNIX test for CDirEntry::CreateRelativePath()
 *
 * Revision 1.34  2004/01/05 21:41:26  gorelenk
 * += Exception throwing in CDirEntry::CreateRelativePath()
 *
 * Revision 1.33  2004/01/05 20:05:06  gorelenk
 * + Test for CDirEntry::CreateRelativePath()
 *
 * Revision 1.32  2003/12/15 15:42:48  ivanov
 * Removed incorrect last access time check
 *
 * Revision 1.31  2003/12/01 12:17:35  ivanov
 * Fixed comparison operator for assert( atime >= ctime )
 *
 * Revision 1.30  2003/11/28 16:23:42  ivanov
 * Added tests for CDirEntry::SetTime()
 *
 * Revision 1.29  2003/11/05 16:27:45  kuznets
 * + test for FindFile
 *
 * Revision 1.28  2003/11/05 15:38:14  kuznets
 * Added test for new GetEntries()
 *
 * Revision 1.27  2003/10/08 15:45:53  ivanov
 * Added tests for [Add|Delete]TrailingPathSeparator()
 *
 * Revision 1.26  2003/09/30 15:09:05  ucko
 * Reworked CDirEntry::NormalizePath, which now handles .. correctly in
 * all cases and optionally resolves symlinks (on Unix).
 *
 * Revision 1.25  2003/09/16 18:55:16  ivanov
 * Added more test for NormalizePath()
 *
 * Revision 1.24  2003/09/16 16:28:08  ivanov
 * Added tests for SplitPath(). Minor cosmetics.
 *
 * Revision 1.23  2003/08/08 14:30:06  meric
 * Accommodate rename of CFile::GetTmpNameExt() to CFile::GetTmpNameEx()
 *
 * Revision 1.22  2003/03/10 18:57:08  kuznets
 * iterate->ITERATE
 *
 * Revision 1.21  2003/03/06 21:22:30  ivanov
 * Rollback to R1.19 -- accidentally commit work version
 *
 * Revision 1.20  2003/03/06 21:12:35  ivanov
 * *** empty log message ***
 *
 * Revision 1.19  2003/02/14 19:26:51  ivanov
 * Added more checks for CreateTmpFile()
 *
 * Revision 1.18  2003/02/05 22:08:26  ivanov
 * Enlarged CMemoryFile test
 *
 * Revision 1.17  2002/06/29 06:45:50  vakatov
 * Get rid of some compilation warnings
 *
 * Revision 1.16  2002/06/07 16:11:38  ivanov
 * Chenget GetTime(): using CTime instead time_t, modification time by default
 *
 * Revision 1.15  2002/06/07 15:21:33  ivanov
 * Added CDirEntry::GetTime() test example
 *
 * Revision 1.14  2002/04/16 18:49:08  ivanov
 * Centralize threatment of assert() in tests.
 * Added #include <test/test_assert.h>. CVS log moved to end of file.
 *
 * Revision 1.13  2002/04/01 18:50:18  ivanov
 * Added test for class CMemoryFile
 *
 * Revision 1.12  2002/03/25 17:08:19  ucko
 * Centralize treatment of Cygwin as Unix rather than Windows in configure.
 *
 * Revision 1.11  2002/03/22 20:00:24  ucko
 * Tweak to work on Cygwin.  (Use Unix rather than MSWIN code).
 *
 * Revision 1.10  2002/03/01 16:06:24  kans
 * removed erroneous __MACOS__ section
 *
 * Revision 1.9  2002/01/24 22:18:52  ivanov
 * Changed tests for CDirEntry::Remove() and CDir::Remove()
 *
 * Revision 1.8  2002/01/22 19:29:09  ivanov
 * Added test for ConcatPathEx()
 *
 * Revision 1.7  2002/01/21 04:48:15  vakatov
 * Use #_MSC_VER instead of #NCBI_OS_MSWIN
 *
 * Revision 1.6  2002/01/20 04:42:22  vakatov
 * Do not #define _DEBUG on NCBI_OS_MSWIN
 *
 * Revision 1.5  2002/01/10 16:48:06  ivanov
 * Added test s_TEST_CheckPath()
 *
 * Revision 1.4  2001/11/19 18:13:10  juran
 * Add s_TEST_MatchesMask().
 * Use GetEntries() instead of Contents().
 *
 * Revision 1.3  2001/11/15 16:36:29  ivanov
 * Moved from util to corelib
 *
 * Revision 1.2  2001/11/01 20:06:50  juran
 * Replace directory streams with Contents() method.
 * Implement and test Mac OS platform.
 *
 * Revision 1.1  2001/09/19 13:08:15  ivanov
 * Initial revision
 *
 * ===========================================================================
 */