dbsteps/sqlite_test.go at master · godogx/dbsteps · GitHub

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
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
package dbsteps_test

import (
	"bytes"
	"database/sql"
	"os"
	"strings"
	"testing"
	"time"

	"github.com/bool64/sqluct"
	"github.com/cucumber/godog"
	"github.com/godogx/dbsteps"
	"github.com/godogx/vars"
	"github.com/stretchr/testify/require"
	_ "modernc.org/sqlite" // DB driver.
)

func TestNewManager(t *testing.T) {
	sqlDB, err := sql.Open("sqlite", ":memory:")
	require.NoError(t, err)

	fixture, err := os.ReadFile("_testdata/fixture.sql")
	require.NoError(t, err)

	for _, st := range sqluct.SplitStatements(string(fixture)) {
		_, err := sqlDB.Exec(st)
		require.NoError(t, err, st)
	}

	vs := vars.Steps{}

	m := dbsteps.NewManager()
	m.AddDB(sqlDB)

	buf := bytes.NewBuffer(nil)

	suite := godog.TestSuite{
		Name:                 "DatabaseContext",
		TestSuiteInitializer: nil,
		ScenarioInitializer: func(s *godog.ScenarioContext) {
			m.RegisterSteps(s)
			vs.Register(s)
		},
		Options: &godog.Options{
			Format:    "pretty",
			Output:    buf,
			Paths:     []string{"_testdata/DatabaseE2E.feature"},
			Strict:    true,
			Randomize: time.Now().UTC().UnixNano(),
		},
	}
	status := suite.Run()

	if status != 0 {
		t.Fatal(buf.String())
	}
}

func TestDiffClosestRowFeature(t *testing.T) {
	sqlDB, err := sql.Open("sqlite", ":memory:")
	require.NoError(t, err)

	fixture, err := os.ReadFile("_testdata/fixture.sql")
	require.NoError(t, err)

	for _, st := range sqluct.SplitStatements(string(fixture)) {
		_, err := sqlDB.Exec(st)
		require.NoError(t, err, st)
	}

	vs := vars.Steps{}

	m := dbsteps.NewManager()
	m.AddDB(sqlDB)

	buf := bytes.NewBuffer(nil)

	suite := godog.TestSuite{
		Name:                 "DatabaseDiffClosestRow",
		TestSuiteInitializer: nil,
		ScenarioInitializer: func(s *godog.ScenarioContext) {
			m.RegisterSteps(s)
			vs.Register(s)
		},
		Options: &godog.Options{
			Format:    "pretty",
			Output:    buf,
			Paths:     []string{"_testdata/DiffClosestRow.feature"},
			NoColors:  true,
			Strict:    true,
			Randomize: time.Now().UTC().UnixNano(),
		},
	}
	status := suite.Run()

	if status == 0 {
		t.Fatal("expected scenario failure")
	}

	out := buf.String()
	require.Contains(t, out, "Diff vs closest row")
	require.Contains(t, out, "Scenario: Diff With Vars")
	require.Contains(t, out, "Scenario: Diff Transposed")
	require.Contains(t, out, "matched 4/5 columns")
	require.Contains(t, out, "matched 5/6 columns")
	require.Contains(t, out, "id     | 99")
	require.Contains(t, out, "age    | 32")

	if block := diffBlock(out); block != "" {
		require.NotContains(t, block, "$created_at")
	}
}

func diffBlock(out string) string {
	start := strings.Index(out, "Diff vs closest row")
	if start == -1 {
		return ""
	}

	block := out[start:]
	if next := strings.Index(block, "Scenario: "); next > 0 {
		block = block[:next]
	}

	return block
}