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) {
|
func TestHandleSigterm(t *testing.T) {
|
||||||
|
const (
|
||||||
|
podName = "test"
|
||||||
|
namespace = "test"
|
||||||
|
)
|
||||||
|
|
||||||
clientSet := fake.NewSimpleClientset()
|
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{
|
pod := corev1.Pod{
|
||||||
ObjectMeta: metav1.ObjectMeta{
|
ObjectMeta: metav1.ObjectMeta{
|
||||||
Name: name,
|
Name: podName,
|
||||||
Namespace: ns,
|
Namespace: namespace,
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
|
|
||||||
_, err := clientSet.CoreV1().Pods(ns).Create(&pod)
|
_, err := clientSet.CoreV1().Pods(namespace).Create(&pod)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatalf("error creating pod %v: %v", pod, err)
|
t.Fatalf("error creating pod %v: %v", pod, err)
|
||||||
}
|
}
|
||||||
|
|
||||||
resetForTesting(func() { t.Fatal("bad parse") })
|
resetForTesting(func() { t.Fatal("bad parse") })
|
||||||
|
|
||||||
os.Setenv("POD_NAME", name)
|
os.Setenv("POD_NAME", podName)
|
||||||
os.Setenv("POD_NAMESPACE", ns)
|
os.Setenv("POD_NAMESPACE", namespace)
|
||||||
defer os.Setenv("POD_NAME", "")
|
|
||||||
defer os.Setenv("POD_NAMESPACE", "")
|
|
||||||
|
|
||||||
oldArgs := os.Args
|
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()
|
_, conf, err := parseFlags()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Errorf("Unexpected error creating NGINX controller: %v", err)
|
t.Errorf("Unexpected error creating NGINX controller: %v", err)
|
||||||
|
@ -102,16 +106,10 @@ func TestHandleSigterm(t *testing.T) {
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Error("Unexpected error sending SIGTERM signal.")
|
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 {
|
func createConfigMap(clientSet kubernetes.Interface, ns string, t *testing.T) string {
|
||||||
t.Helper()
|
t.Helper()
|
||||||
t.Log("Creating temporal config map")
|
|
||||||
|
|
||||||
configMap := &corev1.ConfigMap{
|
configMap := &corev1.ConfigMap{
|
||||||
ObjectMeta: metav1.ObjectMeta{
|
ObjectMeta: metav1.ObjectMeta{
|
||||||
|
@ -124,18 +122,6 @@ func createConfigMap(clientSet kubernetes.Interface, ns string, t *testing.T) st
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Errorf("error creating the configuration map: %v", err)
|
t.Errorf("error creating the configuration map: %v", err)
|
||||||
}
|
}
|
||||||
t.Logf("Temporal configmap %v created", cm)
|
|
||||||
|
|
||||||
return cm.Name
|
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