You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 

1555 lines
44 KiB

  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. }