25개 이상의 토픽을 선택하실 수 없습니다. Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

mock_test.go 44 KiB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554
  1. // Copyright 2018 Google LLC
  2. //
  3. // Licensed under the Apache License, Version 2.0 (the "License");
  4. // you may not use this file except in compliance with the License.
  5. // You may obtain a copy of the License at
  6. //
  7. // https://www.apache.org/licenses/LICENSE-2.0
  8. //
  9. // Unless required by applicable law or agreed to in writing, software
  10. // distributed under the License is distributed on an "AS IS" BASIS,
  11. // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12. // See the License for the specific language governing permissions and
  13. // limitations under the License.
  14. // AUTO-GENERATED CODE. DO NOT EDIT.
  15. package cloudtasks
  16. import (
  17. durationpb "github.com/golang/protobuf/ptypes/duration"
  18. emptypb "github.com/golang/protobuf/ptypes/empty"
  19. timestamppb "github.com/golang/protobuf/ptypes/timestamp"
  20. taskspb "google.golang.org/genproto/googleapis/cloud/tasks/v2beta2"
  21. iampb "google.golang.org/genproto/googleapis/iam/v1"
  22. )
  23. import (
  24. "flag"
  25. "fmt"
  26. "io"
  27. "log"
  28. "net"
  29. "os"
  30. "strings"
  31. "testing"
  32. "github.com/golang/protobuf/proto"
  33. "github.com/golang/protobuf/ptypes"
  34. "golang.org/x/net/context"
  35. "google.golang.org/api/option"
  36. status "google.golang.org/genproto/googleapis/rpc/status"
  37. "google.golang.org/grpc"
  38. "google.golang.org/grpc/codes"
  39. "google.golang.org/grpc/metadata"
  40. gstatus "google.golang.org/grpc/status"
  41. )
  42. var _ = io.EOF
  43. var _ = ptypes.MarshalAny
  44. var _ status.Status
  45. type mockCloudTasksServer struct {
  46. // Embed for forward compatibility.
  47. // Tests will keep working if more methods are added
  48. // in the future.
  49. taskspb.CloudTasksServer
  50. reqs []proto.Message
  51. // If set, all calls return this error.
  52. err error
  53. // responses to return if err == nil
  54. resps []proto.Message
  55. }
  56. func (s *mockCloudTasksServer) ListQueues(ctx context.Context, req *taskspb.ListQueuesRequest) (*taskspb.ListQueuesResponse, error) {
  57. md, _ := metadata.FromIncomingContext(ctx)
  58. if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
  59. return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
  60. }
  61. s.reqs = append(s.reqs, req)
  62. if s.err != nil {
  63. return nil, s.err
  64. }
  65. return s.resps[0].(*taskspb.ListQueuesResponse), nil
  66. }
  67. func (s *mockCloudTasksServer) GetQueue(ctx context.Context, req *taskspb.GetQueueRequest) (*taskspb.Queue, error) {
  68. md, _ := metadata.FromIncomingContext(ctx)
  69. if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
  70. return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
  71. }
  72. s.reqs = append(s.reqs, req)
  73. if s.err != nil {
  74. return nil, s.err
  75. }
  76. return s.resps[0].(*taskspb.Queue), nil
  77. }
  78. func (s *mockCloudTasksServer) CreateQueue(ctx context.Context, req *taskspb.CreateQueueRequest) (*taskspb.Queue, error) {
  79. md, _ := metadata.FromIncomingContext(ctx)
  80. if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
  81. return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
  82. }
  83. s.reqs = append(s.reqs, req)
  84. if s.err != nil {
  85. return nil, s.err
  86. }
  87. return s.resps[0].(*taskspb.Queue), nil
  88. }
  89. func (s *mockCloudTasksServer) UpdateQueue(ctx context.Context, req *taskspb.UpdateQueueRequest) (*taskspb.Queue, error) {
  90. md, _ := metadata.FromIncomingContext(ctx)
  91. if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
  92. return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
  93. }
  94. s.reqs = append(s.reqs, req)
  95. if s.err != nil {
  96. return nil, s.err
  97. }
  98. return s.resps[0].(*taskspb.Queue), nil
  99. }
  100. func (s *mockCloudTasksServer) DeleteQueue(ctx context.Context, req *taskspb.DeleteQueueRequest) (*emptypb.Empty, error) {
  101. md, _ := metadata.FromIncomingContext(ctx)
  102. if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
  103. return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
  104. }
  105. s.reqs = append(s.reqs, req)
  106. if s.err != nil {
  107. return nil, s.err
  108. }
  109. return s.resps[0].(*emptypb.Empty), nil
  110. }
  111. func (s *mockCloudTasksServer) PurgeQueue(ctx context.Context, req *taskspb.PurgeQueueRequest) (*taskspb.Queue, error) {
  112. md, _ := metadata.FromIncomingContext(ctx)
  113. if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
  114. return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
  115. }
  116. s.reqs = append(s.reqs, req)
  117. if s.err != nil {
  118. return nil, s.err
  119. }
  120. return s.resps[0].(*taskspb.Queue), nil
  121. }
  122. func (s *mockCloudTasksServer) PauseQueue(ctx context.Context, req *taskspb.PauseQueueRequest) (*taskspb.Queue, error) {
  123. md, _ := metadata.FromIncomingContext(ctx)
  124. if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
  125. return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
  126. }
  127. s.reqs = append(s.reqs, req)
  128. if s.err != nil {
  129. return nil, s.err
  130. }
  131. return s.resps[0].(*taskspb.Queue), nil
  132. }
  133. func (s *mockCloudTasksServer) ResumeQueue(ctx context.Context, req *taskspb.ResumeQueueRequest) (*taskspb.Queue, error) {
  134. md, _ := metadata.FromIncomingContext(ctx)
  135. if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
  136. return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
  137. }
  138. s.reqs = append(s.reqs, req)
  139. if s.err != nil {
  140. return nil, s.err
  141. }
  142. return s.resps[0].(*taskspb.Queue), nil
  143. }
  144. func (s *mockCloudTasksServer) GetIamPolicy(ctx context.Context, req *iampb.GetIamPolicyRequest) (*iampb.Policy, error) {
  145. md, _ := metadata.FromIncomingContext(ctx)
  146. if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
  147. return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
  148. }
  149. s.reqs = append(s.reqs, req)
  150. if s.err != nil {
  151. return nil, s.err
  152. }
  153. return s.resps[0].(*iampb.Policy), nil
  154. }
  155. func (s *mockCloudTasksServer) SetIamPolicy(ctx context.Context, req *iampb.SetIamPolicyRequest) (*iampb.Policy, error) {
  156. md, _ := metadata.FromIncomingContext(ctx)
  157. if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
  158. return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
  159. }
  160. s.reqs = append(s.reqs, req)
  161. if s.err != nil {
  162. return nil, s.err
  163. }
  164. return s.resps[0].(*iampb.Policy), nil
  165. }
  166. func (s *mockCloudTasksServer) TestIamPermissions(ctx context.Context, req *iampb.TestIamPermissionsRequest) (*iampb.TestIamPermissionsResponse, error) {
  167. md, _ := metadata.FromIncomingContext(ctx)
  168. if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
  169. return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
  170. }
  171. s.reqs = append(s.reqs, req)
  172. if s.err != nil {
  173. return nil, s.err
  174. }
  175. return s.resps[0].(*iampb.TestIamPermissionsResponse), nil
  176. }
  177. func (s *mockCloudTasksServer) ListTasks(ctx context.Context, req *taskspb.ListTasksRequest) (*taskspb.ListTasksResponse, error) {
  178. md, _ := metadata.FromIncomingContext(ctx)
  179. if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
  180. return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
  181. }
  182. s.reqs = append(s.reqs, req)
  183. if s.err != nil {
  184. return nil, s.err
  185. }
  186. return s.resps[0].(*taskspb.ListTasksResponse), nil
  187. }
  188. func (s *mockCloudTasksServer) GetTask(ctx context.Context, req *taskspb.GetTaskRequest) (*taskspb.Task, error) {
  189. md, _ := metadata.FromIncomingContext(ctx)
  190. if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
  191. return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
  192. }
  193. s.reqs = append(s.reqs, req)
  194. if s.err != nil {
  195. return nil, s.err
  196. }
  197. return s.resps[0].(*taskspb.Task), nil
  198. }
  199. func (s *mockCloudTasksServer) CreateTask(ctx context.Context, req *taskspb.CreateTaskRequest) (*taskspb.Task, error) {
  200. md, _ := metadata.FromIncomingContext(ctx)
  201. if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
  202. return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
  203. }
  204. s.reqs = append(s.reqs, req)
  205. if s.err != nil {
  206. return nil, s.err
  207. }
  208. return s.resps[0].(*taskspb.Task), nil
  209. }
  210. func (s *mockCloudTasksServer) DeleteTask(ctx context.Context, req *taskspb.DeleteTaskRequest) (*emptypb.Empty, error) {
  211. md, _ := metadata.FromIncomingContext(ctx)
  212. if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
  213. return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
  214. }
  215. s.reqs = append(s.reqs, req)
  216. if s.err != nil {
  217. return nil, s.err
  218. }
  219. return s.resps[0].(*emptypb.Empty), nil
  220. }
  221. func (s *mockCloudTasksServer) LeaseTasks(ctx context.Context, req *taskspb.LeaseTasksRequest) (*taskspb.LeaseTasksResponse, error) {
  222. md, _ := metadata.FromIncomingContext(ctx)
  223. if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
  224. return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
  225. }
  226. s.reqs = append(s.reqs, req)
  227. if s.err != nil {
  228. return nil, s.err
  229. }
  230. return s.resps[0].(*taskspb.LeaseTasksResponse), nil
  231. }
  232. func (s *mockCloudTasksServer) AcknowledgeTask(ctx context.Context, req *taskspb.AcknowledgeTaskRequest) (*emptypb.Empty, error) {
  233. md, _ := metadata.FromIncomingContext(ctx)
  234. if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
  235. return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
  236. }
  237. s.reqs = append(s.reqs, req)
  238. if s.err != nil {
  239. return nil, s.err
  240. }
  241. return s.resps[0].(*emptypb.Empty), nil
  242. }
  243. func (s *mockCloudTasksServer) RenewLease(ctx context.Context, req *taskspb.RenewLeaseRequest) (*taskspb.Task, error) {
  244. md, _ := metadata.FromIncomingContext(ctx)
  245. if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
  246. return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
  247. }
  248. s.reqs = append(s.reqs, req)
  249. if s.err != nil {
  250. return nil, s.err
  251. }
  252. return s.resps[0].(*taskspb.Task), nil
  253. }
  254. func (s *mockCloudTasksServer) CancelLease(ctx context.Context, req *taskspb.CancelLeaseRequest) (*taskspb.Task, error) {
  255. md, _ := metadata.FromIncomingContext(ctx)
  256. if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
  257. return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
  258. }
  259. s.reqs = append(s.reqs, req)
  260. if s.err != nil {
  261. return nil, s.err
  262. }
  263. return s.resps[0].(*taskspb.Task), nil
  264. }
  265. func (s *mockCloudTasksServer) RunTask(ctx context.Context, req *taskspb.RunTaskRequest) (*taskspb.Task, error) {
  266. md, _ := metadata.FromIncomingContext(ctx)
  267. if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
  268. return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
  269. }
  270. s.reqs = append(s.reqs, req)
  271. if s.err != nil {
  272. return nil, s.err
  273. }
  274. return s.resps[0].(*taskspb.Task), nil
  275. }
  276. // clientOpt is the option tests should use to connect to the test server.
  277. // It is initialized by TestMain.
  278. var clientOpt option.ClientOption
  279. var (
  280. mockCloudTasks mockCloudTasksServer
  281. )
  282. func TestMain(m *testing.M) {
  283. flag.Parse()
  284. serv := grpc.NewServer()
  285. taskspb.RegisterCloudTasksServer(serv, &mockCloudTasks)
  286. lis, err := net.Listen("tcp", "localhost:0")
  287. if err != nil {
  288. log.Fatal(err)
  289. }
  290. go serv.Serve(lis)
  291. conn, err := grpc.Dial(lis.Addr().String(), grpc.WithInsecure())
  292. if err != nil {
  293. log.Fatal(err)
  294. }
  295. clientOpt = option.WithGRPCConn(conn)
  296. os.Exit(m.Run())
  297. }
  298. func TestCloudTasksListQueues(t *testing.T) {
  299. var nextPageToken string = ""
  300. var queuesElement *taskspb.Queue = &taskspb.Queue{}
  301. var queues = []*taskspb.Queue{queuesElement}
  302. var expectedResponse = &taskspb.ListQueuesResponse{
  303. NextPageToken: nextPageToken,
  304. Queues: queues,
  305. }
  306. mockCloudTasks.err = nil
  307. mockCloudTasks.reqs = nil
  308. mockCloudTasks.resps = append(mockCloudTasks.resps[:0], expectedResponse)
  309. var formattedParent string = fmt.Sprintf("projects/%s/locations/%s", "[PROJECT]", "[LOCATION]")
  310. var request = &taskspb.ListQueuesRequest{
  311. Parent: formattedParent,
  312. }
  313. c, err := NewClient(context.Background(), clientOpt)
  314. if err != nil {
  315. t.Fatal(err)
  316. }
  317. resp, err := c.ListQueues(context.Background(), request).Next()
  318. if err != nil {
  319. t.Fatal(err)
  320. }
  321. if want, got := request, mockCloudTasks.reqs[0]; !proto.Equal(want, got) {
  322. t.Errorf("wrong request %q, want %q", got, want)
  323. }
  324. want := (interface{})(expectedResponse.Queues[0])
  325. got := (interface{})(resp)
  326. var ok bool
  327. switch want := (want).(type) {
  328. case proto.Message:
  329. ok = proto.Equal(want, got.(proto.Message))
  330. default:
  331. ok = want == got
  332. }
  333. if !ok {
  334. t.Errorf("wrong response %q, want %q)", got, want)
  335. }
  336. }
  337. func TestCloudTasksListQueuesError(t *testing.T) {
  338. errCode := codes.PermissionDenied
  339. mockCloudTasks.err = gstatus.Error(errCode, "test error")
  340. var formattedParent string = fmt.Sprintf("projects/%s/locations/%s", "[PROJECT]", "[LOCATION]")
  341. var request = &taskspb.ListQueuesRequest{
  342. Parent: formattedParent,
  343. }
  344. c, err := NewClient(context.Background(), clientOpt)
  345. if err != nil {
  346. t.Fatal(err)
  347. }
  348. resp, err := c.ListQueues(context.Background(), request).Next()
  349. if st, ok := gstatus.FromError(err); !ok {
  350. t.Errorf("got error %v, expected grpc error", err)
  351. } else if c := st.Code(); c != errCode {
  352. t.Errorf("got error code %q, want %q", c, errCode)
  353. }
  354. _ = resp
  355. }
  356. func TestCloudTasksGetQueue(t *testing.T) {
  357. var name2 string = "name2-1052831874"
  358. var expectedResponse = &taskspb.Queue{
  359. Name: name2,
  360. }
  361. mockCloudTasks.err = nil
  362. mockCloudTasks.reqs = nil
  363. mockCloudTasks.resps = append(mockCloudTasks.resps[:0], expectedResponse)
  364. var formattedName string = fmt.Sprintf("projects/%s/locations/%s/queues/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]")
  365. var request = &taskspb.GetQueueRequest{
  366. Name: formattedName,
  367. }
  368. c, err := NewClient(context.Background(), clientOpt)
  369. if err != nil {
  370. t.Fatal(err)
  371. }
  372. resp, err := c.GetQueue(context.Background(), request)
  373. if err != nil {
  374. t.Fatal(err)
  375. }
  376. if want, got := request, mockCloudTasks.reqs[0]; !proto.Equal(want, got) {
  377. t.Errorf("wrong request %q, want %q", got, want)
  378. }
  379. if want, got := expectedResponse, resp; !proto.Equal(want, got) {
  380. t.Errorf("wrong response %q, want %q)", got, want)
  381. }
  382. }
  383. func TestCloudTasksGetQueueError(t *testing.T) {
  384. errCode := codes.PermissionDenied
  385. mockCloudTasks.err = gstatus.Error(errCode, "test error")
  386. var formattedName string = fmt.Sprintf("projects/%s/locations/%s/queues/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]")
  387. var request = &taskspb.GetQueueRequest{
  388. Name: formattedName,
  389. }
  390. c, err := NewClient(context.Background(), clientOpt)
  391. if err != nil {
  392. t.Fatal(err)
  393. }
  394. resp, err := c.GetQueue(context.Background(), request)
  395. if st, ok := gstatus.FromError(err); !ok {
  396. t.Errorf("got error %v, expected grpc error", err)
  397. } else if c := st.Code(); c != errCode {
  398. t.Errorf("got error code %q, want %q", c, errCode)
  399. }
  400. _ = resp
  401. }
  402. func TestCloudTasksCreateQueue(t *testing.T) {
  403. var name string = "name3373707"
  404. var expectedResponse = &taskspb.Queue{
  405. Name: name,
  406. }
  407. mockCloudTasks.err = nil
  408. mockCloudTasks.reqs = nil
  409. mockCloudTasks.resps = append(mockCloudTasks.resps[:0], expectedResponse)
  410. var formattedParent string = fmt.Sprintf("projects/%s/locations/%s", "[PROJECT]", "[LOCATION]")
  411. var queue *taskspb.Queue = &taskspb.Queue{}
  412. var request = &taskspb.CreateQueueRequest{
  413. Parent: formattedParent,
  414. Queue: queue,
  415. }
  416. c, err := NewClient(context.Background(), clientOpt)
  417. if err != nil {
  418. t.Fatal(err)
  419. }
  420. resp, err := c.CreateQueue(context.Background(), request)
  421. if err != nil {
  422. t.Fatal(err)
  423. }
  424. if want, got := request, mockCloudTasks.reqs[0]; !proto.Equal(want, got) {
  425. t.Errorf("wrong request %q, want %q", got, want)
  426. }
  427. if want, got := expectedResponse, resp; !proto.Equal(want, got) {
  428. t.Errorf("wrong response %q, want %q)", got, want)
  429. }
  430. }
  431. func TestCloudTasksCreateQueueError(t *testing.T) {
  432. errCode := codes.PermissionDenied
  433. mockCloudTasks.err = gstatus.Error(errCode, "test error")
  434. var formattedParent string = fmt.Sprintf("projects/%s/locations/%s", "[PROJECT]", "[LOCATION]")
  435. var queue *taskspb.Queue = &taskspb.Queue{}
  436. var request = &taskspb.CreateQueueRequest{
  437. Parent: formattedParent,
  438. Queue: queue,
  439. }
  440. c, err := NewClient(context.Background(), clientOpt)
  441. if err != nil {
  442. t.Fatal(err)
  443. }
  444. resp, err := c.CreateQueue(context.Background(), request)
  445. if st, ok := gstatus.FromError(err); !ok {
  446. t.Errorf("got error %v, expected grpc error", err)
  447. } else if c := st.Code(); c != errCode {
  448. t.Errorf("got error code %q, want %q", c, errCode)
  449. }
  450. _ = resp
  451. }
  452. func TestCloudTasksUpdateQueue(t *testing.T) {
  453. var name string = "name3373707"
  454. var expectedResponse = &taskspb.Queue{
  455. Name: name,
  456. }
  457. mockCloudTasks.err = nil
  458. mockCloudTasks.reqs = nil
  459. mockCloudTasks.resps = append(mockCloudTasks.resps[:0], expectedResponse)
  460. var queue *taskspb.Queue = &taskspb.Queue{}
  461. var request = &taskspb.UpdateQueueRequest{
  462. Queue: queue,
  463. }
  464. c, err := NewClient(context.Background(), clientOpt)
  465. if err != nil {
  466. t.Fatal(err)
  467. }
  468. resp, err := c.UpdateQueue(context.Background(), request)
  469. if err != nil {
  470. t.Fatal(err)
  471. }
  472. if want, got := request, mockCloudTasks.reqs[0]; !proto.Equal(want, got) {
  473. t.Errorf("wrong request %q, want %q", got, want)
  474. }
  475. if want, got := expectedResponse, resp; !proto.Equal(want, got) {
  476. t.Errorf("wrong response %q, want %q)", got, want)
  477. }
  478. }
  479. func TestCloudTasksUpdateQueueError(t *testing.T) {
  480. errCode := codes.PermissionDenied
  481. mockCloudTasks.err = gstatus.Error(errCode, "test error")
  482. var queue *taskspb.Queue = &taskspb.Queue{}
  483. var request = &taskspb.UpdateQueueRequest{
  484. Queue: queue,
  485. }
  486. c, err := NewClient(context.Background(), clientOpt)
  487. if err != nil {
  488. t.Fatal(err)
  489. }
  490. resp, err := c.UpdateQueue(context.Background(), request)
  491. if st, ok := gstatus.FromError(err); !ok {
  492. t.Errorf("got error %v, expected grpc error", err)
  493. } else if c := st.Code(); c != errCode {
  494. t.Errorf("got error code %q, want %q", c, errCode)
  495. }
  496. _ = resp
  497. }
  498. func TestCloudTasksDeleteQueue(t *testing.T) {
  499. var expectedResponse *emptypb.Empty = &emptypb.Empty{}
  500. mockCloudTasks.err = nil
  501. mockCloudTasks.reqs = nil
  502. mockCloudTasks.resps = append(mockCloudTasks.resps[:0], expectedResponse)
  503. var formattedName string = fmt.Sprintf("projects/%s/locations/%s/queues/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]")
  504. var request = &taskspb.DeleteQueueRequest{
  505. Name: formattedName,
  506. }
  507. c, err := NewClient(context.Background(), clientOpt)
  508. if err != nil {
  509. t.Fatal(err)
  510. }
  511. err = c.DeleteQueue(context.Background(), request)
  512. if err != nil {
  513. t.Fatal(err)
  514. }
  515. if want, got := request, mockCloudTasks.reqs[0]; !proto.Equal(want, got) {
  516. t.Errorf("wrong request %q, want %q", got, want)
  517. }
  518. }
  519. func TestCloudTasksDeleteQueueError(t *testing.T) {
  520. errCode := codes.PermissionDenied
  521. mockCloudTasks.err = gstatus.Error(errCode, "test error")
  522. var formattedName string = fmt.Sprintf("projects/%s/locations/%s/queues/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]")
  523. var request = &taskspb.DeleteQueueRequest{
  524. Name: formattedName,
  525. }
  526. c, err := NewClient(context.Background(), clientOpt)
  527. if err != nil {
  528. t.Fatal(err)
  529. }
  530. err = c.DeleteQueue(context.Background(), request)
  531. if st, ok := gstatus.FromError(err); !ok {
  532. t.Errorf("got error %v, expected grpc error", err)
  533. } else if c := st.Code(); c != errCode {
  534. t.Errorf("got error code %q, want %q", c, errCode)
  535. }
  536. }
  537. func TestCloudTasksPurgeQueue(t *testing.T) {
  538. var name2 string = "name2-1052831874"
  539. var expectedResponse = &taskspb.Queue{
  540. Name: name2,
  541. }
  542. mockCloudTasks.err = nil
  543. mockCloudTasks.reqs = nil
  544. mockCloudTasks.resps = append(mockCloudTasks.resps[:0], expectedResponse)
  545. var formattedName string = fmt.Sprintf("projects/%s/locations/%s/queues/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]")
  546. var request = &taskspb.PurgeQueueRequest{
  547. Name: formattedName,
  548. }
  549. c, err := NewClient(context.Background(), clientOpt)
  550. if err != nil {
  551. t.Fatal(err)
  552. }
  553. resp, err := c.PurgeQueue(context.Background(), request)
  554. if err != nil {
  555. t.Fatal(err)
  556. }
  557. if want, got := request, mockCloudTasks.reqs[0]; !proto.Equal(want, got) {
  558. t.Errorf("wrong request %q, want %q", got, want)
  559. }
  560. if want, got := expectedResponse, resp; !proto.Equal(want, got) {
  561. t.Errorf("wrong response %q, want %q)", got, want)
  562. }
  563. }
  564. func TestCloudTasksPurgeQueueError(t *testing.T) {
  565. errCode := codes.PermissionDenied
  566. mockCloudTasks.err = gstatus.Error(errCode, "test error")
  567. var formattedName string = fmt.Sprintf("projects/%s/locations/%s/queues/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]")
  568. var request = &taskspb.PurgeQueueRequest{
  569. Name: formattedName,
  570. }
  571. c, err := NewClient(context.Background(), clientOpt)
  572. if err != nil {
  573. t.Fatal(err)
  574. }
  575. resp, err := c.PurgeQueue(context.Background(), request)
  576. if st, ok := gstatus.FromError(err); !ok {
  577. t.Errorf("got error %v, expected grpc error", err)
  578. } else if c := st.Code(); c != errCode {
  579. t.Errorf("got error code %q, want %q", c, errCode)
  580. }
  581. _ = resp
  582. }
  583. func TestCloudTasksPauseQueue(t *testing.T) {
  584. var name2 string = "name2-1052831874"
  585. var expectedResponse = &taskspb.Queue{
  586. Name: name2,
  587. }
  588. mockCloudTasks.err = nil
  589. mockCloudTasks.reqs = nil
  590. mockCloudTasks.resps = append(mockCloudTasks.resps[:0], expectedResponse)
  591. var formattedName string = fmt.Sprintf("projects/%s/locations/%s/queues/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]")
  592. var request = &taskspb.PauseQueueRequest{
  593. Name: formattedName,
  594. }
  595. c, err := NewClient(context.Background(), clientOpt)
  596. if err != nil {
  597. t.Fatal(err)
  598. }
  599. resp, err := c.PauseQueue(context.Background(), request)
  600. if err != nil {
  601. t.Fatal(err)
  602. }
  603. if want, got := request, mockCloudTasks.reqs[0]; !proto.Equal(want, got) {
  604. t.Errorf("wrong request %q, want %q", got, want)
  605. }
  606. if want, got := expectedResponse, resp; !proto.Equal(want, got) {
  607. t.Errorf("wrong response %q, want %q)", got, want)
  608. }
  609. }
  610. func TestCloudTasksPauseQueueError(t *testing.T) {
  611. errCode := codes.PermissionDenied
  612. mockCloudTasks.err = gstatus.Error(errCode, "test error")
  613. var formattedName string = fmt.Sprintf("projects/%s/locations/%s/queues/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]")
  614. var request = &taskspb.PauseQueueRequest{
  615. Name: formattedName,
  616. }
  617. c, err := NewClient(context.Background(), clientOpt)
  618. if err != nil {
  619. t.Fatal(err)
  620. }
  621. resp, err := c.PauseQueue(context.Background(), request)
  622. if st, ok := gstatus.FromError(err); !ok {
  623. t.Errorf("got error %v, expected grpc error", err)
  624. } else if c := st.Code(); c != errCode {
  625. t.Errorf("got error code %q, want %q", c, errCode)
  626. }
  627. _ = resp
  628. }
  629. func TestCloudTasksResumeQueue(t *testing.T) {
  630. var name2 string = "name2-1052831874"
  631. var expectedResponse = &taskspb.Queue{
  632. Name: name2,
  633. }
  634. mockCloudTasks.err = nil
  635. mockCloudTasks.reqs = nil
  636. mockCloudTasks.resps = append(mockCloudTasks.resps[:0], expectedResponse)
  637. var formattedName string = fmt.Sprintf("projects/%s/locations/%s/queues/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]")
  638. var request = &taskspb.ResumeQueueRequest{
  639. Name: formattedName,
  640. }
  641. c, err := NewClient(context.Background(), clientOpt)
  642. if err != nil {
  643. t.Fatal(err)
  644. }
  645. resp, err := c.ResumeQueue(context.Background(), request)
  646. if err != nil {
  647. t.Fatal(err)
  648. }
  649. if want, got := request, mockCloudTasks.reqs[0]; !proto.Equal(want, got) {
  650. t.Errorf("wrong request %q, want %q", got, want)
  651. }
  652. if want, got := expectedResponse, resp; !proto.Equal(want, got) {
  653. t.Errorf("wrong response %q, want %q)", got, want)
  654. }
  655. }
  656. func TestCloudTasksResumeQueueError(t *testing.T) {
  657. errCode := codes.PermissionDenied
  658. mockCloudTasks.err = gstatus.Error(errCode, "test error")
  659. var formattedName string = fmt.Sprintf("projects/%s/locations/%s/queues/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]")
  660. var request = &taskspb.ResumeQueueRequest{
  661. Name: formattedName,
  662. }
  663. c, err := NewClient(context.Background(), clientOpt)
  664. if err != nil {
  665. t.Fatal(err)
  666. }
  667. resp, err := c.ResumeQueue(context.Background(), request)
  668. if st, ok := gstatus.FromError(err); !ok {
  669. t.Errorf("got error %v, expected grpc error", err)
  670. } else if c := st.Code(); c != errCode {
  671. t.Errorf("got error code %q, want %q", c, errCode)
  672. }
  673. _ = resp
  674. }
  675. func TestCloudTasksGetIamPolicy(t *testing.T) {
  676. var version int32 = 351608024
  677. var etag []byte = []byte("21")
  678. var expectedResponse = &iampb.Policy{
  679. Version: version,
  680. Etag: etag,
  681. }
  682. mockCloudTasks.err = nil
  683. mockCloudTasks.reqs = nil
  684. mockCloudTasks.resps = append(mockCloudTasks.resps[:0], expectedResponse)
  685. var formattedResource string = fmt.Sprintf("projects/%s/locations/%s/queues/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]")
  686. var request = &iampb.GetIamPolicyRequest{
  687. Resource: formattedResource,
  688. }
  689. c, err := NewClient(context.Background(), clientOpt)
  690. if err != nil {
  691. t.Fatal(err)
  692. }
  693. resp, err := c.GetIamPolicy(context.Background(), request)
  694. if err != nil {
  695. t.Fatal(err)
  696. }
  697. if want, got := request, mockCloudTasks.reqs[0]; !proto.Equal(want, got) {
  698. t.Errorf("wrong request %q, want %q", got, want)
  699. }
  700. if want, got := expectedResponse, resp; !proto.Equal(want, got) {
  701. t.Errorf("wrong response %q, want %q)", got, want)
  702. }
  703. }
  704. func TestCloudTasksGetIamPolicyError(t *testing.T) {
  705. errCode := codes.PermissionDenied
  706. mockCloudTasks.err = gstatus.Error(errCode, "test error")
  707. var formattedResource string = fmt.Sprintf("projects/%s/locations/%s/queues/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]")
  708. var request = &iampb.GetIamPolicyRequest{
  709. Resource: formattedResource,
  710. }
  711. c, err := NewClient(context.Background(), clientOpt)
  712. if err != nil {
  713. t.Fatal(err)
  714. }
  715. resp, err := c.GetIamPolicy(context.Background(), request)
  716. if st, ok := gstatus.FromError(err); !ok {
  717. t.Errorf("got error %v, expected grpc error", err)
  718. } else if c := st.Code(); c != errCode {
  719. t.Errorf("got error code %q, want %q", c, errCode)
  720. }
  721. _ = resp
  722. }
  723. func TestCloudTasksSetIamPolicy(t *testing.T) {
  724. var version int32 = 351608024
  725. var etag []byte = []byte("21")
  726. var expectedResponse = &iampb.Policy{
  727. Version: version,
  728. Etag: etag,
  729. }
  730. mockCloudTasks.err = nil
  731. mockCloudTasks.reqs = nil
  732. mockCloudTasks.resps = append(mockCloudTasks.resps[:0], expectedResponse)
  733. var formattedResource string = fmt.Sprintf("projects/%s/locations/%s/queues/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]")
  734. var policy *iampb.Policy = &iampb.Policy{}
  735. var request = &iampb.SetIamPolicyRequest{
  736. Resource: formattedResource,
  737. Policy: policy,
  738. }
  739. c, err := NewClient(context.Background(), clientOpt)
  740. if err != nil {
  741. t.Fatal(err)
  742. }
  743. resp, err := c.SetIamPolicy(context.Background(), request)
  744. if err != nil {
  745. t.Fatal(err)
  746. }
  747. if want, got := request, mockCloudTasks.reqs[0]; !proto.Equal(want, got) {
  748. t.Errorf("wrong request %q, want %q", got, want)
  749. }
  750. if want, got := expectedResponse, resp; !proto.Equal(want, got) {
  751. t.Errorf("wrong response %q, want %q)", got, want)
  752. }
  753. }
  754. func TestCloudTasksSetIamPolicyError(t *testing.T) {
  755. errCode := codes.PermissionDenied
  756. mockCloudTasks.err = gstatus.Error(errCode, "test error")
  757. var formattedResource string = fmt.Sprintf("projects/%s/locations/%s/queues/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]")
  758. var policy *iampb.Policy = &iampb.Policy{}
  759. var request = &iampb.SetIamPolicyRequest{
  760. Resource: formattedResource,
  761. Policy: policy,
  762. }
  763. c, err := NewClient(context.Background(), clientOpt)
  764. if err != nil {
  765. t.Fatal(err)
  766. }
  767. resp, err := c.SetIamPolicy(context.Background(), request)
  768. if st, ok := gstatus.FromError(err); !ok {
  769. t.Errorf("got error %v, expected grpc error", err)
  770. } else if c := st.Code(); c != errCode {
  771. t.Errorf("got error code %q, want %q", c, errCode)
  772. }
  773. _ = resp
  774. }
  775. func TestCloudTasksTestIamPermissions(t *testing.T) {
  776. var expectedResponse *iampb.TestIamPermissionsResponse = &iampb.TestIamPermissionsResponse{}
  777. mockCloudTasks.err = nil
  778. mockCloudTasks.reqs = nil
  779. mockCloudTasks.resps = append(mockCloudTasks.resps[:0], expectedResponse)
  780. var formattedResource string = fmt.Sprintf("projects/%s/locations/%s/queues/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]")
  781. var permissions []string = nil
  782. var request = &iampb.TestIamPermissionsRequest{
  783. Resource: formattedResource,
  784. Permissions: permissions,
  785. }
  786. c, err := NewClient(context.Background(), clientOpt)
  787. if err != nil {
  788. t.Fatal(err)
  789. }
  790. resp, err := c.TestIamPermissions(context.Background(), request)
  791. if err != nil {
  792. t.Fatal(err)
  793. }
  794. if want, got := request, mockCloudTasks.reqs[0]; !proto.Equal(want, got) {
  795. t.Errorf("wrong request %q, want %q", got, want)
  796. }
  797. if want, got := expectedResponse, resp; !proto.Equal(want, got) {
  798. t.Errorf("wrong response %q, want %q)", got, want)
  799. }
  800. }
  801. func TestCloudTasksTestIamPermissionsError(t *testing.T) {
  802. errCode := codes.PermissionDenied
  803. mockCloudTasks.err = gstatus.Error(errCode, "test error")
  804. var formattedResource string = fmt.Sprintf("projects/%s/locations/%s/queues/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]")
  805. var permissions []string = nil
  806. var request = &iampb.TestIamPermissionsRequest{
  807. Resource: formattedResource,
  808. Permissions: permissions,
  809. }
  810. c, err := NewClient(context.Background(), clientOpt)
  811. if err != nil {
  812. t.Fatal(err)
  813. }
  814. resp, err := c.TestIamPermissions(context.Background(), request)
  815. if st, ok := gstatus.FromError(err); !ok {
  816. t.Errorf("got error %v, expected grpc error", err)
  817. } else if c := st.Code(); c != errCode {
  818. t.Errorf("got error code %q, want %q", c, errCode)
  819. }
  820. _ = resp
  821. }
  822. func TestCloudTasksListTasks(t *testing.T) {
  823. var nextPageToken string = ""
  824. var tasksElement *taskspb.Task = &taskspb.Task{}
  825. var tasks = []*taskspb.Task{tasksElement}
  826. var expectedResponse = &taskspb.ListTasksResponse{
  827. NextPageToken: nextPageToken,
  828. Tasks: tasks,
  829. }
  830. mockCloudTasks.err = nil
  831. mockCloudTasks.reqs = nil
  832. mockCloudTasks.resps = append(mockCloudTasks.resps[:0], expectedResponse)
  833. var formattedParent string = fmt.Sprintf("projects/%s/locations/%s/queues/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]")
  834. var request = &taskspb.ListTasksRequest{
  835. Parent: formattedParent,
  836. }
  837. c, err := NewClient(context.Background(), clientOpt)
  838. if err != nil {
  839. t.Fatal(err)
  840. }
  841. resp, err := c.ListTasks(context.Background(), request).Next()
  842. if err != nil {
  843. t.Fatal(err)
  844. }
  845. if want, got := request, mockCloudTasks.reqs[0]; !proto.Equal(want, got) {
  846. t.Errorf("wrong request %q, want %q", got, want)
  847. }
  848. want := (interface{})(expectedResponse.Tasks[0])
  849. got := (interface{})(resp)
  850. var ok bool
  851. switch want := (want).(type) {
  852. case proto.Message:
  853. ok = proto.Equal(want, got.(proto.Message))
  854. default:
  855. ok = want == got
  856. }
  857. if !ok {
  858. t.Errorf("wrong response %q, want %q)", got, want)
  859. }
  860. }
  861. func TestCloudTasksListTasksError(t *testing.T) {
  862. errCode := codes.PermissionDenied
  863. mockCloudTasks.err = gstatus.Error(errCode, "test error")
  864. var formattedParent string = fmt.Sprintf("projects/%s/locations/%s/queues/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]")
  865. var request = &taskspb.ListTasksRequest{
  866. Parent: formattedParent,
  867. }
  868. c, err := NewClient(context.Background(), clientOpt)
  869. if err != nil {
  870. t.Fatal(err)
  871. }
  872. resp, err := c.ListTasks(context.Background(), request).Next()
  873. if st, ok := gstatus.FromError(err); !ok {
  874. t.Errorf("got error %v, expected grpc error", err)
  875. } else if c := st.Code(); c != errCode {
  876. t.Errorf("got error code %q, want %q", c, errCode)
  877. }
  878. _ = resp
  879. }
  880. func TestCloudTasksGetTask(t *testing.T) {
  881. var name2 string = "name2-1052831874"
  882. var expectedResponse = &taskspb.Task{
  883. Name: name2,
  884. }
  885. mockCloudTasks.err = nil
  886. mockCloudTasks.reqs = nil
  887. mockCloudTasks.resps = append(mockCloudTasks.resps[:0], expectedResponse)
  888. var formattedName string = fmt.Sprintf("projects/%s/locations/%s/queues/%s/tasks/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]", "[TASK]")
  889. var request = &taskspb.GetTaskRequest{
  890. Name: formattedName,
  891. }
  892. c, err := NewClient(context.Background(), clientOpt)
  893. if err != nil {
  894. t.Fatal(err)
  895. }
  896. resp, err := c.GetTask(context.Background(), request)
  897. if err != nil {
  898. t.Fatal(err)
  899. }
  900. if want, got := request, mockCloudTasks.reqs[0]; !proto.Equal(want, got) {
  901. t.Errorf("wrong request %q, want %q", got, want)
  902. }
  903. if want, got := expectedResponse, resp; !proto.Equal(want, got) {
  904. t.Errorf("wrong response %q, want %q)", got, want)
  905. }
  906. }
  907. func TestCloudTasksGetTaskError(t *testing.T) {
  908. errCode := codes.PermissionDenied
  909. mockCloudTasks.err = gstatus.Error(errCode, "test error")
  910. var formattedName string = fmt.Sprintf("projects/%s/locations/%s/queues/%s/tasks/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]", "[TASK]")
  911. var request = &taskspb.GetTaskRequest{
  912. Name: formattedName,
  913. }
  914. c, err := NewClient(context.Background(), clientOpt)
  915. if err != nil {
  916. t.Fatal(err)
  917. }
  918. resp, err := c.GetTask(context.Background(), request)
  919. if st, ok := gstatus.FromError(err); !ok {
  920. t.Errorf("got error %v, expected grpc error", err)
  921. } else if c := st.Code(); c != errCode {
  922. t.Errorf("got error code %q, want %q", c, errCode)
  923. }
  924. _ = resp
  925. }
  926. func TestCloudTasksCreateTask(t *testing.T) {
  927. var name string = "name3373707"
  928. var expectedResponse = &taskspb.Task{
  929. Name: name,
  930. }
  931. mockCloudTasks.err = nil
  932. mockCloudTasks.reqs = nil
  933. mockCloudTasks.resps = append(mockCloudTasks.resps[:0], expectedResponse)
  934. var formattedParent string = fmt.Sprintf("projects/%s/locations/%s/queues/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]")
  935. var task *taskspb.Task = &taskspb.Task{}
  936. var request = &taskspb.CreateTaskRequest{
  937. Parent: formattedParent,
  938. Task: task,
  939. }
  940. c, err := NewClient(context.Background(), clientOpt)
  941. if err != nil {
  942. t.Fatal(err)
  943. }
  944. resp, err := c.CreateTask(context.Background(), request)
  945. if err != nil {
  946. t.Fatal(err)
  947. }
  948. if want, got := request, mockCloudTasks.reqs[0]; !proto.Equal(want, got) {
  949. t.Errorf("wrong request %q, want %q", got, want)
  950. }
  951. if want, got := expectedResponse, resp; !proto.Equal(want, got) {
  952. t.Errorf("wrong response %q, want %q)", got, want)
  953. }
  954. }
  955. func TestCloudTasksCreateTaskError(t *testing.T) {
  956. errCode := codes.PermissionDenied
  957. mockCloudTasks.err = gstatus.Error(errCode, "test error")
  958. var formattedParent string = fmt.Sprintf("projects/%s/locations/%s/queues/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]")
  959. var task *taskspb.Task = &taskspb.Task{}
  960. var request = &taskspb.CreateTaskRequest{
  961. Parent: formattedParent,
  962. Task: task,
  963. }
  964. c, err := NewClient(context.Background(), clientOpt)
  965. if err != nil {
  966. t.Fatal(err)
  967. }
  968. resp, err := c.CreateTask(context.Background(), request)
  969. if st, ok := gstatus.FromError(err); !ok {
  970. t.Errorf("got error %v, expected grpc error", err)
  971. } else if c := st.Code(); c != errCode {
  972. t.Errorf("got error code %q, want %q", c, errCode)
  973. }
  974. _ = resp
  975. }
  976. func TestCloudTasksDeleteTask(t *testing.T) {
  977. var expectedResponse *emptypb.Empty = &emptypb.Empty{}
  978. mockCloudTasks.err = nil
  979. mockCloudTasks.reqs = nil
  980. mockCloudTasks.resps = append(mockCloudTasks.resps[:0], expectedResponse)
  981. var formattedName string = fmt.Sprintf("projects/%s/locations/%s/queues/%s/tasks/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]", "[TASK]")
  982. var request = &taskspb.DeleteTaskRequest{
  983. Name: formattedName,
  984. }
  985. c, err := NewClient(context.Background(), clientOpt)
  986. if err != nil {
  987. t.Fatal(err)
  988. }
  989. err = c.DeleteTask(context.Background(), request)
  990. if err != nil {
  991. t.Fatal(err)
  992. }
  993. if want, got := request, mockCloudTasks.reqs[0]; !proto.Equal(want, got) {
  994. t.Errorf("wrong request %q, want %q", got, want)
  995. }
  996. }
  997. func TestCloudTasksDeleteTaskError(t *testing.T) {
  998. errCode := codes.PermissionDenied
  999. mockCloudTasks.err = gstatus.Error(errCode, "test error")
  1000. var formattedName string = fmt.Sprintf("projects/%s/locations/%s/queues/%s/tasks/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]", "[TASK]")
  1001. var request = &taskspb.DeleteTaskRequest{
  1002. Name: formattedName,
  1003. }
  1004. c, err := NewClient(context.Background(), clientOpt)
  1005. if err != nil {
  1006. t.Fatal(err)
  1007. }
  1008. err = c.DeleteTask(context.Background(), request)
  1009. if st, ok := gstatus.FromError(err); !ok {
  1010. t.Errorf("got error %v, expected grpc error", err)
  1011. } else if c := st.Code(); c != errCode {
  1012. t.Errorf("got error code %q, want %q", c, errCode)
  1013. }
  1014. }
  1015. func TestCloudTasksLeaseTasks(t *testing.T) {
  1016. var expectedResponse *taskspb.LeaseTasksResponse = &taskspb.LeaseTasksResponse{}
  1017. mockCloudTasks.err = nil
  1018. mockCloudTasks.reqs = nil
  1019. mockCloudTasks.resps = append(mockCloudTasks.resps[:0], expectedResponse)
  1020. var formattedParent string = fmt.Sprintf("projects/%s/locations/%s/queues/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]")
  1021. var leaseDuration *durationpb.Duration = &durationpb.Duration{}
  1022. var request = &taskspb.LeaseTasksRequest{
  1023. Parent: formattedParent,
  1024. LeaseDuration: leaseDuration,
  1025. }
  1026. c, err := NewClient(context.Background(), clientOpt)
  1027. if err != nil {
  1028. t.Fatal(err)
  1029. }
  1030. resp, err := c.LeaseTasks(context.Background(), request)
  1031. if err != nil {
  1032. t.Fatal(err)
  1033. }
  1034. if want, got := request, mockCloudTasks.reqs[0]; !proto.Equal(want, got) {
  1035. t.Errorf("wrong request %q, want %q", got, want)
  1036. }
  1037. if want, got := expectedResponse, resp; !proto.Equal(want, got) {
  1038. t.Errorf("wrong response %q, want %q)", got, want)
  1039. }
  1040. }
  1041. func TestCloudTasksLeaseTasksError(t *testing.T) {
  1042. errCode := codes.PermissionDenied
  1043. mockCloudTasks.err = gstatus.Error(errCode, "test error")
  1044. var formattedParent string = fmt.Sprintf("projects/%s/locations/%s/queues/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]")
  1045. var leaseDuration *durationpb.Duration = &durationpb.Duration{}
  1046. var request = &taskspb.LeaseTasksRequest{
  1047. Parent: formattedParent,
  1048. LeaseDuration: leaseDuration,
  1049. }
  1050. c, err := NewClient(context.Background(), clientOpt)
  1051. if err != nil {
  1052. t.Fatal(err)
  1053. }
  1054. resp, err := c.LeaseTasks(context.Background(), request)
  1055. if st, ok := gstatus.FromError(err); !ok {
  1056. t.Errorf("got error %v, expected grpc error", err)
  1057. } else if c := st.Code(); c != errCode {
  1058. t.Errorf("got error code %q, want %q", c, errCode)
  1059. }
  1060. _ = resp
  1061. }
  1062. func TestCloudTasksAcknowledgeTask(t *testing.T) {
  1063. var expectedResponse *emptypb.Empty = &emptypb.Empty{}
  1064. mockCloudTasks.err = nil
  1065. mockCloudTasks.reqs = nil
  1066. mockCloudTasks.resps = append(mockCloudTasks.resps[:0], expectedResponse)
  1067. var formattedName string = fmt.Sprintf("projects/%s/locations/%s/queues/%s/tasks/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]", "[TASK]")
  1068. var scheduleTime *timestamppb.Timestamp = &timestamppb.Timestamp{}
  1069. var request = &taskspb.AcknowledgeTaskRequest{
  1070. Name: formattedName,
  1071. ScheduleTime: scheduleTime,
  1072. }
  1073. c, err := NewClient(context.Background(), clientOpt)
  1074. if err != nil {
  1075. t.Fatal(err)
  1076. }
  1077. err = c.AcknowledgeTask(context.Background(), request)
  1078. if err != nil {
  1079. t.Fatal(err)
  1080. }
  1081. if want, got := request, mockCloudTasks.reqs[0]; !proto.Equal(want, got) {
  1082. t.Errorf("wrong request %q, want %q", got, want)
  1083. }
  1084. }
  1085. func TestCloudTasksAcknowledgeTaskError(t *testing.T) {
  1086. errCode := codes.PermissionDenied
  1087. mockCloudTasks.err = gstatus.Error(errCode, "test error")
  1088. var formattedName string = fmt.Sprintf("projects/%s/locations/%s/queues/%s/tasks/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]", "[TASK]")
  1089. var scheduleTime *timestamppb.Timestamp = &timestamppb.Timestamp{}
  1090. var request = &taskspb.AcknowledgeTaskRequest{
  1091. Name: formattedName,
  1092. ScheduleTime: scheduleTime,
  1093. }
  1094. c, err := NewClient(context.Background(), clientOpt)
  1095. if err != nil {
  1096. t.Fatal(err)
  1097. }
  1098. err = c.AcknowledgeTask(context.Background(), request)
  1099. if st, ok := gstatus.FromError(err); !ok {
  1100. t.Errorf("got error %v, expected grpc error", err)
  1101. } else if c := st.Code(); c != errCode {
  1102. t.Errorf("got error code %q, want %q", c, errCode)
  1103. }
  1104. }
  1105. func TestCloudTasksRenewLease(t *testing.T) {
  1106. var name2 string = "name2-1052831874"
  1107. var expectedResponse = &taskspb.Task{
  1108. Name: name2,
  1109. }
  1110. mockCloudTasks.err = nil
  1111. mockCloudTasks.reqs = nil
  1112. mockCloudTasks.resps = append(mockCloudTasks.resps[:0], expectedResponse)
  1113. var formattedName string = fmt.Sprintf("projects/%s/locations/%s/queues/%s/tasks/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]", "[TASK]")
  1114. var scheduleTime *timestamppb.Timestamp = &timestamppb.Timestamp{}
  1115. var leaseDuration *durationpb.Duration = &durationpb.Duration{}
  1116. var request = &taskspb.RenewLeaseRequest{
  1117. Name: formattedName,
  1118. ScheduleTime: scheduleTime,
  1119. LeaseDuration: leaseDuration,
  1120. }
  1121. c, err := NewClient(context.Background(), clientOpt)
  1122. if err != nil {
  1123. t.Fatal(err)
  1124. }
  1125. resp, err := c.RenewLease(context.Background(), request)
  1126. if err != nil {
  1127. t.Fatal(err)
  1128. }
  1129. if want, got := request, mockCloudTasks.reqs[0]; !proto.Equal(want, got) {
  1130. t.Errorf("wrong request %q, want %q", got, want)
  1131. }
  1132. if want, got := expectedResponse, resp; !proto.Equal(want, got) {
  1133. t.Errorf("wrong response %q, want %q)", got, want)
  1134. }
  1135. }
  1136. func TestCloudTasksRenewLeaseError(t *testing.T) {
  1137. errCode := codes.PermissionDenied
  1138. mockCloudTasks.err = gstatus.Error(errCode, "test error")
  1139. var formattedName string = fmt.Sprintf("projects/%s/locations/%s/queues/%s/tasks/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]", "[TASK]")
  1140. var scheduleTime *timestamppb.Timestamp = &timestamppb.Timestamp{}
  1141. var leaseDuration *durationpb.Duration = &durationpb.Duration{}
  1142. var request = &taskspb.RenewLeaseRequest{
  1143. Name: formattedName,
  1144. ScheduleTime: scheduleTime,
  1145. LeaseDuration: leaseDuration,
  1146. }
  1147. c, err := NewClient(context.Background(), clientOpt)
  1148. if err != nil {
  1149. t.Fatal(err)
  1150. }
  1151. resp, err := c.RenewLease(context.Background(), request)
  1152. if st, ok := gstatus.FromError(err); !ok {
  1153. t.Errorf("got error %v, expected grpc error", err)
  1154. } else if c := st.Code(); c != errCode {
  1155. t.Errorf("got error code %q, want %q", c, errCode)
  1156. }
  1157. _ = resp
  1158. }
  1159. func TestCloudTasksCancelLease(t *testing.T) {
  1160. var name2 string = "name2-1052831874"
  1161. var expectedResponse = &taskspb.Task{
  1162. Name: name2,
  1163. }
  1164. mockCloudTasks.err = nil
  1165. mockCloudTasks.reqs = nil
  1166. mockCloudTasks.resps = append(mockCloudTasks.resps[:0], expectedResponse)
  1167. var formattedName string = fmt.Sprintf("projects/%s/locations/%s/queues/%s/tasks/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]", "[TASK]")
  1168. var scheduleTime *timestamppb.Timestamp = &timestamppb.Timestamp{}
  1169. var request = &taskspb.CancelLeaseRequest{
  1170. Name: formattedName,
  1171. ScheduleTime: scheduleTime,
  1172. }
  1173. c, err := NewClient(context.Background(), clientOpt)
  1174. if err != nil {
  1175. t.Fatal(err)
  1176. }
  1177. resp, err := c.CancelLease(context.Background(), request)
  1178. if err != nil {
  1179. t.Fatal(err)
  1180. }
  1181. if want, got := request, mockCloudTasks.reqs[0]; !proto.Equal(want, got) {
  1182. t.Errorf("wrong request %q, want %q", got, want)
  1183. }
  1184. if want, got := expectedResponse, resp; !proto.Equal(want, got) {
  1185. t.Errorf("wrong response %q, want %q)", got, want)
  1186. }
  1187. }
  1188. func TestCloudTasksCancelLeaseError(t *testing.T) {
  1189. errCode := codes.PermissionDenied
  1190. mockCloudTasks.err = gstatus.Error(errCode, "test error")
  1191. var formattedName string = fmt.Sprintf("projects/%s/locations/%s/queues/%s/tasks/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]", "[TASK]")
  1192. var scheduleTime *timestamppb.Timestamp = &timestamppb.Timestamp{}
  1193. var request = &taskspb.CancelLeaseRequest{
  1194. Name: formattedName,
  1195. ScheduleTime: scheduleTime,
  1196. }
  1197. c, err := NewClient(context.Background(), clientOpt)
  1198. if err != nil {
  1199. t.Fatal(err)
  1200. }
  1201. resp, err := c.CancelLease(context.Background(), request)
  1202. if st, ok := gstatus.FromError(err); !ok {
  1203. t.Errorf("got error %v, expected grpc error", err)
  1204. } else if c := st.Code(); c != errCode {
  1205. t.Errorf("got error code %q, want %q", c, errCode)
  1206. }
  1207. _ = resp
  1208. }
  1209. func TestCloudTasksRunTask(t *testing.T) {
  1210. var name2 string = "name2-1052831874"
  1211. var expectedResponse = &taskspb.Task{
  1212. Name: name2,
  1213. }
  1214. mockCloudTasks.err = nil
  1215. mockCloudTasks.reqs = nil
  1216. mockCloudTasks.resps = append(mockCloudTasks.resps[:0], expectedResponse)
  1217. var formattedName string = fmt.Sprintf("projects/%s/locations/%s/queues/%s/tasks/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]", "[TASK]")
  1218. var request = &taskspb.RunTaskRequest{
  1219. Name: formattedName,
  1220. }
  1221. c, err := NewClient(context.Background(), clientOpt)
  1222. if err != nil {
  1223. t.Fatal(err)
  1224. }
  1225. resp, err := c.RunTask(context.Background(), request)
  1226. if err != nil {
  1227. t.Fatal(err)
  1228. }
  1229. if want, got := request, mockCloudTasks.reqs[0]; !proto.Equal(want, got) {
  1230. t.Errorf("wrong request %q, want %q", got, want)
  1231. }
  1232. if want, got := expectedResponse, resp; !proto.Equal(want, got) {
  1233. t.Errorf("wrong response %q, want %q)", got, want)
  1234. }
  1235. }
  1236. func TestCloudTasksRunTaskError(t *testing.T) {
  1237. errCode := codes.PermissionDenied
  1238. mockCloudTasks.err = gstatus.Error(errCode, "test error")
  1239. var formattedName string = fmt.Sprintf("projects/%s/locations/%s/queues/%s/tasks/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]", "[TASK]")
  1240. var request = &taskspb.RunTaskRequest{
  1241. Name: formattedName,
  1242. }
  1243. c, err := NewClient(context.Background(), clientOpt)
  1244. if err != nil {
  1245. t.Fatal(err)
  1246. }
  1247. resp, err := c.RunTask(context.Background(), request)
  1248. if st, ok := gstatus.FromError(err); !ok {
  1249. t.Errorf("got error %v, expected grpc error", err)
  1250. } else if c := st.Code(); c != errCode {
  1251. t.Errorf("got error code %q, want %q", c, errCode)
  1252. }
  1253. _ = resp
  1254. }