local t = require('test.testutil')
local n = require('test.functional.testnvim')()

local clear = n.clear
local fn = n.fn
local api = n.api
local eq = t.eq
local matches = t.matches
local eval = n.eval
local command = n.command
local pcall_err = t.pcall_err
local NIL = vim.NIL
local source = n.source

describe('json_decode() function', function()
  setup(function()
    clear()
    source([[
      language C
      function Eq(exp, act)
        let act = a:act
        let exp = a:exp
        if type(exp) != type(act)
          return 0
        endif
        if type(exp) == type({})
          if sort(keys(exp)) !=# sort(keys(act))
            return 0
          endif
          if sort(keys(exp)) ==# ['_TYPE', '_VAL']
            let exp_typ = v:msgpack_types[exp._TYPE]
            let act_typ = act._TYPE
            if exp_typ isnot act_typ
              return 0
            endif
            return Eq(exp._VAL, act._VAL)
          else
            return empty(filter(copy(exp), '!Eq(v:val, act[v:key])'))
          endif
        else
          if type(exp) == type([])
            if len(exp) != len(act)
              return 0
            endif
            return empty(filter(copy(exp), '!Eq(v:val, act[v:key])'))
          endif
          return exp ==# act
        endif
        return 1
      endfunction
      function EvalEq(exp, act_expr)
        let act = eval(a:act_expr)
        if Eq(a:exp, act)
          return 1
        else
          return string(act)
        endif
      endfunction
    ]])
  end)

  local speq = function(expected, actual_expr)
    eq(1, fn.EvalEq(expected, actual_expr))
  end

  it('accepts readfile()-style list', function()
    eq(
      { Test = 1 },
      fn.json_decode({
        '{',
        '\t"Test": 1',
        '}',
      })
    )
  end)

  it('accepts strings with newlines', function()
    eq(
      { Test = 1 },
      fn.json_decode([[
      {
        "Test": 1
      }
    ]])
    )
  end)

  it('parses null, true, false', function()
    eq(NIL, fn.json_decode('null'))
    eq(true, fn.json_decode('true'))
    eq(false, fn.json_decode('false'))
  end)

  it('fails to parse incomplete null, true, false', function()
    eq('Vim(call):E474: Expected null: n', pcall_err(command, 'call json_decode("n")'))
    eq('Vim(call):E474: Expected null: nu', pcall_err(command, 'call json_decode("nu")'))
    eq('Vim(call):E474: Expected null: nul', pcall_err(command, 'call json_decode("nul")'))
    matches(
      'Vim%(call%):E474: Expected null: nul',
      pcall_err(command, 'call json_decode("nul\\n\\t")')
    )

    eq('Vim(call):E474: Expected true: t', pcall_err(command, 'call json_decode("t")'))
    eq('Vim(call):E474: Expected true: tr', pcall_err(command, 'call json_decode("tr")'))
    eq('Vim(call):E474: Expected true: tru', pcall_err(command, 'call json_decode("tru")'))
    matches(
      'Vim%(call%):E474: Expected true: tru',
      pcall_err(command, 'call json_decode("tru\\t\\n")')
    )

    eq('Vim(call):E474: Expected false: f', pcall_err(command, 'call json_decode("f")'))
    eq('Vim(call):E474: Expected false: fa', pcall_err(command, 'call json_decode("fa")'))
    eq('Vim(call):E474: Expected false: fal', pcall_err(command, 'call json_decode("fal")'))
    eq(
      'Vim(call):E474: Expected false: fal   <',
      pcall_err(command, 'call json_decode("   fal   <")')
    )
    eq('Vim(call):E474: Expected false: fals', pcall_err(command, 'call json_decode("fals")'))
  end)

  it('parses integer numbers', function()
    eq(100000, fn.json_decode('100000'))
    eq(-100000, fn.json_decode('-100000'))
    eq(100000, fn.json_decode('  100000  '))
    eq(-100000, fn.json_decode('  -100000  '))
    eq(0, fn.json_decode('0'))
    eq(0, fn.json_decode('-0'))
  end)

  it('fails to parse +numbers and .number', function()
    eq('Vim(call):E474: Unidentified byte: +1000', pcall_err(command, 'call json_decode("+1000")'))
    eq('Vim(call):E474: Unidentified byte: .1000', pcall_err(command, 'call json_decode(".1000")'))
  end)

  it('fails to parse numbers with leading zeroes', function()
    eq(
      'Vim(call):E474: Leading zeroes are not allowed: 00.1',
      pcall_err(command, 'call json_decode("00.1")')
    )
    eq(
      'Vim(call):E474: Leading zeroes are not allowed: 01',
      pcall_err(command, 'call json_decode("01")')
    )
    eq(
      'Vim(call):E474: Leading zeroes are not allowed: -01',
      pcall_err(command, 'call json_decode("-01")')
    )
    eq(
      'Vim(call):E474: Leading zeroes are not allowed: -001.0',
      pcall_err(command, 'call json_decode("-001.0")')
    )
  end)

  it('fails to parse incomplete numbers', function()
    eq(
      'Vim(call):E474: Missing number after minus sign: -.1',
      pcall_err(command, 'call json_decode("-.1")')
    )
    eq(
      'Vim(call):E474: Missing number after minus sign: -',
      pcall_err(command, 'call json_decode("-")')
    )
    eq(
      'Vim(call):E474: Missing number after decimal dot: -1.',
      pcall_err(command, 'call json_decode("-1.")')
    )
    eq(
      'Vim(call):E474: Missing number after decimal dot: 0.',
      pcall_err(command, 'call json_decode("0.")')
    )
    eq('Vim(call):E474: Missing exponent: 0.0e', pcall_err(command, 'call json_decode("0.0e")'))
    eq('Vim(call):E474: Missing exponent: 0.0e+', pcall_err(command, 'call json_decode("0.0e+")'))
    eq('Vim(call):E474: Missing exponent: 0.0e-', pcall_err(command, 'call json_decode("0.0e-")'))
    eq('Vim(call):E474: Missing exponent: 0.0e-', pcall_err(command, 'call json_decode("0.0e-")'))
    eq(
      'Vim(call):E474: Missing number after decimal dot: 1.e5',
      pcall_err(command, 'call json_decode("1.e5")')
    )
    eq(
      'Vim(call):E474: Missing number after decimal dot: 1.e+5',
      pcall_err(command, 'call json_decode("1.e+5")')
    )
    eq(
      'Vim(call):E474: Missing number after decimal dot: 1.e+',
      pcall_err(command, 'call json_decode("1.e+")')
    )
  end)

  it('parses floating-point numbers', function()
    -- Also test method call (->) syntax
    eq('100000.0', eval('"100000.0"->json_decode()->string()'))
    eq(100000.5, fn.json_decode('100000.5'))
    eq(-100000.5, fn.json_decode('-100000.5'))
    eq(-100000.5e50, fn.json_decode('-100000.5e50'))
    eq(100000.5e50, fn.json_decode('100000.5e50'))
    eq(100000.5e50, fn.json_decode('100000.5e+50'))
    eq(-100000.5e-50, fn.json_decode('-100000.5e-50'))
    eq(100000.5e-50, fn.json_decode('100000.5e-50'))
    eq(100000e-50, fn.json_decode('100000e-50'))
    eq(0.5, fn.json_decode('0.5'))
    eq(0.005, fn.json_decode('0.005'))
    eq(0.005, fn.json_decode('0.00500'))
    eq(0.5, fn.json_decode('0.00500e+002'))
    eq(0.00005, fn.json_decode('0.00500e-002'))

    eq(-0.0, fn.json_decode('-0.0'))
    eq(-0.0, fn.json_decode('-0.0e0'))
    eq(-0.0, fn.json_decode('-0.0e+0'))
    eq(-0.0, fn.json_decode('-0.0e-0'))
    eq(-0.0, fn.json_decode('-0e-0'))
    eq(-0.0, fn.json_decode('-0e-2'))
    eq(-0.0, fn.json_decode('-0e+2'))

    eq(0.0, fn.json_decode('0.0'))
    eq(0.0, fn.json_decode('0.0e0'))
    eq(0.0, fn.json_decode('0.0e+0'))
    eq(0.0, fn.json_decode('0.0e-0'))
    eq(0.0, fn.json_decode('0e-0'))
    eq(0.0, fn.json_decode('0e-2'))
    eq(0.0, fn.json_decode('0e+2'))
  end)

  it('fails to parse numbers with spaces inside', function()
    eq(
      'Vim(call):E474: Missing number after minus sign: - 1000',
      pcall_err(command, 'call json_decode("- 1000")')
    )
    eq(
      'Vim(call):E474: Missing number after decimal dot: 0. ',
      pcall_err(command, 'call json_decode("0. ")')
    )
    eq(
      'Vim(call):E474: Missing number after decimal dot: 0. 0',
      pcall_err(command, 'call json_decode("0. 0")')
    )
    eq('Vim(call):E474: Missing exponent: 0.0e 1', pcall_err(command, 'call json_decode("0.0e 1")'))
    eq(
      'Vim(call):E474: Missing exponent: 0.0e+ 1',
      pcall_err(command, 'call json_decode("0.0e+ 1")')
    )
    eq(
      'Vim(call):E474: Missing exponent: 0.0e- 1',
      pcall_err(command, 'call json_decode("0.0e- 1")')
    )
  end)

  it('fails to parse "," and ":"', function()
    eq(
      'Vim(call):E474: Comma not inside container: ,  ',
      pcall_err(command, 'call json_decode("  ,  ")')
    )
    eq(
      'Vim(call):E474: Colon not inside container: :  ',
      pcall_err(command, 'call json_decode("  :  ")')
    )
  end)

  it('parses empty containers', function()
    eq({}, fn.json_decode('[]'))
    eq('[]', eval('string(json_decode("[]"))'))
  end)

  it('fails to parse "[" and "{"', function()
    eq('Vim(call):E474: Unexpected end of input: {', pcall_err(command, 'call json_decode("{")'))
    eq('Vim(call):E474: Unexpected end of input: [', pcall_err(command, 'call json_decode("[")'))
  end)

  it('fails to parse "}" and "]"', function()
    eq('Vim(call):E474: No container to close: ]', pcall_err(command, 'call json_decode("]")'))
    eq('Vim(call):E474: No container to close: }', pcall_err(command, 'call json_decode("}")'))
  end)

  it('fails to parse containers which are closed by different brackets', function()
    eq(
      'Vim(call):E474: Closing dictionary with square bracket: ]',
      pcall_err(command, 'call json_decode("{]")')
    )
    eq(
      'Vim(call):E474: Closing list with curly bracket: }',
      pcall_err(command, 'call json_decode("[}")')
    )
  end)

  it('fails to parse concat inside container', function()
    eq(
      'Vim(call):E474: Expected comma before list item: []]',
      pcall_err(command, 'call json_decode("[[][]]")')
    )
    eq(
      'Vim(call):E474: Expected comma before list item: {}]',
      pcall_err(command, 'call json_decode("[{}{}]")')
    )
    eq(
      'Vim(call):E474: Expected comma before list item: ]',
      pcall_err(command, 'call json_decode("[1 2]")')
    )
    eq(
      'Vim(call):E474: Expected comma before dictionary key: ": 4}',
      pcall_err(command, 'call json_decode("{\\"1\\": 2 \\"3\\": 4}")')
    )
    eq(
      'Vim(call):E474: Expected colon before dictionary value: , "3" 4}',
      pcall_err(command, 'call json_decode("{\\"1\\" 2, \\"3\\" 4}")')
    )
  end)

  it('fails to parse containers with leading comma or colon', function()
    eq('Vim(call):E474: Leading comma: ,}', pcall_err(command, 'call json_decode("{,}")'))
    eq('Vim(call):E474: Leading comma: ,]', pcall_err(command, 'call json_decode("[,]")'))
    eq(
      'Vim(call):E474: Using colon not in dictionary: :]',
      pcall_err(command, 'call json_decode("[:]")')
    )
    eq('Vim(call):E474: Unexpected colon: :}', pcall_err(command, 'call json_decode("{:}")'))
  end)

  it('fails to parse containers with trailing comma', function()
    eq('Vim(call):E474: Trailing comma: ]', pcall_err(command, 'call json_decode("[1,]")'))
    eq('Vim(call):E474: Trailing comma: }', pcall_err(command, 'call json_decode("{\\"1\\": 2,}")'))
  end)

  it('fails to parse dictionaries with missing value', function()
    eq(
      'Vim(call):E474: Expected value after colon: }',
      pcall_err(command, 'call json_decode("{\\"1\\":}")')
    )
    eq('Vim(call):E474: Expected value: }', pcall_err(command, 'call json_decode("{\\"1\\"}")'))
  end)

  it('fails to parse containers with two commas or colons', function()
    eq(
      'Vim(call):E474: Duplicate comma: , "2": 2}',
      pcall_err(command, 'call json_decode("{\\"1\\": 1,, \\"2\\": 2}")')
    )
    eq(
      'Vim(call):E474: Duplicate comma: , "2", 2]',
      pcall_err(command, 'call json_decode("[\\"1\\", 1,, \\"2\\", 2]")')
    )
    eq(
      'Vim(call):E474: Duplicate colon: : 2}',
      pcall_err(command, 'call json_decode("{\\"1\\": 1, \\"2\\":: 2}")')
    )
    eq(
      'Vim(call):E474: Comma after colon: , 2}',
      pcall_err(command, 'call json_decode("{\\"1\\": 1, \\"2\\":, 2}")')
    )
    eq(
      'Vim(call):E474: Unexpected colon: : "2": 2}',
      pcall_err(command, 'call json_decode("{\\"1\\": 1,: \\"2\\": 2}")')
    )
    eq(
      'Vim(call):E474: Unexpected colon: :, "2": 2}',
      pcall_err(command, 'call json_decode("{\\"1\\": 1:, \\"2\\": 2}")')
    )
  end)

  it('fails to parse concat of two values', function()
    eq('Vim(call):E474: Trailing characters: []', pcall_err(command, 'call json_decode("{}[]")'))
  end)

  it('parses containers', function()
    eq({ 1 }, fn.json_decode('[1]'))
    eq({ NIL, 1 }, fn.json_decode('[null, 1]'))
    eq({ ['1'] = 2 }, fn.json_decode('{"1": 2}'))
    eq(
      { ['1'] = 2, ['3'] = { { ['4'] = { ['5'] = { {}, 1 } } } } },
      fn.json_decode('{"1": 2, "3": [{"4": {"5": [[], 1]}}]}')
    )
  end)

  it('fails to parse incomplete strings', function()
    matches(
      'Vim%(call%):E474: Expected string end:',
      pcall_err(command, 'call json_decode("\\t\\"")')
    )
    matches(
      'Vim%(call%):E474: Expected string end:',
      pcall_err(command, 'call json_decode("\\t\\"abc")')
    )
    matches(
      'Vim%(call%):E474: Unfinished escape sequence:',
      pcall_err(command, 'call json_decode("\\t\\"abc\\\\")')
    )
    matches(
      'Vim%(call%):E474: Unfinished unicode escape sequence:',
      pcall_err(command, 'call json_decode("\\t\\"abc\\\\u")')
    )
    matches(
      'Vim%(call%):E474: Unfinished unicode escape sequence:',
      pcall_err(command, 'call json_decode("\\t\\"abc\\\\u0")')
    )
    matches(
      'Vim%(call%):E474: Unfinished unicode escape sequence:',
      pcall_err(command, 'call json_decode("\\t\\"abc\\\\u00")')
    )
    matches(
      'Vim%(call%):E474: Unfinished unicode escape sequence:',
      pcall_err(command, 'call json_decode("\\t\\"abc\\\\u000")')
    )
    eq(
      'Vim(call):E474: Expected four hex digits after \\u: \\u"    ',
      pcall_err(command, 'call json_decode("\\t\\"abc\\\\u\\"    ")')
    )
    eq(
      'Vim(call):E474: Expected four hex digits after \\u: \\u0"    ',
      pcall_err(command, 'call json_decode("\\t\\"abc\\\\u0\\"    ")')
    )
    eq(
      'Vim(call):E474: Expected four hex digits after \\u: \\u00"    ',
      pcall_err(command, 'call json_decode("\\t\\"abc\\\\u00\\"    ")')
    )
    eq(
      'Vim(call):E474: Expected four hex digits after \\u: \\u000"    ',
      pcall_err(command, 'call json_decode("\\t\\"abc\\\\u000\\"    ")')
    )
    matches(
      'Vim%(call%):E474: Expected string end:',
      pcall_err(command, 'call json_decode("\\t\\"abc\\\\u0000")')
    )
  end)

  it('fails to parse unknown escape sequences', function()
    eq(
      'Vim(call):E474: Unknown escape sequence: \\a"',
      pcall_err(command, 'call json_decode("\\t\\"\\\\a\\"")')
    )
  end)

  it('parses strings properly', function()
    eq('\n', fn.json_decode('"\\n"'))
    eq('', fn.json_decode('""'))
    eq('\\/"\t\b\n\r\f', fn.json_decode([["\\\/\"\t\b\n\r\f"]]))
    eq('/a', fn.json_decode([["\/a"]]))
    -- Unicode characters: 2-byte, 3-byte, 4-byte
    eq(
      {
        '«',
        'ફ',
        '\240\144\128\128',
      },
      fn.json_decode({
        '[',
        '"«",',
        '"ફ",',
        '"\240\144\128\128"',
        ']',
      })
    )
  end)

  it('fails on strings with invalid bytes', function()
    eq(
      'Vim(call):E474: Only UTF-8 strings allowed: \255"',
      pcall_err(command, 'call json_decode("\\t\\"\\xFF\\"")')
    )
    eq(
      'Vim(call):E474: ASCII control characters cannot be present inside string: ',
      pcall_err(command, 'call json_decode(["\\"\\n\\""])')
    )
    -- 0xC2 starts 2-byte unicode character
    eq(
      'Vim(call):E474: Only UTF-8 strings allowed: \194"',
      pcall_err(command, 'call json_decode("\\t\\"\\xC2\\"")')
    )
    -- 0xE0 0xAA starts 3-byte unicode character
    eq(
      'Vim(call):E474: Only UTF-8 strings allowed: \224"',
      pcall_err(command, 'call json_decode("\\t\\"\\xE0\\"")')
    )
    eq(
      'Vim(call):E474: Only UTF-8 strings allowed: \224\170"',
      pcall_err(command, 'call json_decode("\\t\\"\\xE0\\xAA\\"")')
    )
    -- 0xF0 0x90 0x80 starts 4-byte unicode character
    eq(
      'Vim(call):E474: Only UTF-8 strings allowed: \240"',
      pcall_err(command, 'call json_decode("\\t\\"\\xF0\\"")')
    )
    eq(
      'Vim(call):E474: Only UTF-8 strings allowed: \240\144"',
      pcall_err(command, 'call json_decode("\\t\\"\\xF0\\x90\\"")')
    )
    eq(
      'Vim(call):E474: Only UTF-8 strings allowed: \240\144\128"',
      pcall_err(command, 'call json_decode("\\t\\"\\xF0\\x90\\x80\\"")')
    )
    -- 0xF9 0x80 0x80 0x80 starts 5-byte unicode character
    eq(
      'Vim(call):E474: Only UTF-8 strings allowed: \249"',
      pcall_err(command, 'call json_decode("\\t\\"\\xF9\\"")')
    )
    eq(
      'Vim(call):E474: Only UTF-8 strings allowed: \249\128"',
      pcall_err(command, 'call json_decode("\\t\\"\\xF9\\x80\\"")')
    )
    eq(
      'Vim(call):E474: Only UTF-8 strings allowed: \249\128\128"',
      pcall_err(command, 'call json_decode("\\t\\"\\xF9\\x80\\x80\\"")')
    )
    eq(
      'Vim(call):E474: Only UTF-8 strings allowed: \249\128\128\128"',
      pcall_err(command, 'call json_decode("\\t\\"\\xF9\\x80\\x80\\x80\\"")')
    )
    -- 0xFC 0x90 0x80 0x80 0x80 starts 6-byte unicode character
    eq(
      'Vim(call):E474: Only UTF-8 strings allowed: \252"',
      pcall_err(command, 'call json_decode("\\t\\"\\xFC\\"")')
    )
    eq(
      'Vim(call):E474: Only UTF-8 strings allowed: \252\144"',
      pcall_err(command, 'call json_decode("\\t\\"\\xFC\\x90\\"")')
    )
    eq(
      'Vim(call):E474: Only UTF-8 strings allowed: \252\144\128"',
      pcall_err(command, 'call json_decode("\\t\\"\\xFC\\x90\\x80\\"")')
    )
    eq(
      'Vim(call):E474: Only UTF-8 strings allowed: \252\144\128\128"',
      pcall_err(command, 'call json_decode("\\t\\"\\xFC\\x90\\x80\\x80\\"")')
    )
    eq(
      'Vim(call):E474: Only UTF-8 strings allowed: \252\144\128\128\128"',
      pcall_err(command, 'call json_decode("\\t\\"\\xFC\\x90\\x80\\x80\\x80\\"")')
    )
    -- Specification does not allow unquoted characters above 0x10FFFF
    eq(
      'Vim(call):E474: Only UTF-8 code points up to U+10FFFF are allowed to appear unescaped: \249\128\128\128\128"',
      pcall_err(command, 'call json_decode("\\t\\"\\xF9\\x80\\x80\\x80\\x80\\"")')
    )
    eq(
      'Vim(call):E474: Only UTF-8 code points up to U+10FFFF are allowed to appear unescaped: \252\144\128\128\128\128"',
      pcall_err(command, 'call json_decode("\\t\\"\\xFC\\x90\\x80\\x80\\x80\\x80\\"")')
    )
    -- '"\249\128\128\128\128"',
    -- '"\252\144\128\128\128\128"',
  end)

  it('parses surrogate pairs properly', function()
    eq('\240\144\128\128', fn.json_decode('"\\uD800\\uDC00"'))
    eq('\237\160\128a\237\176\128', fn.json_decode('"\\uD800a\\uDC00"'))
    eq('\237\160\128\t\237\176\128', fn.json_decode('"\\uD800\\t\\uDC00"'))

    eq('\237\160\128', fn.json_decode('"\\uD800"'))
    eq('\237\160\128a', fn.json_decode('"\\uD800a"'))
    eq('\237\160\128\t', fn.json_decode('"\\uD800\\t"'))

    eq('\237\176\128', fn.json_decode('"\\uDC00"'))
    eq('\237\176\128a', fn.json_decode('"\\uDC00a"'))
    eq('\237\176\128\t', fn.json_decode('"\\uDC00\\t"'))

    eq('\237\176\128', fn.json_decode('"\\uDC00"'))
    eq('a\237\176\128', fn.json_decode('"a\\uDC00"'))
    eq('\t\237\176\128', fn.json_decode('"\\t\\uDC00"'))

    eq('\237\160\128¬', fn.json_decode('"\\uD800\\u00AC"'))

    eq('\237\160\128\237\160\128', fn.json_decode('"\\uD800\\uD800"'))
  end)

  local sp_decode_eq = function(expected, json)
    api.nvim_set_var('__json', json)
    speq(expected, 'json_decode(g:__json)')
    command('unlet! g:__json')
  end

  it('parses strings with NUL properly', function()
    sp_decode_eq('\000', '"\\u0000"')
    sp_decode_eq('\000\n\000', '"\\u0000\\n\\u0000"')
    sp_decode_eq('\000«\000', '"\\u0000\\u00AB\\u0000"')
  end)

  it('parses dictionaries with duplicate keys to special maps', function()
    sp_decode_eq({ _TYPE = 'map', _VAL = { { 'a', 1 }, { 'a', 2 } } }, '{"a": 1, "a": 2}')
    sp_decode_eq(
      { _TYPE = 'map', _VAL = { { 'b', 3 }, { 'a', 1 }, { 'a', 2 } } },
      '{"b": 3, "a": 1, "a": 2}'
    )
    sp_decode_eq(
      { _TYPE = 'map', _VAL = { { 'b', 3 }, { 'a', 1 }, { 'c', 4 }, { 'a', 2 } } },
      '{"b": 3, "a": 1, "c": 4, "a": 2}'
    )
    sp_decode_eq(
      { _TYPE = 'map', _VAL = { { 'b', 3 }, { 'a', 1 }, { 'c', 4 }, { 'a', 2 }, { 'c', 4 } } },
      '{"b": 3, "a": 1, "c": 4, "a": 2, "c": 4}'
    )
    sp_decode_eq(
      { { _TYPE = 'map', _VAL = { { 'b', 3 }, { 'a', 1 }, { 'c', 4 }, { 'a', 2 }, { 'c', 4 } } } },
      '[{"b": 3, "a": 1, "c": 4, "a": 2, "c": 4}]'
    )
    sp_decode_eq({
      {
        d = {
          _TYPE = 'map',
          _VAL = { { 'b', 3 }, { 'a', 1 }, { 'c', 4 }, { 'a', 2 }, { 'c', 4 } },
        },
      },
    }, '[{"d": {"b": 3, "a": 1, "c": 4, "a": 2, "c": 4}}]')
    sp_decode_eq({
      1,
      {
        d = {
          _TYPE = 'map',
          _VAL = { { 'b', 3 }, { 'a', 1 }, { 'c', 4 }, { 'a', 2 }, { 'c', 4 } },
        },
      },
    }, '[1, {"d": {"b": 3, "a": 1, "c": 4, "a": 2, "c": 4}}]')
    sp_decode_eq({
      1,
      {
        a = {},
        d = {
          _TYPE = 'map',
          _VAL = {
            { 'b', 3 },
            { 'a', 1 },
            { 'c', 4 },
            { 'a', 2 },
            {
              'c',
              4,
            },
          },
        },
      },
    }, '[1, {"a": [], "d": {"b": 3, "a": 1, "c": 4, "a": 2, "c": 4}}]')
    sp_decode_eq(
      { _TYPE = 'map', _VAL = { { '', 3 }, { 'a', 1 }, { 'c', 4 }, { 'd', 2 }, { '', 4 } } },
      '{"": 3, "a": 1, "c": 4, "d": 2, "": 4}'
    )
    sp_decode_eq(
      { { _TYPE = 'map', _VAL = { { '', 3 }, { 'a', 1 }, { 'c', 4 }, { 'd', 2 }, { '', 4 } } } },
      '[{"": 3, "a": 1, "c": 4, "d": 2, "": 4}]'
    )
  end)

  it('parses dictionaries with empty keys', function()
    eq({ [''] = 4 }, fn.json_decode('{"": 4}'))
    eq(
      { b = 3, a = 1, c = 4, d = 2, [''] = 4 },
      fn.json_decode('{"b": 3, "a": 1, "c": 4, "d": 2, "": 4}')
    )
  end)

  it('parses dictionaries with keys with NUL bytes to special maps', function()
    sp_decode_eq({ _TYPE = 'map', _VAL = { { 'a\000\nb', 4 } } }, '{"a\\u0000\\nb": 4}')
    sp_decode_eq({ _TYPE = 'map', _VAL = { { 'a\000\nb\n', 4 } } }, '{"a\\u0000\\nb\\n": 4}')
    sp_decode_eq({
      _TYPE = 'map',
      _VAL = {
        { 'b', 3 },
        { 'a', 1 },
        { 'c', 4 },
        { 'd', 2 },
        { '\000', 4 },
      },
    }, '{"b": 3, "a": 1, "c": 4, "d": 2, "\\u0000": 4}')
  end)

  it('parses U+00C3 correctly', function()
    eq('\195\131', fn.json_decode('"\195\131"'))
  end)

  it('fails to parse empty string', function()
    eq(
      'Vim(call):E474: Attempt to decode a blank string',
      pcall_err(command, 'call json_decode("")')
    )
    eq(
      'Vim(call):E474: Attempt to decode a blank string',
      pcall_err(command, 'call json_decode([])')
    )
    eq(
      'Vim(call):E474: Attempt to decode a blank string',
      pcall_err(command, 'call json_decode([""])')
    )
    eq(
      'Vim(call):E474: Attempt to decode a blank string',
      pcall_err(command, 'call json_decode(" ")')
    )
    eq(
      'Vim(call):E474: Attempt to decode a blank string',
      pcall_err(command, 'call json_decode("\\t")')
    )
    eq(
      'Vim(call):E474: Attempt to decode a blank string',
      pcall_err(command, 'call json_decode("\\n")')
    )
    eq(
      'Vim(call):E474: Attempt to decode a blank string',
      pcall_err(command, 'call json_decode(" \\t\\n \\n\\t\\t \\n\\t\\n \\n \\t\\n\\t ")')
    )
  end)

  it('accepts all spaces in every position where space may be put', function()
    local s =
      ' \t\n\r \t\r\n \n\t\r \n\r\t \r\t\n \r\n\t\t \n\r\t \r\n\t\n \r\t\n\r \t\r \n\t\r\n \n \t\r\n \r\t\n\t \r\n\t\r \n\r \t\n\r\t \r \t\n\r \n\t\r\t \n\r\t\n \r\n \t\r\n\t'
    local str = ('%s{%s"key"%s:%s[%s"val"%s,%s"val2"%s]%s,%s"key2"%s:%s1%s}%s'):gsub('%%s', s)
    eq({ key = { 'val', 'val2' }, key2 = 1 }, fn.json_decode(str))
  end)

  it('does not overflow when writing error message about decoding ["", ""]', function()
    eq(
      'Vim(call):E474: Attempt to decode a blank string',
      pcall_err(command, 'call json_decode(["", ""])')
    )
  end)
end)

