gocql/query_error_test.go at master · mykaul/gocql · 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
//go:build unit
// +build unit

package gocql

import (
	"errors"
	"testing"
)

func TestQueryError_PotentiallyExecuted(t *testing.T) {
	t.Parallel()

	tests := []struct {
		name                string
		potentiallyExecuted bool
		expected            bool
	}{
		{
			name:                "potentially executed true",
			potentiallyExecuted: true,
			expected:            true,
		},
		{
			name:                "potentially executed false",
			potentiallyExecuted: false,
			expected:            false,
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			qErr := &QueryError{
				err:                 errors.New("test error"),
				potentiallyExecuted: tt.potentiallyExecuted,
			}

			got := qErr.PotentiallyExecuted()
			if got != tt.expected {
				t.Fatalf("QueryError.PotentiallyExecuted() = %v, expected %v", got, tt.expected)
			}
		})
	}
}

func TestQueryError_IsIdempotent(t *testing.T) {
	t.Parallel()

	tests := []struct {
		name         string
		isIdempotent bool
		expected     bool
	}{
		{
			name:         "idempotent true",
			isIdempotent: true,
			expected:     true,
		},
		{
			name:         "idempotent false",
			isIdempotent: false,
			expected:     false,
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			qErr := &QueryError{
				err:          errors.New("test error"),
				isIdempotent: tt.isIdempotent,
			}

			got := qErr.IsIdempotent()
			if got != tt.expected {
				t.Errorf("QueryError.IsIdempotent() = %v, expected %v", got, tt.expected)
			}
		})
	}
}

func TestQueryError_Error(t *testing.T) {
	t.Parallel()

	tests := []struct {
		name                string
		err                 error
		potentiallyExecuted bool
		expected            string
	}{
		{
			name:                "with potentially executed true",
			err:                 errors.New("connection error"),
			potentiallyExecuted: true,
			expected:            "connection error (potentially executed: true)",
		},
		{
			name:                "with potentially executed false",
			err:                 errors.New("syntax error"),
			potentiallyExecuted: false,
			expected:            "syntax error (potentially executed: false)",
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			qErr := &QueryError{
				err:                 tt.err,
				potentiallyExecuted: tt.potentiallyExecuted,
			}

			got := qErr.Error()
			if got != tt.expected {
				t.Errorf("QueryError.Error() = %v, expected %v", got, tt.expected)
			}
		})
	}
}