mirror of
https://github.com/vlang/v.git
synced 2023-08-10 21:13:21 +03:00
151 lines
4.2 KiB
V
151 lines
4.2 KiB
V
import encoding.base64
|
|
|
|
struct TestPair {
|
|
decoded string
|
|
encoded string
|
|
}
|
|
|
|
const (
|
|
pairs = [
|
|
// RFC 3548 examples
|
|
TestPair{'\x14\xfb\x9c\x03\xd9\x7e', 'FPucA9l+'},
|
|
TestPair{'\x14\xfb\x9c\x03\xd9', 'FPucA9k='},
|
|
TestPair{'\x14\xfb\x9c\x03', 'FPucAw=='},
|
|
// RFC 4648 examples
|
|
TestPair{'', ''},
|
|
TestPair{'f', 'Zg=='},
|
|
TestPair{'fo', 'Zm8='},
|
|
TestPair{'foo', 'Zm9v'},
|
|
TestPair{'foob', 'Zm9vYg=='},
|
|
TestPair{'fooba', 'Zm9vYmE='},
|
|
TestPair{'foobar', 'Zm9vYmFy'},
|
|
// Wikipedia examples
|
|
TestPair{'sure.', 'c3VyZS4='},
|
|
TestPair{'sure', 'c3VyZQ=='},
|
|
TestPair{'sur', 'c3Vy'},
|
|
TestPair{'su', 'c3U='},
|
|
TestPair{'leasure.', 'bGVhc3VyZS4='},
|
|
TestPair{'easure.', 'ZWFzdXJlLg=='},
|
|
TestPair{'asure.', 'YXN1cmUu'},
|
|
TestPair{'sure.', 'c3VyZS4='},
|
|
]
|
|
|
|
man_pair = TestPair{'Man is distinguished, not only by his reason, but by this singular passion from other animals, which is a lust of the mind, that by a perseverance of delight in the continued and indefatigable generation of knowledge, exceeds the short vehemence of any carnal pleasure.', 'TWFuIGlzIGRpc3Rpbmd1aXNoZWQsIG5vdCBvbmx5IGJ5IGhpcyByZWFzb24sIGJ1dCBieSB0aGlzIHNpbmd1bGFyIHBhc3Npb24gZnJvbSBvdGhlciBhbmltYWxzLCB3aGljaCBpcyBhIGx1c3Qgb2YgdGhlIG1pbmQsIHRoYXQgYnkgYSBwZXJzZXZlcmFuY2Ugb2YgZGVsaWdodCBpbiB0aGUgY29udGludWVkIGFuZCBpbmRlZmF0aWdhYmxlIGdlbmVyYXRpb24gb2Yga25vd2xlZGdlLCBleGNlZWRzIHRoZSBzaG9ydCB2ZWhlbWVuY2Ugb2YgYW55IGNhcm5hbCBwbGVhc3VyZS4='}
|
|
)
|
|
|
|
fn test_decode() {
|
|
assert base64.decode(man_pair.encoded) == man_pair.decoded.bytes()
|
|
|
|
// Test for incorrect padding.
|
|
assert base64.decode('aGk') == ''.bytes()
|
|
assert base64.decode('aGk=') == 'hi'.bytes()
|
|
assert base64.decode('aGk==') == ''.bytes()
|
|
|
|
for i, p in pairs {
|
|
got := base64.decode(p.encoded)
|
|
if got != p.decoded.bytes() {
|
|
eprintln('pairs[$i]: expected = $p.decoded, got = $got')
|
|
assert false
|
|
}
|
|
}
|
|
}
|
|
|
|
fn test_decode_str() {
|
|
assert base64.decode_str(man_pair.encoded) == man_pair.decoded
|
|
|
|
// Test for incorrect padding.
|
|
assert base64.decode_str('aGk') == ''
|
|
assert base64.decode_str('aGk=') == 'hi'
|
|
assert base64.decode_str('aGk==') == ''
|
|
|
|
for i, p in pairs {
|
|
got := base64.decode_str(p.encoded)
|
|
if got != p.decoded {
|
|
eprintln('pairs[$i]: expected = $p.decoded, got = $got')
|
|
assert false
|
|
}
|
|
}
|
|
}
|
|
|
|
fn test_encode() {
|
|
assert base64.encode(man_pair.decoded.bytes()) == man_pair.encoded
|
|
|
|
for i, p in pairs {
|
|
got := base64.encode(p.decoded.bytes())
|
|
if got != p.encoded {
|
|
eprintln('pairs[$i]: expected = $p.encoded, got = $got')
|
|
assert false
|
|
}
|
|
}
|
|
}
|
|
|
|
fn test_encode_str() {
|
|
assert base64.encode_str(man_pair.decoded) == man_pair.encoded
|
|
|
|
for i, p in pairs {
|
|
got := base64.encode_str(p.decoded)
|
|
if got != p.encoded {
|
|
eprintln('pairs[$i]: expected = $p.encoded, got = $got')
|
|
assert false
|
|
}
|
|
}
|
|
}
|
|
|
|
fn test_url_encode() {
|
|
test := base64.url_encode('Hello Base64Url encoding!'.bytes())
|
|
assert test == 'SGVsbG8gQmFzZTY0VXJsIGVuY29kaW5nIQ'
|
|
}
|
|
|
|
fn test_url_encode_str() {
|
|
test := base64.url_encode_str('Hello Base64Url encoding!')
|
|
assert test == 'SGVsbG8gQmFzZTY0VXJsIGVuY29kaW5nIQ'
|
|
}
|
|
|
|
fn test_url_decode() {
|
|
test := base64.url_decode('SGVsbG8gQmFzZTY0VXJsIGVuY29kaW5nIQ')
|
|
assert test == 'Hello Base64Url encoding!'.bytes()
|
|
}
|
|
|
|
fn test_url_decode_str() {
|
|
test := base64.url_decode_str('SGVsbG8gQmFzZTY0VXJsIGVuY29kaW5nIQ')
|
|
assert test == 'Hello Base64Url encoding!'
|
|
}
|
|
|
|
fn test_encode_null_u8() {
|
|
assert base64.encode([u8(`A`), 0, `C`]) == 'QQBD'
|
|
}
|
|
|
|
fn test_encode_null_byte_str() {
|
|
// While this works, bytestr() does a memcpy
|
|
s := [u8(`A`), 0, `C`].bytestr()
|
|
assert base64.encode_str(s) == 'QQBD'
|
|
}
|
|
|
|
fn test_decode_null_u8() {
|
|
assert base64.decode('QQBD') == [u8(`A`), 0, `C`]
|
|
}
|
|
|
|
fn test_decode_null_byte_str() {
|
|
// While this works, bytestr() does a memcpy
|
|
s := [u8(`A`), 0, `C`].bytestr()
|
|
assert base64.decode_str('QQBD') == s
|
|
}
|
|
|
|
fn test_decode_in_buffer_bytes() {
|
|
rfc4648_pairs := [
|
|
TestPair{'foob', 'Zm9vYg=='},
|
|
TestPair{'fooba', 'Zm9vYmE='},
|
|
TestPair{'foobar', 'Zm9vYmFy'},
|
|
]
|
|
mut src_dec_buf := []byte{len: 8}
|
|
mut src_enc_buf := []byte{len: 8}
|
|
mut out_buf := []byte{len: 8}
|
|
|
|
for p in rfc4648_pairs {
|
|
src_dec_buf = p.decoded.bytes()
|
|
src_enc_buf = p.encoded.bytes()
|
|
n := base64.decode_in_buffer_bytes(src_enc_buf, out_buf.data)
|
|
assert src_dec_buf == out_buf[..n]
|
|
}
|
|
}
|