describe('json_encode() function', function()
  setup(function()
    clear()
    command('language C')
  end)

  it('dumps strings', function()
    eq('"Test"', fn.json_encode('Test'))
    eq('""', fn.json_encode(''))
    eq('"\\t"', fn.json_encode('\t'))
    eq('"\\n"', fn.json_encode('\n'))
    eq('"\\u001B"', fn.json_encode('\27'))
    eq('"þÿþ"', fn.json_encode('þÿþ'))
  end)

  it('dumps blobs', function()
    eq('[]', eval('json_encode(0z)'))
    eq('[222, 173, 190, 239]', eval('json_encode(0zDEADBEEF)'))
  end)

  it('dumps numbers', function()
    eq('0', fn.json_encode(0))
    eq('10', fn.json_encode(10))
    eq('-10', fn.json_encode(-10))
  end)

  it('dumps floats', function()
    -- Also test method call (->) syntax
    eq('0.0', eval('0.0->json_encode()'))
    eq('10.5', fn.json_encode(10.5))
    eq('-10.5', fn.json_encode(-10.5))
    eq('-1.0e-5', fn.json_encode(-1e-5))
    eq('1.0e50', eval('1.0e50->json_encode()'))
  end)

  it('fails to dump NaN and infinite values', function()
    eq(
      'Vim(call):E474: Unable to represent NaN value in JSON',
      pcall_err(command, 'call json_encode(str2float("nan"))')
    )
    eq(
      'Vim(call):E474: Unable to represent infinity in JSON',
      pcall_err(command, 'call json_encode(str2float("inf"))')
    )
    eq(
      'Vim(call):E474: Unable to represent infinity in JSON',
      pcall_err(command, 'call json_encode(-str2float("inf"))')
    )
  end)

  it('dumps lists', function()
    eq('[]', fn.json_encode({}))
    eq('[[]]', fn.json_encode({ {} }))
    eq('[[], []]', fn.json_encode({ {}, {} }))
  end)

  it('dumps dictionaries', function()
    eq('{}', eval('json_encode({})'))
    eq('{"d": []}', fn.json_encode({ d = {} }))
    eq('{"d": [], "e": []}', fn.json_encode({ d = {}, e = {} }))
    -- Empty keys are allowed per JSON spec (and Vim dicts, and msgpack).
    eq('{"": []}', fn.json_encode({ [''] = {} }))
  end)

  it('cannot dump generic mapping with generic mapping keys and values', function()
    command('let todump = {"_TYPE": v:msgpack_types.map, "_VAL": []}')
    command('let todumpv1 = {"_TYPE": v:msgpack_types.map, "_VAL": []}')
    command('let todumpv2 = {"_TYPE": v:msgpack_types.map, "_VAL": []}')
    command('call add(todump._VAL, [todumpv1, todumpv2])')
    eq(
      'Vim(call):E474: Invalid key in special dictionary',
      pcall_err(command, 'call json_encode(todump)')
    )
  end)

  it('cannot dump generic mapping with ext key', function()
    command('let todump = {"_TYPE": v:msgpack_types.ext, "_VAL": [5, ["",""]]}')
    command('let todump = {"_TYPE": v:msgpack_types.map, "_VAL": [[todump, 1]]}')
    eq(
      'Vim(call):E474: Invalid key in special dictionary',
      pcall_err(command, 'call json_encode(todump)')
    )
  end)

  it('cannot dump generic mapping with array key', function()
    command('let todump = {"_TYPE": v:msgpack_types.array, "_VAL": [5, [""]]}')
    command('let todump = {"_TYPE": v:msgpack_types.map, "_VAL": [[todump, 1]]}')
    eq(
      'Vim(call):E474: Invalid key in special dictionary',
      pcall_err(command, 'call json_encode(todump)')
    )
  end)

  it('cannot dump generic mapping with UINT64_MAX key', function()
    command('let todump = {"_TYPE": v:msgpack_types.integer}')
    command('let todump._VAL = [1, 3, 0x7FFFFFFF, 0x7FFFFFFF]')
    command('let todump = {"_TYPE": v:msgpack_types.map, "_VAL": [[todump, 1]]}')
    eq(
      'Vim(call):E474: Invalid key in special dictionary',
      pcall_err(command, 'call json_encode(todump)')
    )
  end)

  it('cannot dump generic mapping with floating-point key', function()
    command('let todump = {"_TYPE": v:msgpack_types.float, "_VAL": 0.125}')
    command('let todump = {"_TYPE": v:msgpack_types.map, "_VAL": [[todump, 1]]}')
    eq(
      'Vim(call):E474: Invalid key in special dictionary',
      pcall_err(command, 'call json_encode(todump)')
    )
  end)

  it('can dump generic mapping with STR special key and NUL', function()
    command('let todump = {"_TYPE": v:msgpack_types.string, "_VAL": ["\\n"]}')
    command('let todump = {"_TYPE": v:msgpack_types.map, "_VAL": [[todump, 1]]}')
    eq('{"\\u0000": 1}', eval('json_encode(todump)'))
  end)

  it('can dump STR special mapping with NUL and NL', function()
    command('let todump = {"_TYPE": v:msgpack_types.string, "_VAL": ["\\n", ""]}')
    eq('"\\u0000\\n"', eval('json_encode(todump)'))
  end)

  it('cannot dump special ext mapping', function()
    command('let todump = {"_TYPE": v:msgpack_types.ext, "_VAL": [5, ["",""]]}')
    eq(
      'Vim(call):E474: Unable to convert EXT string to JSON',
      pcall_err(command, 'call json_encode(todump)')
    )
  end)

  it('can dump special array mapping', function()
    command('let todump = {"_TYPE": v:msgpack_types.array, "_VAL": [5, [""]]}')
    eq('[5, [""]]', eval('json_encode(todump)'))
  end)

  it('can dump special UINT64_MAX mapping', function()
    command('let todump = {"_TYPE": v:msgpack_types.integer}')
    command('let todump._VAL = [1, 3, 0x7FFFFFFF, 0x7FFFFFFF]')
    eq('18446744073709551615', eval('json_encode(todump)'))
  end)

  it('can dump special INT64_MIN mapping', function()
    command('let todump = {"_TYPE": v:msgpack_types.integer}')
    command('let todump._VAL = [-1, 2, 0, 0]')
    eq('-9223372036854775808', eval('json_encode(todump)'))
  end)

  it('can dump special BOOLEAN true mapping', function()
    command('let todump = {"_TYPE": v:msgpack_types.boolean, "_VAL": 1}')
    eq('true', eval('json_encode(todump)'))
  end)

  it('can dump special BOOLEAN false mapping', function()
    command('let todump = {"_TYPE": v:msgpack_types.boolean, "_VAL": 0}')
    eq('false', eval('json_encode(todump)'))
  end)

  it('can dump special NIL mapping', function()
    command('let todump = {"_TYPE": v:msgpack_types.nil, "_VAL": 0}')
    eq('null', eval('json_encode(todump)'))
  end)

  it('fails to dump a function reference', function()
    eq(
      'Vim(call):E474: Error while dumping encode_tv2json() argument, itself: attempt to dump function reference',
      pcall_err(command, 'call json_encode(function("tr"))')
    )
  end)

  it('fails to dump a partial', function()
    command('function T() dict\nendfunction')
    eq(
      'Vim(call):E474: Error while dumping encode_tv2json() argument, itself: attempt to dump function reference',
      pcall_err(command, 'call json_encode(function("T", [1, 2], {}))')
    )
  end)

  it('fails to dump a function reference in a list', function()
    eq(
      'Vim(call):E474: Error while dumping encode_tv2json() argument, index 0: attempt to dump function reference',
      pcall_err(command, 'call json_encode([function("tr")])')
    )
  end)

  it('fails to dump a recursive list', function()
    command('let todump = [[[]]]')
    command('call add(todump[0][0], todump)')
    eq(
      'Vim(call):E724: unable to correctly dump variable with self-referencing container',
      pcall_err(command, 'call json_encode(todump)')
    )
  end)

  it('fails to dump a recursive dict', function()
    command('let todump = {"d": {"d": {}}}')
    command('call extend(todump.d.d, {"d": todump})')
    eq(
      'Vim(call):E724: unable to correctly dump variable with self-referencing container',
      pcall_err(command, 'call json_encode([todump])')
    )
  end)

  it('can dump dict with two same dicts inside', function()
    command('let inter = {}')
    command('let todump = {"a": inter, "b": inter}')
    eq('{"a": {}, "b": {}}', eval('json_encode(todump)'))
  end)

  it('can dump list with two same lists inside', function()
    command('let inter = []')
    command('let todump = [inter, inter]')
    eq('[[], []]', eval('json_encode(todump)'))
  end)

  it('fails to dump a recursive list in a special dict', function()
    command('let todump = {"_TYPE": v:msgpack_types.array, "_VAL": []}')
    command('call add(todump._VAL, todump)')
    eq(
      'Vim(call):E724: unable to correctly dump variable with self-referencing container',
      pcall_err(command, 'call json_encode(todump)')
    )
  end)

  it('fails to dump a recursive (val) map in a special dict', function()
    command('let todump = {"_TYPE": v:msgpack_types.map, "_VAL": []}')
    command('call add(todump._VAL, ["", todump])')
    eq(
      'Vim(call):E724: unable to correctly dump variable with self-referencing container',
      pcall_err(command, 'call json_encode([todump])')
    )
  end)

  it('fails to dump a recursive (val) map in a special dict, _VAL reference', function()
    command('let todump = {"_TYPE": v:msgpack_types.map, "_VAL": [["", []]]}')
    command('call add(todump._VAL[0][1], todump._VAL)')
    eq(
      'Vim(call):E724: unable to correctly dump variable with self-referencing container',
      pcall_err(command, 'call json_encode(todump)')
    )
  end)

  it('fails to dump a recursive (val) special list in a special dict', function()
    command('let todump = {"_TYPE": v:msgpack_types.array, "_VAL": []}')
    command('call add(todump._VAL, ["", todump._VAL])')
    eq(
      'Vim(call):E724: unable to correctly dump variable with self-referencing container',
      pcall_err(command, 'call json_encode(todump)')
    )
  end)

  it('fails when called with no arguments', function()
    eq(
      'Vim(call):E119: Not enough arguments for function: json_encode',
      pcall_err(command, 'call json_encode()')
    )
  end)

  it('fails when called with two arguments', function()
    eq(
      'Vim(call):E118: Too many arguments for function: json_encode',
      pcall_err(command, 'call json_encode(["", ""], 1)')
    )
  end)

  it('ignores improper values in &isprint', function()
    api.nvim_set_option_value('isprint', '1', {})
    eq(1, eval('"\1" =~# "\\\\p"'))
    eq('"\\u0001"', fn.json_encode('\1'))
  end)

  it('fails when using surrogate character in a UTF-8 string', function()
    eq(
      'Vim(call):E474: UTF-8 string contains code point which belongs to a surrogate pair: \237\160\128',
      pcall_err(command, 'call json_encode("\237\160\128")')
    )
    eq(
      'Vim(call):E474: UTF-8 string contains code point which belongs to a surrogate pair: \237\175\191',
      pcall_err(command, 'call json_encode("\237\175\191")')
    )
  end)

  it('dumps control characters as expected', function()
    eq(
      [["\u0000\u0001\u0002\u0003\u0004\u0005\u0006\u0007\b\t\n\u000B\f\r\u000E\u000F\u0010\u0011\u0012\u0013"]],
      eval(
        'json_encode({"_TYPE": v:msgpack_types.string, "_VAL": ["\n\1\2\3\4\5\6\7\8\9", "\11\12\13\14\15\16\17\18\19"]})'
      )
    )
  end)

  it('can dump NULL string', function()
    eq('""', eval('json_encode($XXX_UNEXISTENT_VAR_XXX)'))
  end)

  it('can dump NULL blob', function()
    eq('[]', eval('json_encode(v:_null_blob)'))
  end)

  it('can dump NULL list', function()
    eq('[]', eval('json_encode(v:_null_list)'))
  end)

  it('can dump NULL dict', function()
    eq('{}', eval('json_encode(v:_null_dict)'))
  end)

  it('fails to parse NULL strings and lists', function()
    eq(
      'Vim(call):E474: Attempt to decode a blank string',
      pcall_err(command, 'call json_decode($XXX_UNEXISTENT_VAR_XXX)')
    )
    eq(
      'Vim(call):E474: Attempt to decode a blank string',
      pcall_err(command, 'call json_decode(v:_null_list)')
    )
  end)
end)
