-- tests string functions added by dfhack.lua

config.target = 'core'

function test.startswith()

expect.true_(('abcd'):startswith(''))

expect.true_(('abcd'):startswith('abc'))

expect.false_(('abcd'):startswith('bcd'))

expect.false_(('abcd'):startswith('abcde'))

expect.true_((''):startswith(''))

expect.false_((''):startswith('a'))

expect.false_(('str'):startswith('.'),

'ensure we match literals, not patterns')

end

function test.endswith()

expect.true_(('abcd'):endswith(''))

expect.true_(('abcd'):endswith('bcd'))

expect.false_(('abcd'):endswith('abc'))

expect.false_(('abcd'):endswith('zabcd'))

expect.true_((''):endswith(''))

expect.false_((''):endswith('a'))

expect.false_(('str'):endswith('.'),

'ensure we match literals, not patterns')

end

function test.split()

expect.table_eq({'hello','world'}, ('hello world'):split())

expect.table_eq({'hello','','world'}, ('hello world'):split())

expect.table_eq({'hello','world'}, ('hello world'):split(' +'))

expect.table_eq({'hello','world'}, ('hello.world'):split('.', true),

'ensure literal interpretation when plain is true')

expect.table_eq({'', '', '', ''}, ('abc'):split('.'),

'ensure pattern interpretation when plain is false')

-- we don't actually care what this returns, just that it does return

expect.true_(('hello world'):split('.*'), 'ensure no infinite loop')

expect.table_eq({'hello ', ' world'}, ('hello , world'):split(','),

'ensure spaces are kept when they are not the delimiter')

expect.table_eq({'hello'}, ('hello'):split(), 'no delimiter')

end

function test.trim()

expect.eq('hello', ('hello'):trim())

expect.eq('hello', (' hello'):trim())

expect.eq('hello', ('hello '):trim())

expect.eq('hello', (' hello '):trim())

expect.eq('', (''):trim())

expect.eq('', (' '):trim())

expect.eq('', (' \t \n \v '):trim())

expect.eq('hel lo', (' hel lo '):trim(), 'keep interior spaces')

expect.eq('hel \n lo', (' hel \n lo '):trim(),

'keep interior spaces across newlines')

end

function test.wrap()

expect.eq('', (''):wrap())

expect.eq('', (' '):wrap())

expect.eq('\n', (' '):wrap(3))

expect.eq('\n', (' '):wrap(3))

expect.eq('', (''):wrap(3, {keep_trailing_spaces=true}))

expect.eq(' ', (' '):wrap(3, {keep_trailing_spaces=true}))

expect.eq(' \n ', (' '):wrap(3, {keep_trailing_spaces=true}))

expect.eq(' \n \n ', (' '):wrap(3, {keep_trailing_spaces=true}))

expect.eq('hello world', ('hello world'):wrap(20))

expect.eq('hello world', ('hello world'):wrap(20))

expect.eq(' hello world', (' hello world '):wrap(20))

expect.eq(' hello world', (' hello world '):wrap(20))

expect.eq('hello world\nhow are you?',('hello world how are you?'):wrap(12))

expect.eq('hello\nworld', ('hello world'):wrap(5))

expect.eq('hello\nworld\n\n', ('hello world\n'):wrap(5))

expect.eq('hello\nworld', ('hello world'):wrap(5))

expect.eq('\nhello\nworld', (' hello world '):wrap(5))

expect.eq('hello\nworld', ('hello world'):wrap(8))

expect.eq('hel\nlo\nwor\nld', ('hello world'):wrap(3))

expect.eq('hel\nloo\nwor\nldo', ('helloo worldo'):wrap(3))

expect.eq('hel\nloo\n \nwor\nldo', ('helloo worldo'):wrap(3, {keep_trailing_spaces=true}))

expect.table_eq({'hel', 'lo', 'wor', 'ld'}, ('hello world'):wrap(3, {return_as_table=true}))

expect.error_match('expected width > 0', function() ('somestr'):wrap(0) end)

-- extended tests based on gui/journal

local journal_opts = {return_as_table=true, keep_trailing_spaces=true, keep_original_newlines=true}

expect.table_eq({

'this ',

'is a ',

'simple ',

'text'

}, ('this is a simple text'):wrap(7, journal_opts), 'simple text')

expect.table_eq({

'this ',

'is ',

'a ',

'simple ',

'text'

}, ('this is a simple text'):wrap(7, journal_opts), 'keep endline spaces')

expect.table_eq({

' this ',

'is a ',

'simple ',

'text'

}, (' this is a simple text'):wrap(7, journal_opts), 'keep text leading spaces')

expect.table_eq({

'this ',

'is a ',

'simple ',

'text '

}, ('this is a simple text '):wrap(7, journal_opts), 'keep text trailing spaces')

expect.table_eq({

'thiswor',

'distool',

'ong is ',

'a ',

'simple ',

'text'

}, ('thiswordistoolong is a simple text'):wrap(7, journal_opts),

'cut words longer than max width')

expect.table_eq({

'this ',

'is a ',

'sim\n',

'ple ',

'text\n',

''

}, ('this is a sim\nple text\n'):wrap(7, journal_opts),

'take into account existing new line')

expect.table_eq({

'this ',

'is a ',

'sim\n',

'\n',

'\n',

'ple ',

'text'

}, ('this is a sim\n\n\nple text'):wrap(7, journal_opts),

'take into account existing new lines')

end

function test.escape_pattern()

-- no change expected

expect.eq('', (''):escape_pattern())

expect.eq(' ', (' '):escape_pattern())

expect.eq('abc', ('abc'):escape_pattern())

expect.eq('a,b', ('a,b'):escape_pattern())

expect.eq('"a,b"', ('"a,b"'):escape_pattern())

-- excape regex chars

expect.eq('iz for me%?', ('iz for me?'):escape_pattern())

expect.eq('%.%*', ('.*'):escape_pattern())

expect.eq('%( %) %. %% %+ %- %* %? %[ %] %^ %$',

('( ) . % + - * ? [ ] ^ $'):escape_pattern())

end