kubelet server test cases

pkg/kubelet/server_test.go

@@ -381,3 +381,121 @@ func TestServeRunInContainerWithUUID(t *testing.T) {
 		t.Errorf("expected %s, got %s", output, result)
 	}
 }
+
+func TestContainerLogs(t *testing.T) {
+    fw := newServerTest()
+    output := "foo bar"
+    podName := "foo"
+    expectedPodName := podName + ".etcd"
+    expectedContainerName := "baz"
+    expectedTail := ""
+    expectedFollow := false
+    // expected := api.Container{"goodpod": docker.Container{ID: "myContainerID"}}
+    fw.fakeKubelet.containerLogsFunc = func(podFullName, containerName, tail string, follow bool, writer io.Writer) error {
+            if podFullName != expectedPodName {
+                    t.Errorf("expected %s, got %s", expectedPodName, podFullName)
+            }
+			if containerName != expectedContainerName {
+				t.Errorf("expected %s, got %s", expectedContainerName, containerName)
+			}
+            if tail != expectedTail {
+                    t.Errorf("expected %s, got %s", expectedTail, tail)
+            }
+            if follow != expectedFollow {
+                    t.Errorf("expected %t, got %t", expectedFollow, follow)
+            }
+            return nil
+    }
+    resp, err := http.Get(fw.testHTTPServer.URL+"/containerLogs/" + podName + "/" + expectedContainerName)
+    if err != nil {
+            t.Errorf("Got error GETing: %v", err)
+    }
+    defer resp.Body.Close()
+
+    body, err := ioutil.ReadAll(resp.Body)
+    if err != nil {
+            t.Errorf("Error reading container logs: %v", err)
+    }
+    result := string(body)
+    if result != string(body) {
+            t.Errorf("Expected: '%v', got: '%v'", output, result)
+    }
+}
+
+func TestContainerLogsWithTail(t *testing.T) {
+    fw := newServerTest()
+    output := "foo bar"
+    podName := "foo"
+    expectedPodName := podName + ".etcd"
+    expectedContainerName := "baz"
+    expectedTail := "5"
+    expectedFollow := false
+    fw.fakeKubelet.containerLogsFunc = func(podFullName, containerName, tail string, follow bool, writer io.Writer) error {
+            if podFullName != expectedPodName {
+                    t.Errorf("expected %s, got %s", expectedPodName, podFullName)
+            }
+			if containerName != expectedContainerName {
+				t.Errorf("expected %s, got %s", expectedContainerName, containerName)
+			}
+            if tail != expectedTail {
+                    t.Errorf("expected %s, got %s", expectedTail, tail)
+            }
+            if follow != expectedFollow {
+                    t.Errorf("expected %t, got %t", expectedFollow, follow)
+            }
+            return nil
+    }
+    resp, err := http.Get(fw.testHTTPServer.URL+"/containerLogs/" + podName + "/" + expectedContainerName + "?tail=5")
+    if err != nil {
+            t.Errorf("Got error GETing: %v", err)
+    }
+    defer resp.Body.Close()
+
+    body, err := ioutil.ReadAll(resp.Body)
+    if err != nil {
+            t.Errorf("Error reading container logs: %v", err)
+    }
+    result := string(body)
+    if result != string(body) {
+            t.Errorf("Expected: '%v', got: '%v'", output, result)
+    }
+}
+
+func TestContainerLogsWithFollow(t *testing.T) {
+    fw := newServerTest()
+    output := "foo bar"
+    podName := "foo"
+    expectedPodName := podName + ".etcd"
+    expectedContainerName := "baz"
+    expectedTail := ""
+    expectedFollow := true
+    fw.fakeKubelet.containerLogsFunc = func(podFullName, containerName, tail string, follow bool, writer io.Writer) error {
+            if podFullName != expectedPodName {
+                    t.Errorf("expected %s, got %s", expectedPodName, podFullName)
+            }
+			if containerName != expectedContainerName {
+				t.Errorf("expected %s, got %s", expectedContainerName, containerName)
+			}
+            if tail != expectedTail {
+                    t.Errorf("expected %s, got %s", expectedTail, tail)
+            }
+            if follow != expectedFollow {
+                    t.Errorf("expected %t, got %t", expectedFollow, follow)
+            }
+            return nil
+    }
+    resp, err := http.Get(fw.testHTTPServer.URL+"/containerLogs/" + podName + "/" + expectedContainerName + "?follow=1")
+    if err != nil {
+            t.Errorf("Got error GETing: %v", err)
+    }
+    defer resp.Body.Close()
+
+    body, err := ioutil.ReadAll(resp.Body)
+    if err != nil {
+            t.Errorf("Error reading container logs: %v", err)
+    }
+    result := string(body)
+    if result != string(body) {
+            t.Errorf("Expected: '%v', got: '%v'", output, result)
+    }
+}