mountpoint-s3-csi-driver/cmd/aws-s3-csi-controller/csicontroller/expectations_test.go at eeb1699f3bdd8681bad7d67b1a6a489ad05d32d0 · yerzhan7/mountpoint-s3-csi-driver · 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
package csicontroller

import (
	"testing"

	"github.com/awslabs/aws-s3-csi-driver/pkg/util/testutil/assert"
	"sigs.k8s.io/controller-runtime/pkg/client"
)

func TestDeriveExpectationKeyFromFilters(t *testing.T) {
	tests := []struct {
		name         string
		fieldFilters client.MatchingFields
		want         string
	}{
		{
			name:         "empty filters",
			fieldFilters: client.MatchingFields{},
			want:         "",
		},
		{
			name: "single filter",
			fieldFilters: client.MatchingFields{
				"key1": "value1",
			},
			want: "key1=value1;",
		},
		{
			name: "multiple filters",
			fieldFilters: client.MatchingFields{
				"key2": "value2",
				"key1": "value1",
				"key3": "value3",
			},
			want: "key1=value1;key2=value2;key3=value3;",
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			got := deriveExpectationKeyFromFilters(tt.fieldFilters)
			assert.Equals(t, tt.want, got)
		})
	}
}

func TestExpectations(t *testing.T) {
	tests := []struct {
		name         string
		fieldFilters client.MatchingFields
		operations   func(*expectations)
		wantPending  bool
	}{
		{
			name: "set and check pending",
			fieldFilters: client.MatchingFields{
				"key1": "value1",
			},
			operations: func(e *expectations) {
				e.setPending(client.MatchingFields{"key1": "value1"})
			},
			wantPending: true,
		},
		{
			name: "set and clear pending",
			fieldFilters: client.MatchingFields{
				"key1": "value1",
			},
			operations: func(e *expectations) {
				e.setPending(client.MatchingFields{"key1": "value1"})
				e.clear(client.MatchingFields{"key1": "value1"})
			},
			wantPending: false,
		},
		{
			name: "check non-existent pending",
			fieldFilters: client.MatchingFields{
				"key1": "value1",
			},
			operations:  func(e *expectations) {},
			wantPending: false,
		},
		{
			name: "multiple operations",
			fieldFilters: client.MatchingFields{
				"key1": "value1",
				"key2": "value2",
			},
			operations: func(e *expectations) {
				e.setPending(client.MatchingFields{"key1": "value1", "key2": "value2"})
				e.clear(client.MatchingFields{"key1": "value1"}) // Different key, shouldn't affect the test
			},
			wantPending: true,
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			e := newExpectations()
			tt.operations(e)
			got := e.isPending(tt.fieldFilters)
			assert.Equals(t, tt.wantPending, got)
		})
	}
}