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.
 
 
 

1516 lines
45 KiB

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