Cleanup test
This commit is contained in:
parent
fe2ae8a1ec
commit
1d46ec2eb2
1 changed files with 18 additions and 32 deletions
|
@ -50,37 +50,41 @@ func init() {
|
|||
}
|
||||
|
||||
func TestHandleSigterm(t *testing.T) {
|
||||
const (
|
||||
podName = "test"
|
||||
namespace = "test"
|
||||
)
|
||||
|
||||
clientSet := fake.NewSimpleClientset()
|
||||
|
||||
ns := "test"
|
||||
createConfigMap(clientSet, namespace, t)
|
||||
|
||||
cm := createConfigMap(clientSet, ns, t)
|
||||
defer deleteConfigMap(cm, ns, clientSet, t)
|
||||
|
||||
name := "test"
|
||||
pod := corev1.Pod{
|
||||
ObjectMeta: metav1.ObjectMeta{
|
||||
Name: name,
|
||||
Namespace: ns,
|
||||
Name: podName,
|
||||
Namespace: namespace,
|
||||
},
|
||||
}
|
||||
|
||||
_, err := clientSet.CoreV1().Pods(ns).Create(&pod)
|
||||
_, err := clientSet.CoreV1().Pods(namespace).Create(&pod)
|
||||
if err != nil {
|
||||
t.Fatalf("error creating pod %v: %v", pod, err)
|
||||
}
|
||||
|
||||
resetForTesting(func() { t.Fatal("bad parse") })
|
||||
|
||||
os.Setenv("POD_NAME", name)
|
||||
os.Setenv("POD_NAMESPACE", ns)
|
||||
defer os.Setenv("POD_NAME", "")
|
||||
defer os.Setenv("POD_NAMESPACE", "")
|
||||
os.Setenv("POD_NAME", podName)
|
||||
os.Setenv("POD_NAMESPACE", namespace)
|
||||
|
||||
oldArgs := os.Args
|
||||
defer func() { os.Args = oldArgs }()
|
||||
os.Args = []string{"cmd", "--default-backend-service", "ingress-nginx/default-backend-http", "--http-port", "0", "--https-port", "0"}
|
||||
|
||||
defer func() {
|
||||
os.Setenv("POD_NAME", "")
|
||||
os.Setenv("POD_NAMESPACE", "")
|
||||
os.Args = oldArgs
|
||||
}()
|
||||
|
||||
os.Args = []string{"cmd", "--default-backend-service", "ingress-nginx/default-backend-http", "--http-port", "0", "--https-port", "0"}
|
||||
_, conf, err := parseFlags()
|
||||
if err != nil {
|
||||
t.Errorf("Unexpected error creating NGINX controller: %v", err)
|
||||
|
@ -102,16 +106,10 @@ func TestHandleSigterm(t *testing.T) {
|
|||
if err != nil {
|
||||
t.Error("Unexpected error sending SIGTERM signal.")
|
||||
}
|
||||
|
||||
err = clientSet.CoreV1().Pods(ns).Delete(name, &metav1.DeleteOptions{})
|
||||
if err != nil {
|
||||
t.Fatalf("error deleting pod %v: %v", pod, err)
|
||||
}
|
||||
}
|
||||
|
||||
func createConfigMap(clientSet kubernetes.Interface, ns string, t *testing.T) string {
|
||||
t.Helper()
|
||||
t.Log("Creating temporal config map")
|
||||
|
||||
configMap := &corev1.ConfigMap{
|
||||
ObjectMeta: metav1.ObjectMeta{
|
||||
|
@ -124,18 +122,6 @@ func createConfigMap(clientSet kubernetes.Interface, ns string, t *testing.T) st
|
|||
if err != nil {
|
||||
t.Errorf("error creating the configuration map: %v", err)
|
||||
}
|
||||
t.Logf("Temporal configmap %v created", cm)
|
||||
|
||||
return cm.Name
|
||||
}
|
||||
|
||||
func deleteConfigMap(cm, ns string, clientSet kubernetes.Interface, t *testing.T) {
|
||||
t.Helper()
|
||||
t.Logf("Deleting temporal configmap %v", cm)
|
||||
|
||||
err := clientSet.CoreV1().ConfigMaps(ns).Delete(cm, &metav1.DeleteOptions{})
|
||||
if err != nil {
|
||||
t.Errorf("error deleting the configmap: %v", err)
|
||||
}
|
||||
t.Logf("Temporal configmap %v deleted", cm)
|
||||
}
|
||||
|
|
Loading…
Reference in a new issue