go-jar/plugins_test.go at master · cognusion/go-jar · GitHub - github.com

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
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
//go:build plugins

package jar

import (
	"net/http"
	"net/http/httptest"
	"testing"

	. "github.com/smartystreets/goconvey/convey"
	"github.com/spf13/viper"
)

func Test_PluginConfig(t *testing.T) {
	req, err := http.NewRequest("GET", "/", nil)
	if err != nil {
		t.Fatal(err)
	}

	Convey("Creating a PluginConfig from a known good config works as expected.", t, func() {

		c := viper.New()
		c.SetConfigFile("tests/plugins/hw.yaml")
		err := c.ReadInConfig()
		So(err, ShouldBeNil)

		pc, err := NewPluginConfig("plugins.helloworld", c)
		So(err, ShouldBeNil)

		h, err := pc.CreateHandler()
		So(err, ShouldBeNil)

		Convey("... and when testing the handler, it too works as expected", func() {
			// Give it a whirl
			testHandler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
			})

			rr := httptest.NewRecorder()

			handler := h(testHandler)

			handler.ServeHTTP(rr, req)

			So(rr.Code, ShouldEqual, http.StatusOK)
			So(rr.Body.String(), ShouldResemble, "Hello World")
		})
	})

	Convey("Creating a PluginConfig from a known good config with its own config works as expected.", t, func() {

		c := viper.New()
		c.SetConfigFile("tests/plugins/hw.yaml")
		err := c.ReadInConfig()
		So(err, ShouldBeNil)

		pc, err := NewPluginConfig("plugins.conftest", c)
		So(err, ShouldBeNil)

		h, err := pc.CreateHandler()
		So(err, ShouldBeNil)

		Convey("... and when testing the handler, it too works as expected", func() {
			// Give it a whirl
			testHandler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
			})

			rr := httptest.NewRecorder()

			handler := h(testHandler)

			handler.ServeHTTP(rr, req)

			So(rr.Code, ShouldEqual, http.StatusOK)
			So(rr.Body.String(), ShouldContainSubstring, "hello = world")
		})
	})
}

func Test_PluginHandler(t *testing.T) {
	req, err := http.NewRequest("GET", "/", nil)
	if err != nil {
		t.Fatal(err)
	}

	Convey("When invoking PluginHandler to glue it all together, everything works as expected", t, func() {

		c := viper.New()
		c.SetConfigFile("tests/plugins/hw.yaml")
		err := c.ReadInConfig()
		So(err, ShouldBeNil)

		h, err := PluginHandler("plugins.helloworld", c)
		So(err, ShouldBeNil)

		Convey("... and when testing the handler, it too works as expected", func() {
			// Give it a whirl
			testHandler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
			})

			rr := httptest.NewRecorder()

			handler := h(testHandler)

			handler.ServeHTTP(rr, req)

			So(rr.Code, ShouldEqual, http.StatusOK)
			So(rr.Body.String(), ShouldResemble, "Hello World")
		})
	})
}