1
0
mirror of https://github.com/vlang/v.git synced 2023-08-10 21:13:21 +03:00
v/vlib/builtin/string_test.v

689 lines
14 KiB
V
Raw Normal View History

2020-02-03 07:00:36 +03:00
// Copyright (c) 2019-2020 Alexander Medvednikov. All rights reserved.
2019-06-23 05:21:30 +03:00
// Use of this source code is governed by an MIT license
// that can be found in the LICENSE file.
struct Foo {
bar int
mut:
str string
}
2019-06-23 04:12:56 +03:00
fn test_add() {
mut a := 'a'
a += 'b'
assert a==('ab')
2019-06-23 04:12:56 +03:00
a = 'a'
for i := 1; i < 1000; i++ {
a += 'b'
}
assert a.len == 1000
assert a.ends_with('bbbbb')
a += '123'
assert a.ends_with('3')
}
fn test_ends_with() {
a := 'browser.v'
assert a.ends_with('.v')
}
fn test_between() {
s := 'hello [man] how you doing'
assert s.find_between('[', ']') == 'man'
}
fn test_compare() {
a := 'Music'
b := 'src'
assert b>=(a)
2019-06-23 04:12:56 +03:00
}
2019-06-23 21:25:50 +03:00
fn test_lt() {
a := ''
b := 'a'
c := 'a'
d := 'b'
e := 'aa'
f := 'ab'
2019-08-26 13:32:53 +03:00
assert a < (b)
assert !(b < c)
assert c < (d)
assert !(d < e)
assert c < (e)
assert e < (f)
2019-06-23 21:25:50 +03:00
}
fn test_ge() {
a := 'aa'
b := 'aa'
c := 'ab'
d := 'abc'
e := 'aaa'
2019-08-26 13:32:53 +03:00
assert b >= (a)
assert c >= (b)
assert d >= (c)
2019-09-21 16:26:25 +03:00
assert !(c >= d)
2019-08-26 13:32:53 +03:00
assert e >= (a)
2019-06-23 21:25:50 +03:00
}
fn test_compare_strings() {
a := 'aa'
b := 'aa'
c := 'ab'
d := 'abc'
e := 'aaa'
assert compare_strings(a, b) == 0
assert compare_strings(b, c) == -1
assert compare_strings(c, d) == -1
assert compare_strings(d, e) == 1
assert compare_strings(a, e) == -1
assert compare_strings(e, a) == 1
}
2019-06-23 04:12:56 +03:00
fn test_sort() {
mut vals := [
2019-06-23 21:25:50 +03:00
'arr', 'an', 'a', 'any'
2019-06-23 04:12:56 +03:00
]
len := vals.len
vals.sort()
assert len == vals.len
2019-06-23 21:25:50 +03:00
assert vals[0] == 'a'
assert vals[1] == 'an'
assert vals[2] == 'any'
assert vals[3] == 'arr'
2019-06-23 04:12:56 +03:00
}
2019-12-01 16:10:13 +03:00
fn test_split_nth() {
a := "1,2,3"
assert (a.split(',').len == 3)
assert (a.split_nth(',', -1).len == 3)
assert (a.split_nth(',', 0).len == 3)
assert (a.split_nth(',', 1).len == 1)
assert (a.split_nth(',', 2).len == 2)
assert (a.split_nth(',', 10).len == 3)
b := "1::2::3"
assert (b.split('::').len == 3)
assert (b.split_nth('::', -1).len == 3)
assert (b.split_nth('::', 0).len == 3)
assert (b.split_nth('::', 1).len == 1)
assert (b.split_nth('::', 2).len == 2)
assert (b.split_nth('::', 10).len == 3)
c := "ABCDEF"
println(c.split('').len)
2019-12-01 16:10:13 +03:00
assert (c.split('').len == 6)
assert (c.split_nth('', 3).len == 3)
assert (c.split_nth('BC', -1).len == 2)
d := ","
assert (d.split(',').len == 2)
assert (d.split_nth('', 3).len == 1)
assert (d.split_nth(',', -1).len == 2)
assert (d.split_nth(',', 3).len == 2)
e := ",,,0,,,,,a,,b,"
// assert (e.split(',,').len == 5)
// assert (e.split_nth(',,', 3).len == 2)
assert (e.split_nth(',', -1).len == 12)
assert (e.split_nth(',', 3).len == 3)
}
fn test_split_nth_values() {
line := 'CMD=eprintln(phase=1)'
a0 := line.split_nth('=', 0)
assert a0.len == 3
assert a0[0] == 'CMD'
assert a0[1] == 'eprintln(phase'
assert a0[2] == '1)'
a1 := line.split_nth('=', 1)
assert a1.len == 1
assert a1[0] == 'CMD=eprintln(phase=1)'
a2 := line.split_nth('=', 2)
assert a2.len == 2
assert a2[0] == 'CMD'
assert a2[1] == 'eprintln(phase=1)'
a3 := line.split_nth('=', 3)
assert a3.len == 3
assert a3[0] == 'CMD'
assert a3[1] == 'eprintln(phase'
assert a3[2] == '1)'
a4 := line.split_nth('=', 4)
assert a4.len == 3
assert a4[0] == 'CMD'
assert a4[1] == 'eprintln(phase'
assert a4[2] == '1)'
}
2019-06-23 04:12:56 +03:00
fn test_split() {
mut s := 'volt/twitch.v:34'
mut vals := s.split(':')
assert vals.len == 2
2019-08-26 13:32:53 +03:00
assert vals[0] == 'volt/twitch.v'
assert vals[1] == '34'
// /////////
2019-06-23 04:12:56 +03:00
s = '2018-01-01z13:01:02'
vals = s.split('z')
assert vals.len == 2
2019-08-26 13:32:53 +03:00
assert vals[0] =='2018-01-01'
assert vals[1] == '13:01:02'
// //////////
2019-06-23 04:12:56 +03:00
s = '4627a862c3dec29fb3182a06b8965e0025759e18___1530207969___blue'
vals = s.split('___')
assert vals.len == 3
assert vals[0]== '4627a862c3dec29fb3182a06b8965e0025759e18'
assert vals[1]=='1530207969'
assert vals[2]== 'blue'
2019-07-31 08:26:22 +03:00
// /////////
s = 'lalala'
2019-08-18 15:02:30 +03:00
vals = s.split('a')
2019-12-01 16:10:13 +03:00
assert vals.len == 4
2019-08-18 15:02:30 +03:00
assert vals[0] == 'l'
assert vals[1] == 'l'
assert vals[2] == 'l'
2019-12-01 16:10:13 +03:00
assert vals[3] == ''
2019-11-10 19:37:36 +03:00
// /////////
s = 'awesome'
a := s.split('')
assert a.len == 7
assert a[0] == 'a'
assert a[1] == 'w'
assert a[2] == 'e'
assert a[3] == 's'
assert a[4] == 'o'
assert a[5] == 'm'
assert a[6] == 'e'
2019-06-23 04:12:56 +03:00
}
fn test_trim_space() {
a := ' a '
assert a.trim_space() == 'a'
code := '
fn main() {
println(2)
}
'
code_clean := 'fn main() {
println(2)
}'
assert code.trim_space() == code_clean
}
fn test_join() {
mut strings := [ 'a', 'b', 'c' ]
mut s := strings.join(' ')
assert s == 'a b c'
strings = ['one
two ',
2019-08-18 15:02:30 +03:00
'three!
2019-06-23 04:12:56 +03:00
four!']
s = strings.join(' ')
assert s.contains('one') && s.contains('two ') && s.contains('four')
}
fn test_clone() {
mut a := 'a'
a += 'a'
a += 'a'
2019-07-25 15:06:26 +03:00
b := a
2019-07-25 15:13:35 +03:00
c := a.clone()
2019-06-23 04:12:56 +03:00
assert c == a
assert c == 'aaa'
assert b == 'aaa'
}
fn test_replace() {
a := 'hello man!'
mut b := a.replace('man', 'world')
assert b==('hello world!')
2019-06-23 04:12:56 +03:00
b = b.replace('!', '')
assert b==('hello world')
2019-06-23 04:12:56 +03:00
b = b.replace('h', 'H')
assert b==('Hello world')
2019-07-10 10:48:10 +03:00
b = b.replace('foo', 'bar')
assert b==('Hello world')
2019-06-23 04:12:56 +03:00
s := 'hey man how are you'
assert s.replace('man ', '') == 'hey how are you'
lol := 'lol lol lol'
2019-07-10 10:48:10 +03:00
assert lol.replace('lol', 'LOL') == 'LOL LOL LOL'
2019-06-23 04:12:56 +03:00
b = 'oneBtwoBBthree'
assert b.replace('B', '') == 'onetwothree'
2019-12-01 10:33:26 +03:00
b = '*charptr'
assert b.replace('charptr', 'byteptr') == '*byteptr'
c :='abc'
2019-06-28 18:17:54 +03:00
assert c.replace('','-') == c
2019-12-27 07:20:06 +03:00
v :='a b c d'
assert v.replace(' ',' ') == 'a b c d'
2019-06-23 04:12:56 +03:00
}
2019-12-10 14:32:12 +03:00
fn test_replace_each() {
s := 'hello man man :)'
q := s.replace_each([
'man', 'dude',
'hello', 'hey'
])
assert q == 'hey dude dude :)'
bb := '[b]bold[/b] [code]code[/code]'
assert bb.replace_each([
'[b]', '<b>',
'[/b]', '</b>',
'[code]', '<code>',
'[/code]', '</code>'
]) == '<b>bold</b> <code>code</code>'
2019-12-11 03:20:30 +03:00
bb2 := '[b]cool[/b]'
assert bb2.replace_each([
'[b]', '<b>',
'[/b]', '</b>',
]) == '<b>cool</b>'
2019-12-10 14:32:12 +03:00
}
2019-06-23 04:12:56 +03:00
fn test_itoa() {
num := 777
assert num.str() == '777'
big := 7779998
assert big.str() == '7779998'
a := 3
assert a.str() == '3'
b := 5555
assert b.str() == '5555'
zero := 0
assert zero.str() == '0'
neg := -7
assert neg.str() == '-7'
}
fn test_reassign() {
a := 'hi'
mut b := a
b += '!'
assert a == 'hi'
assert b == 'hi!'
}
fn test_runes() {
s := 'привет'
assert s.len == 12
s2 := 'privet'
assert s2.len == 6
u := s.ustring()
assert u.len == 6
assert s2.substr(1, 4).len == 3
assert s2.substr(1, 4) == 'riv'
2019-10-27 01:37:29 +03:00
assert s2[1..4].len == 3
assert s2[1..4] == 'riv'
assert s2[..4].len == 4
assert s2[..4] == 'priv'
assert s2[2..].len == 4
assert s2[2..] == 'ivet'
2019-06-23 04:12:56 +03:00
assert u.substr(1, 4).len == 6
assert u.substr(1, 4) == 'рив'
assert s2.substr(1, 2) == 'r'
assert u.substr(1, 2) == 'р'
assert s2.ustring().at(1) == 'r'
assert u.at(1) == 'р'
first := u.at(0)
last := u.at(u.len - 1)
assert first.len == 2
assert last.len == 2
}
fn test_lower() {
mut s := 'A'
assert s.to_lower() == 'a'
assert s.to_lower().len == 1
s = 'HELLO'
assert s.to_lower() == 'hello'
assert s.to_lower().len == 5
s = 'Aloha'
assert s.to_lower() == 'aloha'
s = 'Have A nice Day!'
assert s.to_lower() == 'have a nice day!'
s = 'hi'
assert s.to_lower() == 'hi'
}
2019-08-26 13:32:53 +03:00
fn test_upper() {
mut s := 'a'
assert s.to_upper() == 'A'
assert s.to_upper().len == 1
s = 'hello'
assert s.to_upper() == 'HELLO'
assert s.to_upper().len == 5
s = 'Aloha'
assert s.to_upper() == 'ALOHA'
s = 'have a nice day!'
assert s.to_upper() == 'HAVE A NICE DAY!'
s = 'hi'
assert s.to_upper() == 'HI'
}
2019-06-23 04:12:56 +03:00
fn test_left_right() {
s := 'ALOHA'
assert s.left(3) == 'ALO'
2020-02-20 22:14:21 +03:00
assert s.left(0) == ''
assert s.left(8) == s
assert s.right(3) == 'HA'
assert s.right(6) == ''
assert s[3..] == 'HA'
2019-06-23 04:12:56 +03:00
u := s.ustring()
assert u.left(3) == 'ALO'
2020-02-20 22:14:21 +03:00
assert u.left(0) == ''
assert s.left(8) == s
2019-06-23 04:12:56 +03:00
assert u.right(3) == 'HA'
2020-02-20 22:14:21 +03:00
assert u.right(6) == ''
2019-06-23 04:12:56 +03:00
}
fn test_contains() {
s := 'view.v'
assert s.contains('vi')
assert !s.contains('random')
}
fn test_arr_contains() {
a := ['a', 'b', 'c']
assert a.contains('b')
ints := [1, 2, 3]
assert ints.contains(2)
}
fn test_to_num() {
s := '7'
2019-06-25 15:56:34 +03:00
assert s.int() == 7
2019-10-15 04:26:19 +03:00
assert s.u64() == 7
2019-06-23 04:12:56 +03:00
f := '71.5 hasdf'
2020-04-02 11:15:35 +03:00
// QTODO
//assert f.f32() == 71.5
2019-06-23 04:12:56 +03:00
b := 1.52345
mut a := '${b:.03f}'
assert a == '1.523'
num := 7
a = '${num:03d}'
vals := ['9']
2019-06-25 15:56:34 +03:00
assert vals[0].int() == 9
2019-10-15 04:26:19 +03:00
big := '93993993939322'
assert big.u64() == 93993993939322
assert big.i64() == 93993993939322
2019-06-23 04:12:56 +03:00
}
fn test_hash() {
s := '10000'
assert s.hash() == 46730161
s2 := '24640'
assert s2.hash() == 47778736
s3 := 'Content-Type'
assert s3.hash() == 949037134
s4 := 'bad_key'
assert s4.hash() == -346636507
s5 := '24640'
// From a map collision test
assert s5.hash() % ((1 << 20) -1) == s.hash() % ((1 << 20) -1)
assert s5.hash() % ((1 << 20) -1) == 592861
}
fn test_trim() {
assert 'banana'.trim('bna') == ''
assert 'abc'.trim('ac') == 'b'
assert 'aaabccc'.trim('ac') == 'b'
}
2019-06-23 04:12:56 +03:00
fn test_trim_left() {
mut s := 'module main'
assert s.trim_left(' ') == 'module main'
s = ' module main'
assert s.trim_left(' ') == 'module main'
// test cutset
s = 'banana'
assert s.trim_left('ba') == 'nana'
2019-06-23 04:12:56 +03:00
}
2019-08-17 20:07:40 +03:00
fn test_trim_right() {
mut s := 'module main'
assert s.trim_right(' ') == 'module main'
s = 'module main '
assert s.trim_right(' ') == 'module main'
// test cutset
s = 'banana'
assert s.trim_right('na') == 'b'
2019-08-17 20:07:40 +03:00
}
2020-02-20 22:14:21 +03:00
fn test_all_before() {
s := 'fn hello fn'
assert s.all_before(' ') == 'fn'
assert s.all_before('2') == s
assert s.all_before('') == s
}
fn test_all_before_last() {
s := 'fn hello fn'
assert s.all_before_last(' ') == 'fn hello'
assert s.all_before_last('2') == s
assert s.all_before_last('') == s
}
2019-06-23 04:12:56 +03:00
fn test_all_after() {
s := 'fn hello'
2020-02-20 22:14:21 +03:00
assert s.all_after('fn ') == 'hello'
assert s.all_after('test') == s
assert s.all_after('') == s
2019-06-26 20:03:35 +03:00
}
2019-06-25 15:56:34 +03:00
2019-07-07 18:43:34 +03:00
fn test_reverse() {
2020-02-20 22:14:21 +03:00
assert 'hello'.reverse() == 'olleh'
assert ''.reverse() == ''
assert 'a'.reverse() == 'a'
2019-07-07 18:43:34 +03:00
}
2019-07-16 18:59:07 +03:00
2019-07-18 05:42:00 +03:00
fn test_bytes_to_string() {
2020-03-04 19:39:27 +03:00
mut buf := vcalloc(10)
2019-08-18 15:02:30 +03:00
buf[0] = `h`
buf[1] = `e`
buf[2] = `l`
buf[3] = `l`
buf[4] = `o`
assert string(buf) == 'hello'
assert string(buf, 2) == 'he'
bytes := [`h`, `e`, `l`, `l`, `o`]
assert string(bytes, 5) == 'hello'
}
fn test_count() {
assert ''.count('') == 0
assert ''.count('a') == 0
assert 'a'.count('') == 0
assert 'aa'.count('a') == 2
assert 'aa'.count('aa') == 1
assert 'aabbaa'.count('aa') == 2
assert 'bbaabb'.count('aa') == 1
}
2019-08-26 13:32:53 +03:00
fn test_capitalize() {
mut s := 'hello'
assert s.capitalize() == 'Hello'
s = 'test'
assert s.capitalize() == 'Test'
s = 'i am ray'
assert s.capitalize() == 'I am ray'
2020-02-20 13:33:38 +03:00
s = ''
assert s.capitalize() == ''
2019-08-26 13:32:53 +03:00
}
fn test_title() {
s := 'hello world'
2019-08-26 13:32:53 +03:00
assert s.title() == 'Hello World'
s.to_upper()
assert s.title() == 'Hello World'
s.to_lower()
2019-09-21 16:26:25 +03:00
assert s.title() == 'Hello World'
}
fn test_for_loop() {
mut i := 0
s := 'abcd'
for c in s {
assert c == s[i]
i++
}
}
fn test_for_loop_two() {
s := 'abcd'
for i, c in s {
assert c == s[i]
}
}
2019-09-11 15:21:20 +03:00
fn test_quote() {
a := `'`
2019-09-21 16:26:25 +03:00
println("testing double quotes")
b := "hi"
assert b == 'hi'
2019-09-11 15:21:20 +03:00
assert a.str() == '\''
}
2020-03-26 13:32:29 +03:00
fn test_ustring_comparisons() {
2020-03-26 13:32:29 +03:00
/*
2020-03-26 19:21:52 +03:00
QTODO
assert ('hllô !'.ustring() == 'hllô !'.ustring()) == true
assert ('hllô !'.ustring() == 'hllô'.ustring()) == false
assert ('hllô !'.ustring() == 'hllo !'.ustring()) == false
assert ('hllô !'.ustring() != 'hllô !'.ustring()) == false
assert ('hllô !'.ustring() != 'hllô'.ustring()) == true
assert ('hllô'.ustring() < 'hllô!'.ustring()) == true
assert ('hllô'.ustring() < 'hllo'.ustring()) == false
assert ('hllo'.ustring() < 'hllô'.ustring()) == true
assert ('hllô'.ustring() <= 'hllô!'.ustring()) == true
assert ('hllô'.ustring() <= 'hllô'.ustring()) == true
assert ('hllô!'.ustring() <= 'hllô'.ustring()) == false
assert ('hllô!'.ustring() > 'hllô'.ustring()) == true
assert ('hllô'.ustring() > 'hllô'.ustring()) == false
assert ('hllô!'.ustring() >= 'hllô'.ustring()) == true
assert ('hllô'.ustring() >= 'hllô'.ustring()) == true
assert ('hllô'.ustring() >= 'hllô!'.ustring()) == false
2020-03-26 13:32:29 +03:00
*/
}
fn test_ustring_count() {
a := 'hllô hllô '.ustring()
assert (a.count('l'.ustring())) == 4
assert (a.count(''.ustring())) == 2
assert (a.count('hllô'.ustring())) == 2
assert (a.count(''.ustring())) == 2
assert (a.count('a'.ustring())) == 0
}
2019-09-26 22:54:53 +03:00
2020-02-20 22:14:21 +03:00
fn test_limit() {
s := 'hello'
assert s.limit(2) == 'he'
assert s.limit(9) == s
assert s.limit(0) == ''
// assert s.limit(-1) == ''
}
2019-09-26 22:54:53 +03:00
fn test_repeat() {
s1 := 'V! '
assert s1.repeat(5) == 'V! V! V! V! V! '
assert s1.repeat(1) == s1
assert s1.repeat(0) == ''
s2 := ''
assert s2.repeat(5) == s2
assert s2.repeat(1) == s2
assert s2.repeat(0) == s2
// TODO Add test for negative values
2019-09-26 22:54:53 +03:00
}
2019-10-06 03:56:08 +03:00
fn test_raw() {
raw := r'raw\nstring'
lines := raw.split('\n')
2020-04-02 11:15:35 +03:00
println(lines)
2019-10-06 03:56:08 +03:00
assert lines.len == 1
println('raw string: "$raw"')
}
2019-10-07 01:09:11 +03:00
2020-02-09 18:29:05 +03:00
fn test_raw_with_quotes() {
raw := r"some'" + r'"thing' // " should be escaped in the generated C code
assert raw[0] == `s`
assert raw[5] == `"`
assert raw[6] == `t`
}
2019-10-07 01:09:11 +03:00
fn test_escape() {
// TODO
//a := 10
//println("\"$a")
}
2019-10-15 04:26:19 +03:00
2019-10-25 23:41:18 +03:00
fn test_atoi() {
assert '234232'.int() == 234232
assert '-9009'.int() == -9009
assert '0'.int() == 0
for n in -10000 .. 100000 {
s := n.str()
assert s.int() == n
}
}
2019-10-25 23:41:18 +03:00
fn test_raw_inter() {
world := 'world'
println(world)
s := r'hello\n$world'
assert s == r'hello\n$world'
assert s.contains('$')
}
2019-11-16 22:16:58 +03:00
fn test_c_r() {
// This used to break because of r'' and c''
c := 42
println('$c')
r := 50
println('$r')
2019-12-03 13:08:57 +03:00
}
2019-12-03 13:08:57 +03:00
fn test_inter_before_comp_if() {
s := '123'
// This used to break ('123 $....')
$if linux {
println(s)
}
2019-12-12 13:51:05 +03:00
assert s == '123'
}
2019-11-16 22:16:58 +03:00
2019-12-08 20:06:00 +03:00
fn test_double_quote_inter() {
a := 1
b := 2
println("${a} ${b}")
assert "${a} ${b}" == "1 2"
assert '${a} ${b}' == "1 2"
}
2020-02-09 11:23:57 +03:00
fn test_split_into_lines() {
line_content := 'Line'
text_crlf := '${line_content}\r\n${line_content}\r\n${line_content}'
lines_crlf := text_crlf.split_into_lines()
assert lines_crlf.len == 3
for line in lines_crlf {
assert line == line_content
}
text_lf := '${line_content}\n${line_content}\n${line_content}'
lines_lf := text_lf.split_into_lines()
assert lines_lf.len == 3
for line in lines_lf {
assert line == line_content
}
}
2020-03-15 07:46:12 +03:00