FilesExpand file tree

develop

/

token_test.go

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

52 lines (45 loc) · 1.14 KB

develop

/

token_test.go

Copy path

Top

File metadata and controls

• Code
• Blame

52 lines (45 loc) · 1.14 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

package parser

import (

"bufio"

_ "embed"

"io/fs"

"os"

"path/filepath"

"strings"

"testing"

"github.com/stretchr/testify/assert"

"github.com/stretchr/testify/require"

)

func TestSplit(t *testing.T) {

const testdata = "testdata"

var fixture []string

require.NoError(t, filepath.WalkDir(testdata, func(path string, f fs.DirEntry, err error) error {

if err != nil {

return err

}

if !strings.HasPrefix(f.Name(), "split_") {

return nil

}

contents, err := os.ReadFile(path)

if err != nil {

return err

}

fixture = append(fixture, string(contents))

return nil

}))

require.Len(t, fixture, 18)

fixture = append(fixture, "\n")

sql, err := os.Open(filepath.Join(testdata, "all.sql"))

require.NoError(t, err)

stats, err := Split(sql)

require.NoError(t, err)

assert.ElementsMatch(t, fixture, stats)

}

func TestSplitAndTrim(t *testing.T) {

sql := "\tBEGIN; " + strings.Repeat("a", MaxScannerCapacity)

stats, err := SplitAndTrim(strings.NewReader(sql))

// Check error

assert.ErrorIs(t, err, bufio.ErrTooLong)

assert.ErrorContains(t, err, "After statement 1: \tBEGIN;")

assert.ElementsMatch(t, []string{"BEGIN"}, stats)

}