63 lines
2.2 KiB
Zig
63 lines
2.2 KiB
Zig
const std = @import("index.zig");
|
|
const io = std.io;
|
|
const DefaultPrng = std.rand.DefaultPrng;
|
|
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" {
|
|
var raw_bytes: [200 * 1024]u8 = undefined;
|
|
var allocator = &std.heap.FixedBufferAllocator.init(raw_bytes[0..]).allocator;
|
|
|
|
var data: [1024]u8 = undefined;
|
|
var prng = DefaultPrng.init(1234);
|
|
prng.random.bytes(data[0..]);
|
|
const tmp_file_name = "temp_test_file.txt";
|
|
{
|
|
var file = try os.File.openWrite(allocator, tmp_file_name);
|
|
defer file.close();
|
|
|
|
var file_out_stream = io.FileOutStream.init(&file);
|
|
var buf_stream = io.BufferedOutStream(io.FileOutStream.Error).init(&file_out_stream.stream);
|
|
const st = &buf_stream.stream;
|
|
try st.print("begin");
|
|
try st.write(data[0..]);
|
|
try st.print("end");
|
|
try buf_stream.flush();
|
|
}
|
|
{
|
|
var file = try os.File.openRead(allocator, tmp_file_name);
|
|
defer file.close();
|
|
|
|
const file_size = try 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(io.FileInStream.Error).init(&file_in_stream.stream);
|
|
const st = &buf_stream.stream;
|
|
const contents = try 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"));
|
|
}
|
|
try os.deleteFile(allocator, tmp_file_name);
|
|
}
|
|
|
|
test "BufferOutStream" {
|
|
var bytes: [100]u8 = undefined;
|
|
var allocator = &std.heap.FixedBufferAllocator.init(bytes[0..]).allocator;
|
|
|
|
var buffer = try std.Buffer.initSize(allocator, 0);
|
|
var buf_stream = &std.io.BufferOutStream.init(&buffer).stream;
|
|
|
|
const x: i32 = 42;
|
|
const y: i32 = 1234;
|
|
try buf_stream.print("x: {}\ny: {}\n", x, y);
|
|
|
|
assert(mem.eql(u8, buffer.toSlice(), "x: 42\ny: 1234\n"));
|
|
}
|