hatmakers.go

// Copyright 2018 Twitch Interactive, Inc.  All Rights Reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License"). You may not
// use this file except in compliance with the License. A copy of the License is
// located at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// or in the "license" file accompanying this file. This file 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 twirptest

import (
	"context"
	"net/http"
	"net/http/httptest"
	"time"

	"github.com/twitchtv/twirp"
)

type hatmaker func(ctx context.Context, s *Size) (*Hat, error)

func (h hatmaker) MakeHat(ctx context.Context, s *Size) (*Hat, error) { return h(ctx, s) }

// HaberdasherFunc is a convenience to convert a function into a Haberdasher service.
func HaberdasherFunc(f func(ctx context.Context, s *Size) (*Hat, error)) Haberdasher {
	return hatmaker(f)
}

// Always makes a blank hat.
func NoopHatmaker() Haberdasher {
	return hatmaker(func(context.Context, *Size) (*Hat, error) {
		return &Hat{}, nil
	})
}

// Makes a hat, as long as its the size they like
func PickyHatmaker(want int32) Haberdasher {
	return hatmaker(func(ctx context.Context, s *Size) (*Hat, error) {
		if s.Inches != want {
			return nil, twirp.InvalidArgumentError("Inches", "I can't make a hat that size!")
		}
		return &Hat{Size: s.Inches, Color: "blue", Name: "top hat"}, nil
	})
}

// Makes a hat, but sure takes their time
func SlowHatmaker(dur time.Duration) Haberdasher {
	return hatmaker(func(ctx context.Context, s *Size) (*Hat, error) {
		time.Sleep(dur)
		return &Hat{Size: s.Inches, Color: "blue", Name: "top hat"}, nil
	})
}

// Always errors.
func ErroringHatmaker(err error) Haberdasher {
	return hatmaker(func(ctx context.Context, s *Size) (*Hat, error) {
		return nil, err
	})
}

// Panics.
func PanickyHatmaker(p interface{}) Haberdasher {
	return hatmaker(func(ctx context.Context, s *Size) (*Hat, error) {
		panic(p)
	})
}

// Returns nil, nil
func NilHatmaker() Haberdasher {
	return hatmaker(func(context.Context, *Size) (*Hat, error) {
		return nil, nil
	})
}

func ServerAndClient(h Haberdasher, hooks *twirp.ServerHooks) (*httptest.Server, Haberdasher) {
	s := httptest.NewServer(NewHaberdasherServer(h, hooks))
	c := NewHaberdasherProtobufClient(s.URL, http.DefaultClient)
	return s, c
}

func TwirpServerAndClient(hooks *twirp.ServerHooks) (*httptest.Server, Haberdasher) {
	return ServerAndClient(NoopHatmaker(), hooks)
}