Fix tests

weaveworks · Oct 24, 2016 · 9881a6b · 9881a6b
1 parent 529759b
commit 9881a6b
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 2 deletions.
diff --git a/app/api_topologies_test.go b/app/api_topologies_test.go
@@ -83,6 +83,9 @@ func TestAPITopologyAddsKubernetes(t *testing.T) {
 				{ContainerID: "container2"},
 			},
 		},
+		Spec: api.PodSpec{
+			SecurityContext: &api.PodSecurityContext{},
+		},
 	}).GetNode("")
 	buf := &bytes.Buffer{}
 	encoder := codec.NewEncoder(buf, &codec.MsgpackHandle{})

diff --git a/probe/kubernetes/pod.go b/probe/kubernetes/pod.go
@@ -64,7 +64,7 @@ func (p *pod) GetNode(probeID string) report.Node {
 		report.ControlProbeID: probeID,
 	}
 
-	if p.Pod.Spec.SecurityContext.HostNetwork {
+	if sc := p.Pod.Spec.SecurityContext; sc != nil && sc.HostNetwork {
 		latests[IsInHostNetwork] = "true"
 	}
 

diff --git a/probe/kubernetes/reporter_test.go b/probe/kubernetes/reporter_test.go
@@ -46,6 +46,9 @@ var (
 		},
 		Spec: api.PodSpec{
 			NodeName: nodeName,
+			SecurityContext: &api.PodSecurityContext{
+				HostNetwork: true,
+			},
 		},
 	}
 	apiPod2 = api.Pod{
@@ -65,7 +68,8 @@ var (
 			},
 		},
 		Spec: api.PodSpec{
-			NodeName: nodeName,
+			NodeName:        nodeName,
+			SecurityContext: &api.PodSecurityContext{},
 		},
 	}
 	apiService1 = api.Service{