You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
78 lines
1.7 KiB
Go
78 lines
1.7 KiB
Go
// Copyright 2019 Drone.IO Inc. All rights reserved.
|
|
// Use of this source code is governed by the Polyform License
|
|
// that can be found in the LICENSE file.
|
|
|
|
package client
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"testing"
|
|
|
|
"git.awesome-for.me/liuzhiguo/drone-go/drone"
|
|
)
|
|
|
|
var noContext = context.Background()
|
|
|
|
func TestSingleFlight(t *testing.T) {
|
|
mock := &mockRequestClient{
|
|
out: &drone.Stage{},
|
|
err: errors.New("some random error"),
|
|
}
|
|
client := NewSingleFlight("", "", false)
|
|
client.Client = mock
|
|
out, err := client.Request(noContext, nil)
|
|
if got, want := out, mock.out; got != want {
|
|
t.Errorf("Expect stage returned from request")
|
|
}
|
|
if got, want := err, mock.err; got != want {
|
|
t.Errorf("Expect error returned from request")
|
|
}
|
|
}
|
|
|
|
func TestSingleFlightPanic(t *testing.T) {
|
|
mock := &mockRequestClientPanic{}
|
|
client := NewSingleFlight("", "", false)
|
|
client.Client = mock
|
|
|
|
defer func() {
|
|
if recover() != nil {
|
|
t.Errorf("Expect Request to recover from panic")
|
|
}
|
|
client.mu.Lock()
|
|
client.mu.Unlock()
|
|
}()
|
|
|
|
client.Request(noContext, nil)
|
|
}
|
|
|
|
func TestSingleFlightCancel(t *testing.T) {
|
|
ctx, cancel := context.WithCancel(noContext)
|
|
cancel()
|
|
client := NewSingleFlight("", "", false)
|
|
client.Request(ctx, nil)
|
|
}
|
|
|
|
// mock client that returns a static stage and error
|
|
// from the request method.
|
|
type mockRequestClient struct {
|
|
Client
|
|
|
|
out *drone.Stage
|
|
err error
|
|
}
|
|
|
|
func (m *mockRequestClient) Request(ctx context.Context, args *Filter) (*drone.Stage, error) {
|
|
return m.out, m.err
|
|
}
|
|
|
|
// mock client that returns panics when the request
|
|
// method is invoked.
|
|
type mockRequestClientPanic struct {
|
|
Client
|
|
}
|
|
|
|
func (m *mockRequestClientPanic) Request(ctx context.Context, args *Filter) (*drone.Stage, error) {
|
|
panic("method not implemented")
|
|
}
|