mirror of
https://github.com/k3s-io/kubernetes.git
synced 2025-07-24 12:15:52 +00:00
Converted TestPodHasServiceEnvVars to native ginkgo syntax #3941
This commit is contained in:
Robert Rati
parent
eb5d340851
commit
da2594710a
@ -17,156 +17,154 @@ limitations under the License.
|
||||
package e2e
|
||||
|
||||
import (
|
||||
"strings"
|
||||
"fmt"
|
||||
"time"
|
||||
|
||||
"github.com/GoogleCloudPlatform/kubernetes/pkg/api"
|
||||
"github.com/GoogleCloudPlatform/kubernetes/pkg/client"
|
||||
"github.com/golang/glog"
|
||||
|
||||
. "github.com/onsi/ginkgo"
|
||||
. "github.com/onsi/gomega"
|
||||
)
|
||||
|
||||
// TestPodHasServiceEnvVars checks that kubelets and scheduler send events about pods scheduling and running.
|
||||
func TestPodHasServiceEnvVars(c *client.Client) bool {
|
||||
// Make a pod that will be a service.
|
||||
// This pod serves its hostname via HTTP.
|
||||
serverPod := parsePodOrDie(`{
|
||||
"kind": "Pod",
|
||||
"apiVersion": "v1beta1",
|
||||
"id": "srv",
|
||||
"desiredState": {
|
||||
"manifest": {
|
||||
"version": "v1beta1",
|
||||
"id": "srv",
|
||||
"containers": [{
|
||||
"name": "srv",
|
||||
"image": "kubernetes/serve_hostname",
|
||||
"ports": [{
|
||||
"containerPort": 9376,
|
||||
"hostPort": 8080
|
||||
}]
|
||||
}]
|
||||
}
|
||||
},
|
||||
"labels": {
|
||||
"name": "srv"
|
||||
}
|
||||
}`)
|
||||
_, err := c.Pods(api.NamespaceDefault).Create(serverPod)
|
||||
if err != nil {
|
||||
glog.Errorf("Failed to create serverPod: %v", err)
|
||||
return false
|
||||
}
|
||||
defer c.Pods(api.NamespaceDefault).Delete(serverPod.Name)
|
||||
waitForPodRunning(c, serverPod.Name)
|
||||
|
||||
// This service exposes pod p's port 8080 as a service on port 8765
|
||||
svc := parseServiceOrDie(`{
|
||||
"id": "fooservice",
|
||||
"kind": "Service",
|
||||
"apiVersion": "v1beta1",
|
||||
"port": 8765,
|
||||
"containerPort": 8080,
|
||||
"selector": {
|
||||
"name": "p"
|
||||
}
|
||||
}`)
|
||||
if err != nil {
|
||||
glog.Errorf("Failed to delete service: %v", err)
|
||||
return false
|
||||
}
|
||||
time.Sleep(2)
|
||||
_, err = c.Services(api.NamespaceDefault).Create(svc)
|
||||
if err != nil {
|
||||
glog.Errorf("Failed to create service: %v", err)
|
||||
return false
|
||||
}
|
||||
defer c.Services(api.NamespaceDefault).Delete(svc.Name)
|
||||
|
||||
// TODO: we don't have a way to wait for a service to be "running".
|
||||
// If this proves flaky, then we will need to retry the clientPod or insert a sleep.
|
||||
|
||||
// Make a client pod that verifies that it has the service environment variables.
|
||||
clientPod := parsePodOrDie(`{
|
||||
"apiVersion": "v1beta1",
|
||||
"kind": "Pod",
|
||||
"id": "env3",
|
||||
"desiredState": {
|
||||
"manifest": {
|
||||
"version": "v1beta1",
|
||||
"id": "env3",
|
||||
"restartPolicy": { "never": {} },
|
||||
"containers": [{
|
||||
"name": "env3cont",
|
||||
"image": "busybox",
|
||||
"command": ["sh", "-c", "env"]
|
||||
}]
|
||||
}
|
||||
},
|
||||
"labels": { "name": "env3" }
|
||||
}`)
|
||||
_, err = c.Pods(api.NamespaceDefault).Create(clientPod)
|
||||
if err != nil {
|
||||
glog.Errorf("Failed to create pod: %v", err)
|
||||
return false
|
||||
}
|
||||
defer c.Pods(api.NamespaceDefault).Delete(clientPod.Name)
|
||||
|
||||
// Wait for client pod to complete.
|
||||
success := waitForPodSuccess(c, clientPod.Name, clientPod.Spec.Containers[0].Name)
|
||||
if !success {
|
||||
glog.Errorf("Failed to run client pod to detect service env vars.")
|
||||
}
|
||||
|
||||
// Grab its logs. Get host first.
|
||||
clientPodStatus, err := c.Pods(api.NamespaceDefault).Get(clientPod.Name)
|
||||
if err != nil {
|
||||
glog.Errorf("Failed to get clientPod to know host: %v", err)
|
||||
return false
|
||||
}
|
||||
glog.Infof("Trying to get logs from host %s pod %s container %s: %v",
|
||||
clientPodStatus.Status.Host, clientPodStatus.Name, clientPodStatus.Spec.Containers[0].Name, err)
|
||||
logs, err := c.Get().
|
||||
Prefix("proxy").
|
||||
Resource("minions").
|
||||
Name(clientPodStatus.Status.Host).
|
||||
Suffix("containerLogs", api.NamespaceDefault, clientPodStatus.Name, clientPodStatus.Spec.Containers[0].Name).
|
||||
Do().
|
||||
Raw()
|
||||
if err != nil {
|
||||
glog.Errorf("Failed to get logs from host %s pod %s container %s: %v",
|
||||
clientPodStatus.Status.Host, clientPodStatus.Name, clientPodStatus.Spec.Containers[0].Name, err)
|
||||
return false
|
||||
}
|
||||
glog.Info("clientPod logs:", string(logs))
|
||||
|
||||
toFind := []string{
|
||||
"FOOSERVICE_SERVICE_HOST=",
|
||||
"FOOSERVICE_SERVICE_PORT=",
|
||||
"FOOSERVICE_PORT=",
|
||||
"FOOSERVICE_PORT_8765_TCP_PORT=",
|
||||
"FOOSERVICE_PORT_8765_TCP_PROTO=",
|
||||
"FOOSERVICE_PORT_8765_TCP=",
|
||||
"FOOSERVICE_PORT_8765_TCP_ADDR=",
|
||||
}
|
||||
|
||||
for _, m := range toFind {
|
||||
if !strings.Contains(string(logs), m) {
|
||||
glog.Errorf("Unable to find env var %q in client env vars.", m)
|
||||
success = false
|
||||
}
|
||||
}
|
||||
|
||||
// We could try a wget the service from the client pod. But services.sh e2e test covers that pretty well.
|
||||
return success
|
||||
}
|
||||
|
||||
var _ = Describe("TestPodHasServiceEnvVars", func() {
|
||||
It("should pass", func() {
|
||||
// TODO: Instead of OrDie, client should Fail the test if there's a problem.
|
||||
// In general tests should Fail() instead of glog.Fatalf().
|
||||
Expect(TestPodHasServiceEnvVars(loadClientOrDie())).To(BeTrue())
|
||||
var (
|
||||
c *client.Client
|
||||
)
|
||||
|
||||
BeforeEach(func() {
|
||||
c = loadClientOrDie()
|
||||
})
|
||||
|
||||
It("should contain environment variables for services", func() {
|
||||
// Make a pod that will be a service.
|
||||
// This pod serves its hostname via HTTP.
|
||||
serverPod := parsePodOrDie(`{
|
||||
"kind": "Pod",
|
||||
"apiVersion": "v1beta1",
|
||||
"id": "srv",
|
||||
"desiredState": {
|
||||
"manifest": {
|
||||
"version": "v1beta1",
|
||||
"id": "srv",
|
||||
"containers": [{
|
||||
"name": "srv",
|
||||
"image": "kubernetes/serve_hostname",
|
||||
"ports": [{
|
||||
"containerPort": 9376,
|
||||
"hostPort": 8080
|
||||
}]
|
||||
}]
|
||||
}
|
||||
},
|
||||
"labels": {
|
||||
"name": "srv"
|
||||
}
|
||||
}`)
|
||||
_, err := c.Pods(api.NamespaceDefault).Create(serverPod)
|
||||
if err != nil {
|
||||
Fail(fmt.Sprintf("Failed to create serverPod: %v", err))
|
||||
}
|
||||
defer func() {
|
||||
defer GinkgoRecover()
|
||||
c.Pods(api.NamespaceDefault).Delete(serverPod.Name)
|
||||
}()
|
||||
waitForPodRunning(c, serverPod.Name)
|
||||
|
||||
// This service exposes pod p's port 8080 as a service on port 8765
|
||||
svc := parseServiceOrDie(`{
|
||||
"id": "fooservice",
|
||||
"kind": "Service",
|
||||
"apiVersion": "v1beta1",
|
||||
"port": 8765,
|
||||
"containerPort": 8080,
|
||||
"selector": {
|
||||
"name": "p"
|
||||
}
|
||||
}`)
|
||||
if err != nil {
|
||||
Fail(fmt.Sprintf("Failed to delete service: %v", err))
|
||||
}
|
||||
time.Sleep(2)
|
||||
_, err = c.Services(api.NamespaceDefault).Create(svc)
|
||||
if err != nil {
|
||||
Fail(fmt.Sprintf("Failed to create service: %v", err))
|
||||
}
|
||||
defer func() {
|
||||
defer GinkgoRecover()
|
||||
c.Services(api.NamespaceDefault).Delete(svc.Name)
|
||||
}()
|
||||
|
||||
// TODO: we don't have a way to wait for a service to be "running". // If this proves flaky, then we will need to retry the clientPod or insert a sleep.
|
||||
|
||||
// Make a client pod that verifies that it has the service environment variables.
|
||||
clientPod := parsePodOrDie(`{
|
||||
"apiVersion": "v1beta1",
|
||||
"kind": "Pod",
|
||||
"id": "env3",
|
||||
"desiredState": {
|
||||
"manifest": {
|
||||
"version": "v1beta1",
|
||||
"id": "env3",
|
||||
"restartPolicy": { "never": {} },
|
||||
"containers": [{
|
||||
"name": "env3cont",
|
||||
"image": "busybox",
|
||||
"command": ["sh", "-c", "env"]
|
||||
}]
|
||||
}
|
||||
},
|
||||
"labels": { "name": "env3" }
|
||||
}`)
|
||||
_, err = c.Pods(api.NamespaceDefault).Create(clientPod)
|
||||
if err != nil {
|
||||
Fail(fmt.Sprintf("Failed to create pod: %v", err))
|
||||
}
|
||||
defer func() {
|
||||
defer GinkgoRecover()
|
||||
c.Pods(api.NamespaceDefault).Delete(clientPod.Name)
|
||||
}()
|
||||
|
||||
// Wait for client pod to complete.
|
||||
success := waitForPodSuccess(c, clientPod.Name, clientPod.Spec.Containers[0].Name)
|
||||
if !success {
|
||||
Fail(fmt.Sprintf("Failed to run client pod to detect service env vars."))
|
||||
}
|
||||
|
||||
// Grab its logs. Get host first.
|
||||
clientPodStatus, err := c.Pods(api.NamespaceDefault).Get(clientPod.Name)
|
||||
if err != nil {
|
||||
Fail(fmt.Sprintf("Failed to get clientPod to know host: %v", err))
|
||||
}
|
||||
By(fmt.Sprintf("Trying to get logs from host %s pod %s container %s: %v",
|
||||
clientPodStatus.Status.Host, clientPodStatus.Name, clientPodStatus.Spec.Containers[0].Name, err))
|
||||
logs, err := c.Get().
|
||||
Prefix("proxy").
|
||||
Resource("minions").
|
||||
Name(clientPodStatus.Status.Host).
|
||||
Suffix("containerLogs", api.NamespaceDefault, clientPodStatus.Name, clientPodStatus.Spec.Containers[0].Name).
|
||||
Do().
|
||||
Raw()
|
||||
if err != nil {
|
||||
Fail(fmt.Sprintf("Failed to get logs from host %s pod %s container %s: %v",
|
||||
clientPodStatus.Status.Host, clientPodStatus.Name, clientPodStatus.Spec.Containers[0].Name, err))
|
||||
}
|
||||
fmt.Sprintf("clientPod logs:%v\n", string(logs))
|
||||
|
||||
toFind := []string{
|
||||
"FOOSERVICE_SERVICE_HOST=",
|
||||
"FOOSERVICE_SERVICE_PORT=",
|
||||
"FOOSERVICE_PORT=",
|
||||
"FOOSERVICE_PORT_8765_TCP_PORT=",
|
||||
"FOOSERVICE_PORT_8765_TCP_PROTO=",
|
||||
"FOOSERVICE_PORT_8765_TCP=",
|
||||
"FOOSERVICE_PORT_8765_TCP_ADDR=",
|
||||
}
|
||||
|
||||
for _, m := range toFind {
|
||||
Expect(string(logs)).To(ContainSubstring(m), "%q in client env vars", m)
|
||||
}
|
||||
|
||||
// We could try a wget the service from the client pod. But services.sh e2e test covers that pretty well.
|
||||
})
|
||||
})
|
||||
|
||||
Loading…
Reference in New Issue
Block a user