104 lines
4.1 KiB
Go
104 lines
4.1 KiB
Go
package common
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/truecharts/apps/tests/helmunit"
|
|
"github.com/stretchr/testify/suite"
|
|
)
|
|
|
|
type ServiceTestSuite struct {
|
|
suite.Suite
|
|
Chart helmunit.HelmChart
|
|
}
|
|
|
|
func (suite *ServiceTestSuite) SetupSuite() {
|
|
suite.Chart = helmunit.New("common-test", "../../common-test")
|
|
suite.Chart.UpdateDependencies()
|
|
}
|
|
|
|
// We need this function to kick off the test suite, otherwise
|
|
// "go test" won't know about our tests
|
|
func TestService(t *testing.T) {
|
|
suite.Run(t, new(ServiceTestSuite))
|
|
}
|
|
|
|
func (suite *ServiceTestSuite) TestServiceName() {
|
|
tests := map[string]struct {
|
|
values []string
|
|
expectedName string
|
|
}{
|
|
"Default": {values: nil, expectedName: "common-test"},
|
|
"CustomName": {values: []string{"service.main.nameOverride=main"}, expectedName: "common-test-main"},
|
|
}
|
|
for name, tc := range tests {
|
|
suite.Suite.Run(name, func() {
|
|
err := suite.Chart.Render(nil, tc.values, nil)
|
|
if err != nil {
|
|
suite.FailNow(err.Error())
|
|
}
|
|
|
|
serviceManifest := suite.Chart.Manifests.Get("Service", tc.expectedName)
|
|
suite.Assertions.NotEmpty(serviceManifest)
|
|
})
|
|
}
|
|
}
|
|
|
|
func (suite *ServiceTestSuite) TestPortNames() {
|
|
tests := map[string]struct {
|
|
values []string
|
|
expectedRenderFailure bool
|
|
expectedName string
|
|
expectedTargetPort interface{}
|
|
}{
|
|
"Default": {values: nil, expectedRenderFailure: false, expectedName: "main", expectedTargetPort: "main"},
|
|
"CustomName": {values: []string{"service.main.ports.main.enabled=false", "service.main.ports.server.enabled=true", "service.main.ports.server.port=8080"}, expectedRenderFailure: false, expectedName: "server", expectedTargetPort: "server"},
|
|
"CustomTargetPort": {values: []string{"service.main.ports.main.targetPort=80"}, expectedRenderFailure: false, expectedName: "main", expectedTargetPort: 80},
|
|
"NamedTargetPort": {values: []string{"service.main.ports.main.targetPort=name"}, expectedRenderFailure: true, expectedName: "", expectedTargetPort: nil},
|
|
}
|
|
for name, tc := range tests {
|
|
suite.Suite.Run(name, func() {
|
|
err := suite.Chart.Render(nil, tc.values, nil)
|
|
if tc.expectedRenderFailure {
|
|
suite.Assertions.Error(err)
|
|
return
|
|
}
|
|
if err != nil {
|
|
suite.FailNow(err.Error())
|
|
}
|
|
|
|
serviceManifest := suite.Chart.Manifests.Get("Service", "common-test")
|
|
suite.Assertions.NotEmpty(serviceManifest)
|
|
servicePorts, _ := serviceManifest.Path("spec.ports").Children()
|
|
suite.Assertions.EqualValues(tc.expectedName, servicePorts[0].Path("name").Data())
|
|
suite.Assertions.EqualValues(tc.expectedTargetPort, servicePorts[0].Path("targetPort").Data())
|
|
})
|
|
}
|
|
}
|
|
|
|
func (suite *ServiceTestSuite) TestPortProtocol() {
|
|
tests := map[string]struct {
|
|
values []string
|
|
expectedProtocol string
|
|
}{
|
|
"Default": {values: nil, expectedProtocol: "TCP"},
|
|
"ExplicitTCP": {values: []string{"service.main.ports.main.protocol=TCP"}, expectedProtocol: "TCP"},
|
|
"ExplicitHTTP": {values: []string{"service.main.ports.main.protocol=HTTP"}, expectedProtocol: "TCP"},
|
|
"ExplicitHTTPS": {values: []string{"service.main.ports.main.protocol=HTTPS"}, expectedProtocol: "TCP"},
|
|
"ExplicitUDP": {values: []string{"service.main.ports.main.protocol=UDP"}, expectedProtocol: "UDP"},
|
|
}
|
|
for name, tc := range tests {
|
|
suite.Suite.Run(name, func() {
|
|
err := suite.Chart.Render(nil, tc.values, nil)
|
|
if err != nil {
|
|
suite.FailNow(err.Error())
|
|
}
|
|
|
|
serviceManifest := suite.Chart.Manifests.Get("Service", "common-test")
|
|
suite.Assertions.NotEmpty(serviceManifest)
|
|
servicePorts, _ := serviceManifest.Path("spec.ports").Children()
|
|
suite.Assertions.EqualValues(tc.expectedProtocol, servicePorts[0].Path("protocol").Data())
|
|
})
|
|
}
|
|
}
|