DevFW-CICD/ingress-nginx-helm
6
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Move the resetting logic into framework

Browse source 
Stylistic fixes based on feedback
This commit is contained in:
Andrew Louis 2018-04-16 15:30:35 -04:00
parent 19337f05fb
commit 444914b764
4 changed files with 22 additions and 47 deletions
Download patch file Download diff file Expand all files Collapse all files

31
test/e2e/framework/framework.go
View file

@ -61,6 +61,9 @@ type Framework struct {
// should abort, the AfterSuite hook should run all Cleanup actions.
cleanupHandle CleanupActionHandle
// Map to hold the default nginx-configuration configmap data before a test run
DefaultNginxConfigMapData map[string]string
NginxHTTPURL string
NginxHTTPSURL string
}
@ -69,7 +72,8 @@ type Framework struct {
// you (you can write additional before/after each functions).
func NewDefaultFramework(baseName string) *Framework {
f := &Framework{
BaseName: baseName,
BaseName: baseName,
DefaultNginxConfigMapData: nil,
}
BeforeEach(f.BeforeEach)
@ -101,6 +105,13 @@ func (f *Framework) BeforeEach() {
By("Building NGINX HTTPS URL")
f.NginxHTTPSURL, err = f.GetNginxURL(HTTPS)
Expect(err).NotTo(HaveOccurred())
By("Persist nginx-configuration configMap Data")
if f.DefaultNginxConfigMapData == nil {
f.DefaultNginxConfigMapData, err = f.GetNginxConfigMapData()
Expect(err).NotTo(HaveOccurred())
Expect(f.DefaultNginxConfigMapData).NotTo(BeNil())
}
}
// AfterEach deletes the namespace, after reading its events.
@ -114,6 +125,10 @@ func (f *Framework) AfterEach() {
By("Waiting for test namespace to no longer exist")
err = WaitForNoPodsInNamespace(f.KubeClientSet, f.Namespace.Name)
Expect(err).NotTo(HaveOccurred())
By("Reset nginx-configuration configMap to default state")
err = f.SetNginxConfigMapData(f.DefaultNginxConfigMapData)
Expect(err).NotTo(HaveOccurred())
}
// IngressNginxDescribe wrapper function for ginkgo describe. Adds namespacing.
@ -259,8 +274,7 @@ func (f *Framework) matchNginxConditions(name string, matcher func(cfg string) b
}
}
// GetNginxConfigMap gets ingress-nginx's nginx-configuration map
func (f *Framework) GetNginxConfigMap() (*v1.ConfigMap, error) {
func (f *Framework) getNginxConfigMap() (*v1.ConfigMap, error) {
if f.KubeClientSet == nil {
return nil, fmt.Errorf("KubeClientSet not initialized")
}
@ -270,22 +284,15 @@ func (f *Framework) GetNginxConfigMap() (*v1.ConfigMap, error) {
return nil, err
}
time.Sleep(1 * time.Second)
return config, err
}
// GetNginxConfigMapData gets ingress-nginx's nginx-configuration map's data
func (f *Framework) GetNginxConfigMapData() (map[string]string, error) {
config, err := f.GetNginxConfigMap()
config, err := f.getNginxConfigMap()
if err != nil {
return nil, err
}
if config == nil {
return nil, fmt.Errorf("Unable to get nginx-configuration configMap")
}
if config.Data == nil {
config.Data = map[string]string{}
}
@ -297,7 +304,7 @@ func (f *Framework) GetNginxConfigMapData() (map[string]string, error) {
func (f *Framework) SetNginxConfigMapData(cmData map[string]string) error {
// Needs to do a Get and Set, Update will not take just the Data field
// or a configMap that is not the very last revision
config, err := f.GetNginxConfigMap()
config, err := f.getNginxConfigMap()
if err != nil {
return err
}