util_test.go 2.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172
  1. // +build linux
  2. package daemon
  3. import (
  4. "context"
  5. "time"
  6. "github.com/containerd/containerd"
  7. libcontainerdtypes "github.com/docker/docker/libcontainerd/types"
  8. specs "github.com/opencontainers/runtime-spec/specs-go"
  9. )
  10. type mockProcess struct {
  11. }
  12. func (m *mockProcess) Delete(_ context.Context) (uint32, time.Time, error) {
  13. return 0, time.Time{}, nil
  14. }
  15. // Mock containerd client implementation, for unit tests.
  16. type MockContainerdClient struct {
  17. }
  18. func (c *MockContainerdClient) Version(ctx context.Context) (containerd.Version, error) {
  19. return containerd.Version{}, nil
  20. }
  21. func (c *MockContainerdClient) Restore(ctx context.Context, containerID string, attachStdio libcontainerdtypes.StdioCallback) (alive bool, pid int, p libcontainerdtypes.Process, err error) {
  22. return false, 0, &mockProcess{}, nil
  23. }
  24. func (c *MockContainerdClient) Create(ctx context.Context, containerID string, spec *specs.Spec, runtimeOptions interface{}) error {
  25. return nil
  26. }
  27. func (c *MockContainerdClient) Start(ctx context.Context, containerID, checkpointDir string, withStdin bool, attachStdio libcontainerdtypes.StdioCallback) (pid int, err error) {
  28. return 0, nil
  29. }
  30. func (c *MockContainerdClient) SignalProcess(ctx context.Context, containerID, processID string, signal int) error {
  31. return nil
  32. }
  33. func (c *MockContainerdClient) Exec(ctx context.Context, containerID, processID string, spec *specs.Process, withStdin bool, attachStdio libcontainerdtypes.StdioCallback) (int, error) {
  34. return 0, nil
  35. }
  36. func (c *MockContainerdClient) ResizeTerminal(ctx context.Context, containerID, processID string, width, height int) error {
  37. return nil
  38. }
  39. func (c *MockContainerdClient) CloseStdin(ctx context.Context, containerID, processID string) error {
  40. return nil
  41. }
  42. func (c *MockContainerdClient) Pause(ctx context.Context, containerID string) error { return nil }
  43. func (c *MockContainerdClient) Resume(ctx context.Context, containerID string) error { return nil }
  44. func (c *MockContainerdClient) Stats(ctx context.Context, containerID string) (*libcontainerdtypes.Stats, error) {
  45. return nil, nil
  46. }
  47. func (c *MockContainerdClient) ListPids(ctx context.Context, containerID string) ([]uint32, error) {
  48. return nil, nil
  49. }
  50. func (c *MockContainerdClient) Summary(ctx context.Context, containerID string) ([]libcontainerdtypes.Summary, error) {
  51. return nil, nil
  52. }
  53. func (c *MockContainerdClient) DeleteTask(ctx context.Context, containerID string) (uint32, time.Time, error) {
  54. return 0, time.Time{}, nil
  55. }
  56. func (c *MockContainerdClient) Delete(ctx context.Context, containerID string) error { return nil }
  57. func (c *MockContainerdClient) Status(ctx context.Context, containerID string) (containerd.ProcessStatus, error) {
  58. return "null", nil
  59. }
  60. func (c *MockContainerdClient) UpdateResources(ctx context.Context, containerID string, resources *libcontainerdtypes.Resources) error {
  61. return nil
  62. }
  63. func (c *MockContainerdClient) CreateCheckpoint(ctx context.Context, containerID, checkpointDir string, exit bool) error {
  64. return nil
  65. }