From 4df4b0ee2e5c1988c587778a06e0aff767b69d1c Mon Sep 17 00:00:00 2001
From: Nan Deng <dengnan@google.com>
Date: Thu, 3 Jul 2014 22:08:24 -0700
Subject: [PATCH] unit tests for containerData

---
 manager/container_test.go | 136 ++++++++++++++++++++++++++++++++++++++
 1 file changed, 136 insertions(+)
 create mode 100644 manager/container_test.go

diff --git a/manager/container_test.go b/manager/container_test.go
new file mode 100644
index 00000000..a67cb34b
--- /dev/null
+++ b/manager/container_test.go
@@ -0,0 +1,136 @@
+// 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.
+
+// Per-container manager.
+
+package manager
+
+import (
+	"fmt"
+	"testing"
+	"time"
+
+	"github.com/google/cadvisor/container"
+	ctest "github.com/google/cadvisor/container/test"
+	"github.com/google/cadvisor/info"
+	itest "github.com/google/cadvisor/info/test"
+	"github.com/google/cadvisor/storage"
+	stest "github.com/google/cadvisor/storage/test"
+)
+
+func createContainerDataAndSetHandler(
+	driver storage.StorageDriver,
+	f func(*ctest.MockContainerHandler),
+	t *testing.T,
+) *containerData {
+	factory := &ctest.FactoryForMockContainerHandler{
+		Name: "factoryForMockContainer",
+		PrepareContainerHandlerFunc: func(name string, handler *ctest.MockContainerHandler) {
+			handler.Name = name
+			f(handler)
+		},
+	}
+	container.RegisterContainerHandlerFactory("/", factory)
+
+	if driver == nil {
+		driver = &stest.MockStorageDriver{}
+	}
+
+	ret, err := NewContainerData("/container", driver)
+	if err != nil {
+		t.Fatal(err)
+	}
+	return ret
+}
+
+func TestContainerUpdateSubcontainers(t *testing.T) {
+	var handler *ctest.MockContainerHandler
+	subcontainers := []info.ContainerReference{
+		{Name: "/container/ee0103"},
+		{Name: "/container/abcd"},
+		{Name: "/container/something"},
+	}
+	cd := createContainerDataAndSetHandler(
+		nil,
+		func(h *ctest.MockContainerHandler) {
+			h.On("ListContainers", container.LIST_SELF).Return(
+				subcontainers,
+				nil,
+			)
+			handler = h
+		},
+		t,
+	)
+
+	err := cd.updateSubcontainers()
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	handler.AssertExpectations(t)
+}
+
+func TestContainerUpdateSubcontainersWithError(t *testing.T) {
+	var handler *ctest.MockContainerHandler
+	cd := createContainerDataAndSetHandler(
+		nil,
+		func(h *ctest.MockContainerHandler) {
+			h.On("ListContainers", container.LIST_SELF).Return(
+				[]info.ContainerReference{},
+				fmt.Errorf("some error"),
+			)
+			handler = h
+		},
+		t,
+	)
+
+	err := cd.updateSubcontainers()
+	if err == nil {
+		t.Fatal("updateSubcontainers should return error")
+	}
+
+	handler.AssertExpectations(t)
+}
+
+func TestContainerUpdateStats(t *testing.T) {
+	var handler *ctest.MockContainerHandler
+	var ref info.ContainerReference
+
+	driver := &stest.MockStorageDriver{}
+
+	statsList := itest.GenerateRandomStats(1, 4, 1*time.Second)
+	stats := statsList[0]
+
+	cd := createContainerDataAndSetHandler(
+		driver,
+		func(h *ctest.MockContainerHandler) {
+			h.On("GetStats").Return(
+				stats,
+				nil,
+			)
+			handler = h
+			ref.Name = h.Name
+		},
+		t,
+	)
+
+	driver.On("AddStats", ref, stats).Return(nil)
+
+	err := cd.updateStats()
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	handler.AssertExpectations(t)
+}