shrimp/src/parser/tests/function-blocks.test.ts

import { expect, describe, test } from 'bun:test'

import '../shrimp.grammar' // Importing this so changes cause it to retest!

describe('single line function blocks', () => {
  test('work with no args', () => {
    expect(`trap: echo bye bye end`).toMatchTree(`
  FunctionCallWithBlock
    FunctionCallOrIdentifier
      Identifier trap
    colon :
    FunctionCall
      Identifier echo
      PositionalArg
        Identifier bye
      PositionalArg
        Identifier bye
    keyword end`
    )
  })

  test('work with one arg', () => {
    expect(`trap EXIT: echo bye bye end`).toMatchTree(`
  FunctionCallWithBlock
    FunctionCall
      Identifier trap
      PositionalArg
        Word EXIT
    colon :
    FunctionCall
      Identifier echo
      PositionalArg
        Identifier bye
      PositionalArg
        Identifier bye
    keyword end`
    )
  })

  test('work with named args', () => {
    expect(`attach signal='exit': echo bye bye end`).toMatchTree(`
  FunctionCallWithBlock
    FunctionCall
      Identifier attach
      NamedArg
        NamedArgPrefix signal=
        String
          StringFragment exit
    colon :
    FunctionCall
      Identifier echo
      PositionalArg
        Identifier bye
      PositionalArg
        Identifier bye
    keyword end`
    )
  })


  test('work with dot-get', () => {
    expect(`signals = [=]; signals.trap 'EXIT': echo bye bye end`).toMatchTree(`
    Assign
      AssignableIdentifier signals
      Eq =
      Dict [=]
    FunctionCallWithBlock
      FunctionCall
        DotGet
          IdentifierBeforeDot signals
          Identifier trap
        PositionalArg
          String
            StringFragment EXIT
      colon :
      FunctionCall
        Identifier echo
        PositionalArg
          Identifier bye
        PositionalArg
          Identifier bye
      keyword end`
    )
  })
})

describe('multi line function blocks', () => {
  test('work with no args', () => {
    expect(`
trap: 
  echo bye bye 
end
`).toMatchTree(`
  FunctionCallWithBlock
    FunctionCallOrIdentifier
      Identifier trap
    colon :
    FunctionCall
      Identifier echo
      PositionalArg
        Identifier bye
      PositionalArg
        Identifier bye
    keyword end`
    )
  })

  test('work with one arg', () => {
    expect(`
trap EXIT: 
  echo bye bye 
end`).toMatchTree(`
  FunctionCallWithBlock
    FunctionCall
      Identifier trap
      PositionalArg
        Word EXIT
    colon :
    FunctionCall
      Identifier echo
      PositionalArg
        Identifier bye
      PositionalArg
        Identifier bye
    keyword end`
    )
  })

  test('work with named args', () => {
    expect(`
attach signal='exit' code=1:
  echo bye bye 
end`).toMatchTree(`
  FunctionCallWithBlock
    FunctionCall
      Identifier attach
      NamedArg
        NamedArgPrefix signal=
        String
          StringFragment exit
      NamedArg
        NamedArgPrefix code=
        Number 1
    colon :
    FunctionCall
      Identifier echo
      PositionalArg
        Identifier bye
      PositionalArg
        Identifier bye
    keyword end`
    )
  })


  test('work with dot-get', () => {
    expect(`
signals = [=]
signals.trap 'EXIT': 
  echo bye bye 
end`).toMatchTree(`
    Assign
      AssignableIdentifier signals
      Eq =
      Dict [=]
    FunctionCallWithBlock
      FunctionCall
        DotGet
          IdentifierBeforeDot signals
          Identifier trap
        PositionalArg
          String
            StringFragment EXIT
      colon :
      FunctionCall
        Identifier echo
        PositionalArg
          Identifier bye
        PositionalArg
          Identifier bye
      keyword end`
    )
  })
})

describe('ribbit', () => {
  test('head tag', () => {
    expect(`
head:
  title What up
  meta charSet=UTF-8
  meta name='viewport' content='width=device-width, initial-scale=1, viewport-fit=cover'
end`).toMatchTree(`
  FunctionCallWithBlock
    FunctionCallOrIdentifier
      Identifier head
    colon :
    FunctionCall
      Identifier title
      PositionalArg
        Word What
      PositionalArg
        Identifier up
    FunctionCall
      Identifier meta
      PositionalArg
        Word charSet=UTF-8
    FunctionCall
      Identifier meta
      NamedArg
        NamedArgPrefix name=
        String
          StringFragment viewport
      NamedArg
        NamedArgPrefix content=
        String
          StringFragment width=device-width, initial-scale=1, viewport-fit=cover
    keyword end
  `)
  })

  test('li', () => {
    expect(`
list:
  li border-bottom='1px solid black' one
  li two
  li three
end`).toMatchTree(`
  FunctionCallWithBlock
    FunctionCallOrIdentifier
      Identifier list
    colon :
    FunctionCall
      Identifier li
      NamedArg
        NamedArgPrefix border-bottom=
        String
          StringFragment 1px solid black
      PositionalArg
        Identifier one
    FunctionCall
      Identifier li
      PositionalArg
        Identifier two
    FunctionCall
      Identifier li
      PositionalArg
        Identifier three
    keyword end`)
  })

  test('inline expressions', () => {
    expect(`
p:
  h1 class=bright style='font-family: helvetica' Heya
  h2 man that is (b wild)!
end`)
      .toMatchTree(`
  FunctionCallWithBlock
    FunctionCallOrIdentifier
      Identifier p
    colon :
    FunctionCall
      Identifier h1
      NamedArg
        NamedArgPrefix class=
        Identifier bright
      NamedArg
        NamedArgPrefix style=
        String
          StringFragment font-family: helvetica
      PositionalArg
        Word Heya
    FunctionCall
      Identifier h2
      PositionalArg
        Identifier man
      PositionalArg
        Identifier that
      PositionalArg
        Identifier is
      PositionalArg
        ParenExpr
          FunctionCall
            Identifier b
            PositionalArg
              Identifier wild
      PositionalArg
        Word !
    keyword end`)
  })
})