/
worker_meta_data_mock_test.go
56 lines (48 loc) · 1.73 KB
/
worker_meta_data_mock_test.go
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
package worker_meta_data
import (
"testing"
"github.com/cloudflare/cloudflare-go"
"github.com/cloudquery/cloudquery/plugins/source/cloudflare/client"
"github.com/cloudquery/cloudquery/plugins/source/cloudflare/client/mocks"
"github.com/cloudquery/plugin-sdk/faker"
"github.com/golang/mock/gomock"
)
func buildWorkerMetaData(t *testing.T, ctrl *gomock.Controller) client.Clients {
mock := mocks.NewMockApi(ctrl)
var workerScript cloudflare.WorkerMetaData
if err := faker.FakeObject(&workerScript); err != nil {
t.Fatal(err)
}
response := cloudflare.WorkerListResponse{
WorkerList: []cloudflare.WorkerMetaData{workerScript},
}
mock.EXPECT().ListWorkers(gomock.Any(), cloudflare.AccountIdentifier(client.TestAccountID), cloudflare.ListWorkersParams{}).Return(
response,
&response.ResultInfo,
nil,
)
var workerCronTrigger cloudflare.WorkerCronTrigger
if err := faker.FakeObject(&workerCronTrigger); err != nil {
t.Fatal(err)
}
mock.EXPECT().ListWorkerCronTriggers(gomock.Any(), cloudflare.AccountIdentifier(client.TestAccountID), cloudflare.ListWorkerCronTriggersParams{ScriptName: workerScript.ID}).Return(
[]cloudflare.WorkerCronTrigger{workerCronTrigger},
nil,
)
var workerSecret cloudflare.WorkersSecret
if err := faker.FakeObject(&workerSecret); err != nil {
t.Fatal(err)
}
mock.EXPECT().ListWorkersSecrets(gomock.Any(), cloudflare.AccountIdentifier(client.TestAccountID), cloudflare.ListWorkersSecretsParams{ScriptName: workerScript.ID}).Return(
cloudflare.WorkersListSecretsResponse{
Result: []cloudflare.WorkersSecret{workerSecret},
},
nil,
)
return client.Clients{
client.TestAccountID: mock,
}
}
func TestWorkerMetaData(t *testing.T) {
client.MockTestHelper(t, WorkerMetaData(), buildWorkerMetaData)
}