4ae2c3a101
This PS makes sure we have linter and unit tests processed. The code has been reformatted to adhere to Go's code formatting conventions. Change-Id: I31f15d6d6c4b9bda7e3837941b6c9c3c3735aea7
115 lines
2.3 KiB
Go
115 lines
2.3 KiB
Go
package entrypoint
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"io"
|
|
"os"
|
|
"time"
|
|
|
|
. "github.com/onsi/ginkgo"
|
|
. "github.com/onsi/gomega"
|
|
|
|
cli "opendev.org/airship/kubernetes-entrypoint/client"
|
|
"opendev.org/airship/kubernetes-entrypoint/logger"
|
|
"opendev.org/airship/kubernetes-entrypoint/mocks"
|
|
)
|
|
|
|
const (
|
|
testNamespace = "test"
|
|
dummyResolverName = "dummy"
|
|
loggerInfoText = "Entrypoint INFO: "
|
|
)
|
|
|
|
var (
|
|
testEntrypoint EntrypointInterface
|
|
testClient cli.ClientInterface
|
|
)
|
|
|
|
type dummyResolver struct {
|
|
name string
|
|
namespace string
|
|
}
|
|
|
|
func (d dummyResolver) IsResolved(ctx context.Context, entry EntrypointInterface) (bool, error) {
|
|
return true, nil
|
|
}
|
|
|
|
func (d dummyResolver) GetName() (name string) {
|
|
return d.name
|
|
}
|
|
|
|
func (d dummyResolver) String() string {
|
|
return fmt.Sprintf("Dummy %s in namespace %s", d.name, d.namespace)
|
|
}
|
|
|
|
func init() {
|
|
testClient = mocks.NewClient()
|
|
testEntrypoint = mocks.NewEntrypointInNamespace(testNamespace)
|
|
}
|
|
|
|
func registerNilResolver() {
|
|
Register(nil)
|
|
}
|
|
|
|
var _ = Describe("Entrypoint", func() {
|
|
dummy := dummyResolver{name: dummyResolverName}
|
|
|
|
BeforeEach(func() {
|
|
logger.Info.SetFlags(0)
|
|
logger.Warning.SetFlags(0)
|
|
logger.Error.SetFlags(0)
|
|
})
|
|
|
|
AfterEach(func() {
|
|
// Clear dependencies
|
|
dependencies = make([]Resolver, 0)
|
|
})
|
|
|
|
It("registers new nil resolver", func() {
|
|
defer GinkgoRecover()
|
|
|
|
Ω(registerNilResolver).Should(Panic())
|
|
})
|
|
|
|
It("registers new non-nil resolver", func() {
|
|
defer GinkgoRecover()
|
|
Register(dummy)
|
|
Expect(len(dependencies)).To(Equal(1))
|
|
})
|
|
|
|
It("checks Client() method", func() {
|
|
client := testEntrypoint.Client()
|
|
Expect(client).To(Equal(testClient))
|
|
})
|
|
|
|
It("resolves main entrypoint with a dummy dependency", func() {
|
|
defer GinkgoRecover()
|
|
|
|
// Set output logger to our reader
|
|
r, w, _ := os.Pipe()
|
|
tmp := os.Stdout
|
|
defer func() {
|
|
os.Stdout = tmp
|
|
}()
|
|
|
|
logger.Info.SetOutput(w)
|
|
|
|
os.Stdout = w
|
|
go func() {
|
|
mainEntrypoint := Entrypoint{client: mocks.NewClient(), namespace: "main"}
|
|
Register(dummy)
|
|
mainEntrypoint.Resolve()
|
|
w.Close()
|
|
}()
|
|
|
|
// Wait for resolver to finish
|
|
time.Sleep(5 * time.Second)
|
|
|
|
stdout, _ := io.ReadAll(r)
|
|
resolvedString := fmt.Sprintf("%sResolving %v\n%sDependency %v is resolved.\n",
|
|
loggerInfoText, dummy, loggerInfoText, dummy)
|
|
Expect(string(stdout)).To(Equal(resolvedString))
|
|
})
|
|
})
|