forked from ardanlabs/gotraining
/
example1.go
53 lines (42 loc) · 1.43 KB
/
example1.go
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
// All material is licensed under the Apache License Version 2.0, January 2004
// http://www.apache.org/licenses/LICENSE-2.0
// Sample program to show how you can personally mock concrete types when
// you need to for your own packages or tests.
package main
import (
"github.com/ardanlabs/gotraining/topics/api/composition/mocking/example1/pubsub"
)
// publisher is an interface to allow this package to mock the pubsub
// package support.
type publisher interface {
Publish(key string, v interface{}) error
Subscribe(key string) error
}
// mock is a concrete type to help support the mocking of the pubsub package.
type mock struct{}
// Publish implements the publisher interface for the mock.
func (m *mock) Publish(key string, v interface{}) error {
// ADD YOUR MOCK FOR THE PUBLISH CALL.
return nil
}
// Subscribe implements the publisher interface for the mock.
func (m *mock) Subscribe(key string) error {
// ADD YOUR MOCK FOR THE SUBSCRIBE CALL.
return nil
}
func main() {
// Create a slice of publisher interface values. Assign
// the address of a pubsub.PubSub value and the address of
// a mock value.
pubs := []publisher{
pubsub.New("localhost"),
&mock{},
}
// Range over the interface value to see how the publisher
// interface provides the level of decoupling the user needs.
// The pubsub package did not need to provide the interface type.
for _, p := range pubs {
p.Publish("key", "value")
p.Subscribe("key")
}
}