mirror of
https://github.com/vlang/v.git
synced 2023-08-10 21:13:21 +03:00
422 lines
10 KiB
V
422 lines
10 KiB
V
import os
|
|
|
|
const tfolder = os.join_path(os.vtmp_dir(), 'v', 'tests', 'os_file_test')
|
|
|
|
const tfile = os.join_path_single(tfolder, 'test_file')
|
|
|
|
fn testsuite_begin() {
|
|
os.rmdir_all(tfolder) or {}
|
|
assert !os.is_dir(tfolder)
|
|
os.mkdir_all(tfolder)!
|
|
os.chdir(tfolder)!
|
|
assert os.is_dir(tfolder)
|
|
}
|
|
|
|
fn testsuite_end() {
|
|
os.rmdir_all(tfolder) or {}
|
|
}
|
|
|
|
struct Point {
|
|
x f64
|
|
y f64
|
|
z f64
|
|
}
|
|
|
|
struct Extended_Point {
|
|
a f64
|
|
b f64
|
|
c f64
|
|
d f64
|
|
e f64
|
|
f f64
|
|
g f64
|
|
h f64
|
|
i f64
|
|
}
|
|
|
|
enum Color {
|
|
red
|
|
green
|
|
blue
|
|
}
|
|
|
|
[flag]
|
|
enum Permissions {
|
|
read
|
|
write
|
|
execute
|
|
}
|
|
|
|
const (
|
|
unit_point = Point{1.0, 1.0, 1.0}
|
|
another_point = Point{0.25, 2.25, 6.25}
|
|
extended_point = Extended_Point{1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0}
|
|
another_byte = u8(123)
|
|
another_color = Color.red
|
|
another_permission = Permissions.read | .write
|
|
)
|
|
|
|
// test_read_bytes_into_newline_text tests reading text from a file with newlines.
|
|
// This test simulates reading a larger text file step by step into a buffer and
|
|
// returning on each newline, even before the buffer is full, and reaching EOF before
|
|
// the buffer is completely filled.
|
|
fn test_read_bytes_into_newline_text() {
|
|
mut f := os.open_file(tfile, 'w')!
|
|
f.write_string('Hello World!\nGood\r morning.')!
|
|
f.close()
|
|
|
|
f = os.open_file(tfile, 'r')!
|
|
mut buf := []u8{len: 8}
|
|
|
|
n0 := f.read_bytes_into_newline(mut buf)!
|
|
assert n0 == 8
|
|
|
|
n1 := f.read_bytes_into_newline(mut buf)!
|
|
assert n1 == 5
|
|
|
|
n2 := f.read_bytes_into_newline(mut buf)!
|
|
assert n2 == 8
|
|
|
|
n3 := f.read_bytes_into_newline(mut buf)!
|
|
assert n3 == 6
|
|
|
|
f.close()
|
|
}
|
|
|
|
// test_read_bytes_into_newline_binary tests reading a binary file with NUL bytes.
|
|
// This test simulates the scenario when a byte stream is read and a newline byte
|
|
// appears in that stream and an EOF occurs before the buffer is full.
|
|
fn test_read_bytes_into_newline_binary() {
|
|
os.rm(tfile) or {} // FIXME This is a workaround for macos, because the file isn't truncated when open with 'w'
|
|
mut bw := []u8{len: 15}
|
|
bw[9] = 0xff
|
|
bw[12] = 10 // newline
|
|
|
|
n0_bytes := bw[0..10]
|
|
n1_bytes := bw[10..13]
|
|
n2_bytes := bw[13..]
|
|
|
|
mut f := os.open_file(tfile, 'w')!
|
|
f.write(bw)!
|
|
f.close()
|
|
|
|
f = os.open_file(tfile, 'r')!
|
|
mut buf := []u8{len: 10}
|
|
|
|
n0 := f.read_bytes_into_newline(mut buf)!
|
|
assert n0 == 10
|
|
assert buf[..n0] == n0_bytes
|
|
|
|
n1 := f.read_bytes_into_newline(mut buf)!
|
|
assert n1 == 3
|
|
assert buf[..n1] == n1_bytes
|
|
|
|
n2 := f.read_bytes_into_newline(mut buf)!
|
|
assert n2 == 2
|
|
assert buf[..n2] == n2_bytes
|
|
f.close()
|
|
}
|
|
|
|
// test_read_eof_last_read_partial_buffer_fill tests that when reading a file
|
|
// the end-of-file is detected and results in a none error being returned. This
|
|
// test simulates file reading where the end-of-file is reached inside an fread
|
|
// containing data.
|
|
fn test_read_eof_last_read_partial_buffer_fill() {
|
|
mut f := os.open_file(tfile, 'w')!
|
|
bw := []u8{len: 199, init: 5}
|
|
f.write(bw)!
|
|
f.close()
|
|
|
|
f = os.open_file(tfile, 'r')!
|
|
mut br := []u8{len: 100}
|
|
// Read first 100 bytes of 199 byte file, should fill buffer with no error.
|
|
n0 := f.read(mut br)!
|
|
assert n0 == 100
|
|
// Read remaining 99 bytes of 199 byte file, should fill buffer with no
|
|
// error, even though end-of-file was reached.
|
|
n1 := f.read(mut br)!
|
|
assert n1 == 99
|
|
// Read again, end-of-file was previously reached so should return none
|
|
// error.
|
|
if _ := f.read(mut br) {
|
|
// This is not intended behavior because the read function should
|
|
// not return a number of bytes read when end-of-file is reached.
|
|
assert false
|
|
} else {
|
|
// Expected an error when received end-of-file.
|
|
assert err is os.Eof
|
|
}
|
|
f.close()
|
|
}
|
|
|
|
// test_read_eof_last_read_full_buffer_fill tests that when reading a file the
|
|
// end-of-file is detected and results in a none error being returned. This test
|
|
// simulates file reading where the end-of-file is reached at the beinning of an
|
|
// fread that returns no data.
|
|
fn test_read_eof_last_read_full_buffer_fill() {
|
|
mut f := os.open_file(tfile, 'w')!
|
|
bw := []u8{len: 200, init: 5}
|
|
f.write(bw)!
|
|
f.close()
|
|
|
|
f = os.open_file(tfile, 'r')!
|
|
mut br := []u8{len: 100}
|
|
// Read first 100 bytes of 200 byte file, should fill buffer with no error.
|
|
n0 := f.read(mut br)!
|
|
assert n0 == 100
|
|
// Read remaining 100 bytes of 200 byte file, should fill buffer with no
|
|
// error. The end-of-file isn't reached yet, but there is no more data.
|
|
n1 := f.read(mut br)!
|
|
assert n1 == 100
|
|
// Read again, end-of-file was previously reached so should return none
|
|
// error.
|
|
if _ := f.read(mut br) {
|
|
// This is not intended behavior because the read function should
|
|
// not return a number of bytes read when end-of-file is reached.
|
|
assert false
|
|
} else {
|
|
// Expect an error at EOF.
|
|
assert err is os.Eof
|
|
}
|
|
f.close()
|
|
}
|
|
|
|
fn test_write_struct() {
|
|
os.rm(tfile) or {} // FIXME This is a workaround for macos, because the file isn't truncated when open with 'w'
|
|
size_of_point := int(sizeof(Point))
|
|
mut f := os.open_file(tfile, 'w')!
|
|
f.write_struct(another_point)!
|
|
f.close()
|
|
x := os.read_file(tfile)!
|
|
pcopy := unsafe { &u8(memdup(&another_point, size_of_point)) }
|
|
y := unsafe { pcopy.vstring_with_len(size_of_point) }
|
|
assert x == y
|
|
$if debug {
|
|
eprintln(x.bytes())
|
|
eprintln(y.bytes())
|
|
}
|
|
}
|
|
|
|
fn test_write_struct_at() {
|
|
mut f := os.open_file(tfile, 'w')!
|
|
f.write_struct(extended_point)!
|
|
f.write_struct_at(another_point, 3)!
|
|
f.close()
|
|
f = os.open_file(tfile, 'r')!
|
|
mut p := Point{}
|
|
f.read_struct_at(mut p, 3)!
|
|
f.close()
|
|
|
|
assert p == another_point
|
|
}
|
|
|
|
fn test_read_struct() {
|
|
mut f := os.open_file(tfile, 'w')!
|
|
f.write_struct(another_point)!
|
|
f.close()
|
|
|
|
f = os.open_file(tfile, 'r')!
|
|
mut p := Point{}
|
|
f.read_struct(mut p)!
|
|
f.close()
|
|
|
|
assert p == another_point
|
|
}
|
|
|
|
fn test_read_struct_at() {
|
|
mut f := os.open_file(tfile, 'w')!
|
|
f.write([u8(1), 2, 3])!
|
|
f.write_struct(another_point)!
|
|
f.close()
|
|
f = os.open_file(tfile, 'r')!
|
|
mut p := Point{}
|
|
f.read_struct_at(mut p, 3)!
|
|
f.close()
|
|
|
|
assert p == another_point
|
|
}
|
|
|
|
fn test_write_raw() {
|
|
os.rm(tfile) or {} // FIXME This is a workaround for macos, because the file isn't truncated when open with 'w'
|
|
size_of_point := int(sizeof(Point))
|
|
mut f := os.open_file(tfile, 'w')!
|
|
f.write_raw(another_point)!
|
|
f.close()
|
|
x := os.read_file(tfile)!
|
|
pcopy := unsafe { &u8(memdup(&another_point, size_of_point)) }
|
|
y := unsafe { pcopy.vstring_with_len(size_of_point) }
|
|
assert x == y
|
|
$if debug {
|
|
eprintln(x.bytes())
|
|
eprintln(y.bytes())
|
|
}
|
|
}
|
|
|
|
fn test_write_raw_at() {
|
|
mut f := os.open_file(tfile, 'w')!
|
|
f.write_raw(extended_point)!
|
|
f.write_raw_at(another_point, 3)!
|
|
f.close()
|
|
f = os.open_file(tfile, 'r')!
|
|
mut p := Point{}
|
|
f.read_struct_at(mut p, 3)!
|
|
f.close()
|
|
|
|
assert p == another_point
|
|
}
|
|
|
|
fn test_write_raw_at_negative_pos() {
|
|
mut f := os.open_file(tfile, 'w')!
|
|
if _ := f.write_raw_at(another_point, u64(-1)) {
|
|
assert false
|
|
}
|
|
f.write_raw_at(another_point, u64(-1)) or { assert err.msg() == 'Invalid argument' }
|
|
f.close()
|
|
}
|
|
|
|
fn test_read_raw() {
|
|
mut f := os.open_file(tfile, 'w')!
|
|
f.write_raw(another_point)!
|
|
f.write_raw(another_byte)!
|
|
f.write_raw(another_color)!
|
|
f.write_raw(another_permission)!
|
|
f.close()
|
|
f = os.open_file(tfile, 'r')!
|
|
p := f.read_raw[Point]()!
|
|
b := f.read_raw[u8]()!
|
|
c := f.read_raw[Color]()!
|
|
x := f.read_raw[Permissions]()!
|
|
f.close()
|
|
|
|
assert p == another_point
|
|
assert b == another_byte
|
|
assert c == another_color
|
|
assert x == another_permission
|
|
}
|
|
|
|
fn test_read_raw_at() {
|
|
mut f := os.open_file(tfile, 'w')!
|
|
f.write([u8(1), 2, 3])!
|
|
f.write_raw(another_point)!
|
|
f.write_raw(another_byte)!
|
|
f.write_raw(another_color)!
|
|
f.write_raw(another_permission)!
|
|
f.close()
|
|
f = os.open_file(tfile, 'r')!
|
|
mut at := u64(3)
|
|
p := f.read_raw_at[Point](at)!
|
|
at += sizeof(Point)
|
|
b := f.read_raw_at[u8](at)!
|
|
at += sizeof(u8)
|
|
c := f.read_raw_at[Color](at)!
|
|
at += sizeof(Color)
|
|
x := f.read_raw_at[Permissions](at)!
|
|
at += sizeof(Permissions)
|
|
f.close()
|
|
|
|
assert p == another_point
|
|
assert b == another_byte
|
|
assert c == another_color
|
|
assert x == another_permission
|
|
}
|
|
|
|
fn test_read_raw_at_negative_pos() {
|
|
mut f := os.open_file(tfile, 'r')!
|
|
if _ := f.read_raw_at[Point](u64(-1)) {
|
|
assert false
|
|
}
|
|
f.read_raw_at[Point](u64(-1)) or { assert err.msg() == 'Invalid argument' }
|
|
f.close()
|
|
}
|
|
|
|
fn test_seek() {
|
|
mut f := os.open_file(tfile, 'w')!
|
|
f.write_raw(another_point)!
|
|
f.write_raw(another_byte)!
|
|
f.write_raw(another_color)!
|
|
f.write_raw(another_permission)!
|
|
f.close()
|
|
|
|
// println('> ${sizeof(Point)} ${sizeof(byte)} ${sizeof(Color)} ${sizeof(Permissions)}')
|
|
f = os.open_file(tfile, 'r')!
|
|
//
|
|
f.seek(i64(sizeof(Point)), .start)!
|
|
assert f.tell()! == sizeof(Point)
|
|
b := f.read_raw[u8]()!
|
|
assert b == another_byte
|
|
|
|
f.seek(i64(sizeof(Color)), .current)!
|
|
x := f.read_raw[Permissions]()!
|
|
assert x == another_permission
|
|
//
|
|
f.close()
|
|
}
|
|
|
|
fn test_tell() {
|
|
for size in 10 .. 30 {
|
|
s := 'x'.repeat(size)
|
|
os.write_file(tfile, s)!
|
|
fs := os.file_size(tfile)
|
|
assert int(fs) == size
|
|
//
|
|
mut f := os.open_file(tfile, 'r')!
|
|
f.seek(-5, .end)!
|
|
pos := f.tell()!
|
|
f.close()
|
|
// dump(pos)
|
|
assert pos == size - 5
|
|
}
|
|
}
|
|
|
|
fn test_reopen() {
|
|
tfile1 := os.join_path_single(tfolder, 'tfile1')
|
|
tfile2 := os.join_path_single(tfolder, 'tfile2')
|
|
os.write_file(tfile1, 'Hello World!\nGood\r morning.\nBye 1.')!
|
|
os.write_file(tfile2, 'Another file\nAnother line.\nBye 2.')!
|
|
assert os.file_size(tfile1) > 0
|
|
assert os.file_size(tfile2) > 0
|
|
|
|
mut line_buffer := []u8{len: 1024}
|
|
|
|
mut f2 := os.open(tfile2)!
|
|
x := f2.read_bytes_into_newline(mut line_buffer)!
|
|
assert !f2.eof()
|
|
assert x > 0
|
|
assert line_buffer#[..x].bytestr() == 'Another file\n'
|
|
|
|
// Note: after this call, f2 should be using the file `tfile1`:
|
|
f2.reopen(tfile1, 'r')!
|
|
assert !f2.eof()
|
|
|
|
z := f2.read(mut line_buffer) or { panic(err) }
|
|
assert f2.eof()
|
|
assert z > 0
|
|
content := line_buffer#[..z].bytestr()
|
|
// dump(content)
|
|
assert content.starts_with('Hello World')
|
|
assert content.ends_with('Bye 1.')
|
|
}
|
|
|
|
fn test_eof() {
|
|
os.write_file(tfile, 'Hello World!\n')!
|
|
|
|
mut f := os.open(tfile)!
|
|
f.read_bytes(10)
|
|
assert !f.eof()
|
|
f.read_bytes(100)
|
|
assert f.eof()
|
|
}
|
|
|
|
fn test_open_file_wb_ab() {
|
|
os.rm(tfile) or {}
|
|
mut wfile := os.open_file('text.txt', 'wb', 0o666)!
|
|
wfile.write_string('hello')!
|
|
wfile.close()
|
|
assert os.read_file('text.txt')! == 'hello'
|
|
//
|
|
mut afile := os.open_file('text.txt', 'ab', 0o666)!
|
|
afile.write_string('hello')!
|
|
afile.close()
|
|
assert os.read_file('text.txt')? == 'hellohello'
|
|
}
|