overwrite.cpp
上传用户:yhdzpy8989
上传日期:2007-06-13
资源大小:13604k
文件大小:4k
源码类别:

生物技术

开发平台:

C/C++

  1. /*
  2.  * ===========================================================================
  3.  * PRODUCTION $Log: overwrite.cpp,v $
  4.  * PRODUCTION Revision 1000.1  2004/06/01 19:09:40  gouriano
  5.  * PRODUCTION PRODUCTION: UPGRADED [GCC34_MSVC7] Dev-tree R6.4
  6.  * PRODUCTION
  7.  * ===========================================================================
  8.  */
  9. /*  $Id: overwrite.cpp,v 1000.1 2004/06/01 19:09:40 gouriano Exp $
  10.  * ===========================================================================
  11.  *
  12.  *                            PUBLIC DOMAIN NOTICE
  13.  *               National Center for Biotechnology Information
  14.  *
  15.  *  This software/database is a "United States Government Work" under the
  16.  *  terms of the United States Copyright Act.  It was written as part of
  17.  *  the author's official duties as a United States Government employee and
  18.  *  thus cannot be copyrighted.  This software/database is freely available
  19.  *  to the public for use. The National Library of Medicine and the U.S.
  20.  *  Government have not placed any restriction on its use or reproduction.
  21.  *
  22.  *  Although all reasonable efforts have been taken to ensure the accuracy
  23.  *  and reliability of the software and data, the NLM and the U.S.
  24.  *  Government do not and cannot warrant the performance or results that
  25.  *  may be obtained by using this software or data. The NLM and the U.S.
  26.  *  Government disclaim all warranties, express or implied, including
  27.  *  warranties of performance, merchantability or fitness for any particular
  28.  *  purpose.
  29.  *
  30.  *  Please cite the author in any work or product based on this material.
  31.  *
  32.  * ===========================================================================
  33.  *
  34.  * Author:  Vasilchenko
  35.  *
  36.  * File Description:
  37.  *
  38.  */
  39. #include <ncbi_pch.hpp>
  40. #include <corelib/ncbistd.hpp>
  41. #include <corelib/ncbistre.hpp>
  42. #if NCBI_OS_MSWIN
  43. # include <stdio.h>
  44. #else
  45. # include <unistd.h>
  46. #endif
  47. #include <sys/types.h>
  48. #include <sys/stat.h>
  49. #include <errno.h>
  50. #include <test/test_assert.h>  /* This header must go last */
  51. USING_NCBI_SCOPE;
  52. void Remove(const string& file)
  53. {
  54.     unlink(file.c_str());
  55. }
  56. bool Exists(const string& file)
  57. {
  58.     struct stat st;
  59.     if ( stat(file.c_str(), &st) == 0 )
  60.         return true;
  61.     if ( errno == ENOENT )
  62.         return false;
  63.     ERR_POST("Unexpected errno after stat(): " << errno);
  64.     return false;
  65. }
  66. long FileSize(const string& file)
  67. {
  68.     struct stat st;
  69.     if ( stat(file.c_str(), &st) == 0 )
  70.         return st.st_size;
  71.     ERR_POST("Cannot get size of file: " << errno);
  72.     return 0;
  73. }
  74. int main()
  75. {
  76.     SetDiagStream(&NcbiCerr);
  77.     SetDiagPostLevel(eDiag_Info);
  78.     SetDiagPostPrefix("test_overwrite");
  79.     
  80.     string fileName = "test_file.tmp";
  81.     Remove(fileName);
  82.     if ( Exists(fileName) ) {
  83.         ERR_POST("cannot remove " << fileName);
  84.         return 1;
  85.     }
  86.     {
  87.         // write data
  88.         CNcbiOfstream o(fileName.c_str(), IOS_BASE::app);
  89.         if ( !o ) {
  90.             ERR_POST("Cannot create file " << fileName);
  91.             return 1;
  92.         }
  93.         o.seekp(0, IOS_BASE::end);
  94.         if ( streampos(o.tellp()) != streampos(0) ) {
  95.             ERR_POST("New file returns non zero size" << streampos(o.tellp()));
  96.             return 1;
  97.         }
  98.         o << "TEST";
  99.     }
  100.     if ( FileSize(fileName) != 4 ) {
  101.         ERR_POST("Wrong file size after write" << FileSize(fileName));
  102.         return 1;
  103.     }
  104.     {
  105.         // try to overwrite data
  106.         CNcbiOfstream o(fileName.c_str(), IOS_BASE::app);
  107.         if ( !o ) {
  108.             ERR_POST("Cannot open file " << fileName);
  109.             return 1;
  110.         }
  111.         o.seekp(0, IOS_BASE::end);
  112.         if ( streampos(o.tellp()) == streampos(0) ) {
  113.             ERR_POST("Non empty file returns zero size");
  114.             return 1;
  115.         }
  116.     }
  117.     if ( FileSize(fileName) != 4 ) {
  118.         ERR_POST("File size was changed after test" << FileSize(fileName));
  119.         return 1;
  120.     }
  121.     Remove(fileName);
  122.     ERR_POST(Info << "Test passed successfully");
  123.     return 0;
  124. }
  125. /*
  126.  * ===========================================================================
  127.  * $Log: overwrite.cpp,v $
  128.  * Revision 1000.1  2004/06/01 19:09:40  gouriano
  129.  * PRODUCTION: UPGRADED [GCC34_MSVC7] Dev-tree R6.4
  130.  *
  131.  * Revision 6.4  2004/05/14 13:59:51  gorelenk
  132.  * Added include of ncbi_pch.hpp
  133.  *
  134.  * Revision 6.3  2002/04/16 18:49:06  ivanov
  135.  * Centralize threatment of assert() in tests.
  136.  * Added #include <test/test_assert.h>. CVS log moved to end of file.
  137.  *
  138.  * ==========================================================================
  139.  */