truenas-go/cloudsync_service_iface.go at main · deevus/truenas-go · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
package truenas

import "context"

// CloudSyncServiceAPI defines the interface for cloud sync credential and task operations.
type CloudSyncServiceAPI interface {
	CreateCredential(ctx context.Context, opts CreateCredentialOpts) (*CloudSyncCredential, error)
	GetCredential(ctx context.Context, id int64) (*CloudSyncCredential, error)
	ListCredentials(ctx context.Context) ([]CloudSyncCredential, error)
	UpdateCredential(ctx context.Context, id int64, opts UpdateCredentialOpts) (*CloudSyncCredential, error)
	DeleteCredential(ctx context.Context, id int64) error
	CreateTask(ctx context.Context, opts CreateCloudSyncTaskOpts) (*CloudSyncTask, error)
	GetTask(ctx context.Context, id int64) (*CloudSyncTask, error)
	ListTasks(ctx context.Context) ([]CloudSyncTask, error)
	UpdateTask(ctx context.Context, id int64, opts UpdateCloudSyncTaskOpts) (*CloudSyncTask, error)
	DeleteTask(ctx context.Context, id int64) error
	Sync(ctx context.Context, id int64) error
}

// Compile-time checks.
var _ CloudSyncServiceAPI = (*CloudSyncService)(nil)
var _ CloudSyncServiceAPI = (*MockCloudSyncService)(nil)

// MockCloudSyncService is a test double for CloudSyncServiceAPI.
type MockCloudSyncService struct {
	CreateCredentialFunc func(ctx context.Context, opts CreateCredentialOpts) (*CloudSyncCredential, error)
	GetCredentialFunc    func(ctx context.Context, id int64) (*CloudSyncCredential, error)
	ListCredentialsFunc  func(ctx context.Context) ([]CloudSyncCredential, error)
	UpdateCredentialFunc func(ctx context.Context, id int64, opts UpdateCredentialOpts) (*CloudSyncCredential, error)
	DeleteCredentialFunc func(ctx context.Context, id int64) error
	CreateTaskFunc       func(ctx context.Context, opts CreateCloudSyncTaskOpts) (*CloudSyncTask, error)
	GetTaskFunc          func(ctx context.Context, id int64) (*CloudSyncTask, error)
	ListTasksFunc        func(ctx context.Context) ([]CloudSyncTask, error)
	UpdateTaskFunc       func(ctx context.Context, id int64, opts UpdateCloudSyncTaskOpts) (*CloudSyncTask, error)
	DeleteTaskFunc       func(ctx context.Context, id int64) error
	SyncFunc             func(ctx context.Context, id int64) error
}

func (m *MockCloudSyncService) CreateCredential(ctx context.Context, opts CreateCredentialOpts) (*CloudSyncCredential, error) {
	if m.CreateCredentialFunc != nil {
		return m.CreateCredentialFunc(ctx, opts)
	}
	return nil, nil
}

func (m *MockCloudSyncService) GetCredential(ctx context.Context, id int64) (*CloudSyncCredential, error) {
	if m.GetCredentialFunc != nil {
		return m.GetCredentialFunc(ctx, id)
	}
	return nil, nil
}

func (m *MockCloudSyncService) ListCredentials(ctx context.Context) ([]CloudSyncCredential, error) {
	if m.ListCredentialsFunc != nil {
		return m.ListCredentialsFunc(ctx)
	}
	return nil, nil
}

func (m *MockCloudSyncService) UpdateCredential(ctx context.Context, id int64, opts UpdateCredentialOpts) (*CloudSyncCredential, error) {
	if m.UpdateCredentialFunc != nil {
		return m.UpdateCredentialFunc(ctx, id, opts)
	}
	return nil, nil
}

func (m *MockCloudSyncService) DeleteCredential(ctx context.Context, id int64) error {
	if m.DeleteCredentialFunc != nil {
		return m.DeleteCredentialFunc(ctx, id)
	}
	return nil
}

func (m *MockCloudSyncService) CreateTask(ctx context.Context, opts CreateCloudSyncTaskOpts) (*CloudSyncTask, error) {
	if m.CreateTaskFunc != nil {
		return m.CreateTaskFunc(ctx, opts)
	}
	return nil, nil
}

func (m *MockCloudSyncService) GetTask(ctx context.Context, id int64) (*CloudSyncTask, error) {
	if m.GetTaskFunc != nil {
		return m.GetTaskFunc(ctx, id)
	}
	return nil, nil
}

func (m *MockCloudSyncService) ListTasks(ctx context.Context) ([]CloudSyncTask, error) {
	if m.ListTasksFunc != nil {
		return m.ListTasksFunc(ctx)
	}
	return nil, nil
}

func (m *MockCloudSyncService) UpdateTask(ctx context.Context, id int64, opts UpdateCloudSyncTaskOpts) (*CloudSyncTask, error) {
	if m.UpdateTaskFunc != nil {
		return m.UpdateTaskFunc(ctx, id, opts)
	}
	return nil, nil
}

func (m *MockCloudSyncService) DeleteTask(ctx context.Context, id int64) error {
	if m.DeleteTaskFunc != nil {
		return m.DeleteTaskFunc(ctx, id)
	}
	return nil
}

func (m *MockCloudSyncService) Sync(ctx context.Context, id int64) error {
	if m.SyncFunc != nil {
		return m.SyncFunc(ctx, id)
	}
	return nil
}