diff --git a/probe/docker/registry_test.go b/probe/docker/registry_test.go index caba06996..73f175455 100644 --- a/probe/docker/registry_test.go +++ b/probe/docker/registry_test.go @@ -56,6 +56,10 @@ func (c *mockContainer) GetNode(_ string, _ []net.IP) report.Node { }) } +func (c *mockContainer) Container() *client.Container { + return c.c +} + func (c *mockContainer) HasTTY() bool { return true } type mockDockerClient struct { diff --git a/probe/docker/tagger_test.go b/probe/docker/tagger_test.go index f1811f6d4..1415d4e1b 100644 --- a/probe/docker/tagger_test.go +++ b/probe/docker/tagger_test.go @@ -23,6 +23,10 @@ func (m *mockProcessTree) GetParent(pid int) (int, error) { return parent, nil } +func (m *mockProcessTree) GetChildren(int) ([]int, error) { + panic("Not implemented") +} + func TestTagger(t *testing.T) { oldProcessTree := docker.NewProcessTreeStub defer func() { docker.NewProcessTreeStub = oldProcessTree }()