From 621e49bb465f500cc46d47e39e828cf76d6381d7 Mon Sep 17 00:00:00 2001 From: Dimitri Sokolyuk Date: Tue, 24 Jul 2018 14:35:44 +0200 Subject: update vendor --- .../github.com/golang/mock/gomock/callset_test.go | 76 ++++++++++++++++++++++ 1 file changed, 76 insertions(+) create mode 100644 vendor/github.com/golang/mock/gomock/callset_test.go (limited to 'vendor/github.com/golang/mock/gomock/callset_test.go') diff --git a/vendor/github.com/golang/mock/gomock/callset_test.go b/vendor/github.com/golang/mock/gomock/callset_test.go new file mode 100644 index 0000000..7fc711a --- /dev/null +++ b/vendor/github.com/golang/mock/gomock/callset_test.go @@ -0,0 +1,76 @@ +// Copyright 2011 Google Inc. +// +// Licensed under the Apache License, Version 2.0 (the "License"); +// you may not use this file except in compliance with the License. +// You may obtain a copy of the License at +// +// http://www.apache.org/licenses/LICENSE-2.0 +// +// Unless required by applicable law or agreed to in writing, software +// distributed under the License is distributed on an "AS IS" BASIS, +// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +// See the License for the specific language governing permissions and +// limitations under the License. + +package gomock + +import ( + "reflect" + "testing" +) + +type receiverType struct{} + +func (receiverType) Func() {} + +func TestCallSetAdd(t *testing.T) { + method := "TestMethod" + var receiver interface{} = "TestReceiver" + cs := newCallSet() + + numCalls := 10 + for i := 0; i < numCalls; i++ { + cs.Add(newCall(t, receiver, method, reflect.TypeOf(receiverType{}.Func))) + } + + call, err := cs.FindMatch(receiver, method, []interface{}{}) + if err != nil { + t.Fatalf("FindMatch: %v", err) + } + if call == nil { + t.Fatalf("FindMatch: Got nil, want non-nil *Call") + } +} + +func TestCallSetRemove(t *testing.T) { + method := "TestMethod" + var receiver interface{} = "TestReceiver" + + cs := newCallSet() + ourCalls := []*Call{} + + numCalls := 10 + for i := 0; i < numCalls; i++ { + // NOTE: abuse the `numCalls` value to convey initial ordering of mocked calls + generatedCall := &Call{receiver: receiver, method: method, numCalls: i} + cs.Add(generatedCall) + ourCalls = append(ourCalls, generatedCall) + } + + // validateOrder validates that the calls in the array are ordered as they were added + validateOrder := func(calls []*Call) { + // lastNum tracks the last `numCalls` (call order) value seen + lastNum := -1 + for _, c := range calls { + if lastNum >= c.numCalls { + t.Errorf("found call %d after call %d", c.numCalls, lastNum) + } + lastNum = c.numCalls + } + } + + for _, c := range ourCalls { + validateOrder(cs.expected[callSetKey{receiver, method}]) + cs.Remove(c) + } +} -- cgit v1.2.3