std/io_test.zig


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

const std = @import("index.zig");
const io = std.io;
const allocator = std.debug.global_allocator;
const Rand = std.rand.Rand;
const assert = std.debug.assert;
const mem = std.mem;
const os = std.os;
const builtin = @import("builtin");

test "write a file, read it, then delete it" {
    if (builtin.os == builtin.Os.windows and builtin.arch == builtin.Arch.i386) {
        // TODO get this test passing
        // https://github.com/zig-lang/zig/issues/537
        return;
    }
    var data: [1024]u8 = undefined;
    var rng = Rand.init(1234);
    rng.fillBytes(data[0..]);
    const tmp_file_name = "temp_test_file.txt";
    {
        var file = %%io.File.openWrite(tmp_file_name, allocator);
        defer file.close();

        var file_out_stream = io.FileOutStream.init(&file);
        var buf_stream = io.BufferedOutStream.init(&file_out_stream.stream);
        const st = &buf_stream.stream;
        %%st.print("begin");
        %%st.write(data[0..]);
        %%st.print("end");
        %%buf_stream.flush();
    }
    {
        var file = %%io.File.openRead(tmp_file_name, allocator);
        defer file.close();

        const file_size = %%file.getEndPos();
        const expected_file_size = "begin".len + data.len + "end".len;
        assert(file_size == expected_file_size);

        var file_in_stream = io.FileInStream.init(&file);
        var buf_stream = io.BufferedInStream.init(&file_in_stream.stream);
        const st = &buf_stream.stream;
        const contents = %%st.readAllAlloc(allocator, 2 * 1024);
        defer allocator.free(contents);

        assert(mem.eql(u8, contents[0.."begin".len], "begin"));
        assert(mem.eql(u8, contents["begin".len..contents.len - "end".len], data));
        assert(mem.eql(u8, contents[contents.len - "end".len ..], "end"));
    }
    %%os.deleteFile(allocator, tmp_file_name);
}