mirror of
				https://github.com/optim-enterprises-bv/kubernetes.git
				synced 2025-11-04 04:08:16 +00:00 
			
		
		
		
	docker.Container directly. Conflicts: pkg/kubelet/dockertools/docker.go pkg/registry/pod/rest.go
		
			
				
	
	
		
			116 lines
		
	
	
		
			2.8 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			116 lines
		
	
	
		
			2.8 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
/*
 | 
						|
Copyright 2014 Google Inc. All rights reserved.
 | 
						|
 | 
						|
Licensed under the Apache License, Version 2.0 (the "License");
 | 
						|
you may not use this file except in compliance with the License.
 | 
						|
You may obtain a copy of the License at
 | 
						|
 | 
						|
    http://www.apache.org/licenses/LICENSE-2.0
 | 
						|
 | 
						|
Unless required by applicable law or agreed to in writing, software
 | 
						|
distributed under the License is distributed on an "AS IS" BASIS,
 | 
						|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 | 
						|
See the License for the specific language governing permissions and
 | 
						|
limitations under the License.
 | 
						|
*/
 | 
						|
 | 
						|
package client
 | 
						|
 | 
						|
import (
 | 
						|
	"encoding/json"
 | 
						|
	"net/http"
 | 
						|
	"net/http/httptest"
 | 
						|
	"net/url"
 | 
						|
	"strconv"
 | 
						|
	"strings"
 | 
						|
	"testing"
 | 
						|
 | 
						|
	"github.com/GoogleCloudPlatform/kubernetes/pkg/api"
 | 
						|
	"github.com/GoogleCloudPlatform/kubernetes/pkg/util"
 | 
						|
	"github.com/fsouza/go-dockerclient"
 | 
						|
)
 | 
						|
 | 
						|
func TestHTTPPodInfoGetter(t *testing.T) {
 | 
						|
	expectObj := api.PodInfo{
 | 
						|
		"myID": api.ContainerStatus{
 | 
						|
			DetailInfo: docker.Container{ID: "myID"},
 | 
						|
		},
 | 
						|
	}
 | 
						|
	body, err := json.Marshal(expectObj)
 | 
						|
	if err != nil {
 | 
						|
		t.Errorf("unexpected error: %v", err)
 | 
						|
	}
 | 
						|
 | 
						|
	fakeHandler := util.FakeHandler{
 | 
						|
		StatusCode:   200,
 | 
						|
		ResponseBody: string(body),
 | 
						|
	}
 | 
						|
	testServer := httptest.NewServer(&fakeHandler)
 | 
						|
 | 
						|
	hostURL, err := url.Parse(testServer.URL)
 | 
						|
	if err != nil {
 | 
						|
		t.Errorf("unexpected error: %v", err)
 | 
						|
	}
 | 
						|
 | 
						|
	parts := strings.Split(hostURL.Host, ":")
 | 
						|
 | 
						|
	port, err := strconv.Atoi(parts[1])
 | 
						|
	if err != nil {
 | 
						|
		t.Errorf("unexpected error: %v", err)
 | 
						|
	}
 | 
						|
 | 
						|
	podInfoGetter := &HTTPPodInfoGetter{
 | 
						|
		Client: http.DefaultClient,
 | 
						|
		Port:   uint(port),
 | 
						|
	}
 | 
						|
	gotObj, err := podInfoGetter.GetPodInfo(parts[0], "foo")
 | 
						|
	if err != nil {
 | 
						|
		t.Errorf("unexpected error: %v", err)
 | 
						|
	}
 | 
						|
 | 
						|
	// reflect.DeepEqual(expectObj, gotObj) doesn't handle blank times well
 | 
						|
	if len(gotObj) != len(expectObj) ||
 | 
						|
		expectObj["myID"].DetailInfo.ID != gotObj["myID"].DetailInfo.ID {
 | 
						|
		t.Errorf("Unexpected response.  Expected: %#v, received %#v", expectObj, gotObj)
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
func TestHTTPPodInfoGetterNotFound(t *testing.T) {
 | 
						|
	expectObj := api.PodInfo{
 | 
						|
		"myID": api.ContainerStatus{
 | 
						|
			DetailInfo: docker.Container{ID: "myID"},
 | 
						|
		},
 | 
						|
	}
 | 
						|
	_, err := json.Marshal(expectObj)
 | 
						|
	if err != nil {
 | 
						|
		t.Errorf("unexpected error: %v", err)
 | 
						|
	}
 | 
						|
 | 
						|
	fakeHandler := util.FakeHandler{
 | 
						|
		StatusCode:   404,
 | 
						|
		ResponseBody: "Pod not found",
 | 
						|
	}
 | 
						|
	testServer := httptest.NewServer(&fakeHandler)
 | 
						|
 | 
						|
	hostURL, err := url.Parse(testServer.URL)
 | 
						|
	if err != nil {
 | 
						|
		t.Errorf("unexpected error: %v", err)
 | 
						|
	}
 | 
						|
 | 
						|
	parts := strings.Split(hostURL.Host, ":")
 | 
						|
 | 
						|
	port, err := strconv.Atoi(parts[1])
 | 
						|
	if err != nil {
 | 
						|
		t.Errorf("unexpected error: %v", err)
 | 
						|
	}
 | 
						|
 | 
						|
	podInfoGetter := &HTTPPodInfoGetter{
 | 
						|
		Client: http.DefaultClient,
 | 
						|
		Port:   uint(port),
 | 
						|
	}
 | 
						|
	_, err = podInfoGetter.GetPodInfo(parts[0], "foo")
 | 
						|
	if err != ErrPodInfoNotAvailable {
 | 
						|
		t.Errorf("Expected %#v, Got %#v", ErrPodInfoNotAvailable, err)
 | 
						|
	}
 | 
						|
}
 |