/
server_suite_test.go
95 lines (77 loc) · 2.58 KB
/
server_suite_test.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
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
package server_test
import (
"fmt"
"os"
"path/filepath"
"time"
"code.cloudfoundry.org/app-autoscaler/src/autoscaler/fakes"
"code.cloudfoundry.org/app-autoscaler/src/autoscaler/helpers"
"code.cloudfoundry.org/app-autoscaler/src/autoscaler/metricsforwarder/config"
. "code.cloudfoundry.org/app-autoscaler/src/autoscaler/metricsforwarder/server"
"code.cloudfoundry.org/app-autoscaler/src/autoscaler/models"
"code.cloudfoundry.org/lager"
. "github.com/onsi/ginkgo/v2"
. "github.com/onsi/gomega"
"github.com/patrickmn/go-cache"
"github.com/tedsuo/ifrit"
"github.com/tedsuo/ifrit/ginkgomon_v2"
"testing"
)
var (
serverProcess ifrit.Process
serverUrl string
policyDB *fakes.FakePolicyDB
rateLimiter *fakes.FakeLimiter
fakeCredentials *fakes.FakeCredentials
allowedMetricCache cache.Cache
)
func TestServer(t *testing.T) {
RegisterFailHandler(Fail)
RunSpecs(t, "Server Suite")
}
var _ = SynchronizedBeforeSuite(func() []byte {
_, err := os.ReadFile("../../../../test-certs/metron.key")
Expect(err).NotTo(HaveOccurred())
_, err = os.ReadFile("../../../../test-certs/metron.crt")
Expect(err).NotTo(HaveOccurred())
_, err = os.ReadFile("../../../../test-certs/loggregator-ca.crt")
Expect(err).NotTo(HaveOccurred())
return nil
}, func(_ []byte) {
testCertDir := "../../../../test-certs"
loggregatorConfig := config.LoggregatorConfig{
TLS: models.TLSCerts{
KeyFile: filepath.Join(testCertDir, "metron.key"),
CertFile: filepath.Join(testCertDir, "metron.crt"),
CACertFile: filepath.Join(testCertDir, "loggregator-ca.crt"),
},
MetronAddress: "invalid-host-name-blah:12345",
}
serverConfig := config.ServerConfig{
Port: 2222 + GinkgoParallelProcess(),
}
loggerConfig := helpers.LoggingConfig{
Level: "debug",
}
conf := &config.Config{
Server: serverConfig,
Logging: loggerConfig,
LoggregatorConfig: loggregatorConfig,
}
policyDB = &fakes.FakePolicyDB{}
allowedMetricCache = *cache.New(10*time.Minute, -1)
httpStatusCollector := &fakes.FakeHTTPStatusCollector{}
rateLimiter = &fakes.FakeLimiter{}
fakeCredentials = &fakes.FakeCredentials{}
logger := lager.NewLogger("server_suite_test")
logger.RegisterSink(lager.NewWriterSink(GinkgoWriter, lager.DEBUG))
httpServer, err := NewServer(logger, conf, policyDB,
fakeCredentials, allowedMetricCache, httpStatusCollector, rateLimiter)
Expect(err).NotTo(HaveOccurred())
serverUrl = fmt.Sprintf("http://127.0.0.1:%d", conf.Server.Port)
serverProcess = ginkgomon_v2.Invoke(httpServer)
})
var _ = SynchronizedAfterSuite(func() {
ginkgomon_v2.Interrupt(serverProcess)
}, func() {
})