Function TestSRem

structure/set_test.go:305–364

(t *testing.T)

}

func TestSRem(t *testing.T) {
	// Creating a list of cases for testing.
	tests := []struct {
		name        string
		setup       func(*SetStructure)
		key         string
		membersRem  []string
		membersWant []string
		wantErr     error
	}{
		{
			name:        "test when key not found",
			setup:       func(s *SetStructure) {},
			key:         "destination",
			membersRem:  []string{"key1", "key2"},
			membersWant: []string{},
			wantErr:     _const.ErrKeyNotFound,
		},
		{
			name: "test when remove one key",
			setup: func(s *SetStructure) {
				_ = s.SAdds("destination", 0, "key1", "key2")
			},
			key:         "destination",
			membersRem:  []string{"key1"},
			membersWant: []string{"key2"},
			wantErr:     nil,
		},
		{
			name: "test remove all keys",
			setup: func(s *SetStructure) {
				_ = s.SAdds("destination", 0, "key3", "key4")
			},
			key:         "destination",
			membersRem:  []string{"key3", "key4"},
			membersWant: []string{},
			wantErr:     nil,
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			s, _ := initTestSetDb()
			tt.setup(s)
			for _, s2 := range tt.membersRem {
				// Call the method with test case parameters.
				err := s.SRem(tt.key, s2)
				if !errors.Is(err, tt.wantErr) {
					t.Errorf("SRems() error = %v, wantErr %v", err, tt.wantErr)
				}
			}
			if tt.wantErr == nil {
				// positive verify
				assert.True(t, s.exists(tt.key, tt.membersWant...))
				// verify removal
				assert.False(t, s.exists(tt.key, tt.membersRem...))
			}
		})