From cd8d8f2dce35a1d4c03a85cb6910454492806e30 Mon Sep 17 00:00:00 2001 From: Arnav Sankaran Date: Mon, 15 Jul 2019 09:56:55 -0700 Subject: [PATCH] Add tracing to GetContainerStats --- pkg/kubelet/dockershim/libdocker/fake_client.go | 15 +++++++++++++-- 1 file changed, 13 insertions(+), 2 deletions(-) diff --git a/pkg/kubelet/dockershim/libdocker/fake_client.go b/pkg/kubelet/dockershim/libdocker/fake_client.go index 7e95cf36516..cc063cc8c3d 100644 --- a/pkg/kubelet/dockershim/libdocker/fake_client.go +++ b/pkg/kubelet/dockershim/libdocker/fake_client.go @@ -77,6 +77,7 @@ type FakeDockerClient struct { execCmd []string EnableSleep bool ImageHistoryMap map[string][]dockerimagetypes.HistoryResponseItem + ContainerStatsMap map[string]*dockertypes.StatsJSON } const ( @@ -913,9 +914,19 @@ func (f *FakeDockerPuller) GetImageRef(image string) (string, error) { return image, err } +func (f *FakeDockerClient) InjectContainerStats(data map[string]*dockertypes.StatsJSON) { + f.Lock() + defer f.Unlock() + f.ContainerStatsMap = data +} + func (f *FakeDockerClient) GetContainerStats(id string) (*dockertypes.StatsJSON, error) { f.Lock() defer f.Unlock() - f.appendCalled(CalledDetail{name: "getContainerStats"}) - return nil, fmt.Errorf("not implemented") + f.appendCalled(CalledDetail{name: "get_container_stats"}) + stats, ok := f.ContainerStatsMap[id] + if !ok { + return nil, fmt.Errorf("container %q not found", id) + } + return stats, nil }