summaryrefslogtreecommitdiff
path: root/src/unit/file.unit
blob: b5d5b1a6d23480e5f6197db9f5da8303e17c85e8 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
// vim: syntax=cpp
/*
 * Copyright (C) 2007-2013 Xagasoft, All rights reserved.
 *
 * This file is part of the libbu++ library and is released under the
 * terms of the license contained in the file LICENSE.
 */

#include "bu/file.h"

#include <sys/types.h>
#include <sys/stat.h>
#include <unistd.h>

suite File
{
    test writeFull
    {
        Bu::File sf("testfile1", Bu::File::WriteNew );
        for( int c = 0; c < 256; c++ )
        {
            unsigned char ch = (unsigned char)c;
            sf.write( &ch, 1 );
            unitTest( sf.tell() == c+1 );
        }
        //unitTest( sf.canRead() == false );
        //unitTest( sf.canWrite() == true );
        //unitTest( sf.canSeek() == true );
        sf.close();
        struct stat sdat;
        stat("testfile1", &sdat );
        unitTest( sdat.st_size == 256 );
    }

    test readBlocks
    {
        Bu::File sf("testfile1", Bu::File::Read );
        unsigned char buf[50];
        size_t total = 0;
        for(;;)
        {
            size_t s = sf.read( buf, 50 );
            for( size_t c = 0; c < s; c++ )
            {
                unitTest( buf[c] == (unsigned char)(c+total) );
            }
            total += s;
            if( s < 50 )
            {
                unitTest( total == 256 );
                unitTest( sf.isEos() == false );
                unitTest( sf.read( buf, 1 ) == 0 );
                unitTest( sf.isEos() == true );
                // EOS is only set when an attempt to read past the end of
                // the stream is made, not when a short block is returned.
                break;
            }
        }
        sf.close();
    }

    test readError1
    {
        try
        {
            Bu::File sf("doesn'texist", Bu::File::Read );
            unitFailed("No exception thrown");
        }
        catch( Bu::FileException &e )
        {
            return;
        }
    }

    test readError2
    {
        Bu::File sf("testfile1", Bu::File::Read );
        char buf[256];
        int r = sf.read( buf, 256 );
        unitTest( r == 256 );
        // You have to read past the end to set the EOS flag.
        unitTest( sf.isEos() == false );
        try
        {
            if( sf.read( buf, 5 ) > 0 )
            {
                unitFailed("Non-zero read result");
            }
            else
            {
                sf.close();
            }
        }
        catch( Bu::FileException &e )
        {
            sf.close();
            return;
        }
    }
}