@@ -47,10 +47,9 @@ def _verifyProtobufCall(self, called_with, URI, conn):
47
47
conn .USER_AGENT )
48
48
49
49
def test_default_url (self ):
50
- from gcloud .connection import API_BASE_URL
51
-
50
+ klass = self ._getTargetClass ()
52
51
conn = self ._makeOne ()
53
- self .assertEqual (conn .api_base_url , API_BASE_URL )
52
+ self .assertEqual (conn .api_base_url , klass . API_BASE_URL )
54
53
55
54
def test_custom_url_from_env (self ):
56
55
import os
@@ -143,11 +142,9 @@ def test__request_w_200(self):
143
142
conn = self ._makeOne ()
144
143
URI = '/' .join ([
145
144
conn .api_base_url ,
146
- 'datastore' ,
147
145
conn .API_VERSION ,
148
- 'datasets' ,
149
- PROJECT ,
150
- METHOD ,
146
+ 'projects' ,
147
+ PROJECT + ':' + METHOD ,
151
148
])
152
149
http = conn ._http = Http ({'status' : '200' }, 'CONTENT' )
153
150
self .assertEqual (conn ._request (PROJECT , METHOD , DATA ), 'CONTENT' )
@@ -189,11 +186,9 @@ def FromString(cls, pb):
189
186
conn = self ._makeOne ()
190
187
URI = '/' .join ([
191
188
conn .api_base_url ,
192
- 'datastore' ,
193
189
conn .API_VERSION ,
194
- 'datasets' ,
195
- PROJECT ,
196
- METHOD ,
190
+ 'projects' ,
191
+ PROJECT + ':' + METHOD ,
197
192
])
198
193
http = conn ._http = Http ({'status' : '200' }, 'CONTENT' )
199
194
response = conn ._rpc (PROJECT , METHOD , ReqPB (), RspPB )
@@ -208,11 +203,9 @@ def test_build_api_url_w_default_base_version(self):
208
203
conn = self ._makeOne ()
209
204
URI = '/' .join ([
210
205
conn .api_base_url ,
211
- 'datastore' ,
212
206
conn .API_VERSION ,
213
- 'datasets' ,
214
- PROJECT ,
215
- METHOD ,
207
+ 'projects' ,
208
+ PROJECT + ':' + METHOD ,
216
209
])
217
210
self .assertEqual (conn .build_api_url (PROJECT , METHOD ), URI )
218
211
@@ -224,11 +217,9 @@ def test_build_api_url_w_explicit_base_version(self):
224
217
conn = self ._makeOne ()
225
218
URI = '/' .join ([
226
219
BASE ,
227
- 'datastore' ,
228
220
VER ,
229
- 'datasets' ,
230
- PROJECT ,
231
- METHOD ,
221
+ 'projects' ,
222
+ PROJECT + ':' + METHOD ,
232
223
])
233
224
self .assertEqual (conn .build_api_url (PROJECT , METHOD , BASE , VER ),
234
225
URI )
@@ -242,11 +233,9 @@ def test_lookup_single_key_empty_response(self):
242
233
conn = self ._makeOne ()
243
234
URI = '/' .join ([
244
235
conn .api_base_url ,
245
- 'datastore' ,
246
236
conn .API_VERSION ,
247
- 'datasets' ,
248
- PROJECT ,
249
- 'lookup' ,
237
+ 'projects' ,
238
+ PROJECT + ':lookup' ,
250
239
])
251
240
http = conn ._http = Http ({'status' : '200' }, rsp_pb .SerializeToString ())
252
241
found , missing , deferred = conn .lookup (PROJECT , [key_pb ])
@@ -271,11 +260,9 @@ def test_lookup_single_key_empty_response_w_eventual(self):
271
260
conn = self ._makeOne ()
272
261
URI = '/' .join ([
273
262
conn .api_base_url ,
274
- 'datastore' ,
275
263
conn .API_VERSION ,
276
- 'datasets' ,
277
- PROJECT ,
278
- 'lookup' ,
264
+ 'projects' ,
265
+ PROJECT + ':lookup' ,
279
266
])
280
267
http = conn ._http = Http ({'status' : '200' }, rsp_pb .SerializeToString ())
281
268
found , missing , deferred = conn .lookup (PROJECT , [key_pb ],
@@ -313,11 +300,9 @@ def test_lookup_single_key_empty_response_w_transaction(self):
313
300
conn = self ._makeOne ()
314
301
URI = '/' .join ([
315
302
conn .api_base_url ,
316
- 'datastore' ,
317
303
conn .API_VERSION ,
318
- 'datasets' ,
319
- PROJECT ,
320
- 'lookup' ,
304
+ 'projects' ,
305
+ PROJECT + ':lookup' ,
321
306
])
322
307
http = conn ._http = Http ({'status' : '200' }, rsp_pb .SerializeToString ())
323
308
found , missing , deferred = conn .lookup (PROJECT , [key_pb ],
@@ -348,11 +333,9 @@ def test_lookup_single_key_nonempty_response(self):
348
333
conn = self ._makeOne ()
349
334
URI = '/' .join ([
350
335
conn .api_base_url ,
351
- 'datastore' ,
352
336
conn .API_VERSION ,
353
- 'datasets' ,
354
- PROJECT ,
355
- 'lookup' ,
337
+ 'projects' ,
338
+ PROJECT + ':lookup' ,
356
339
])
357
340
http = conn ._http = Http ({'status' : '200' }, rsp_pb .SerializeToString ())
358
341
(found ,), missing , deferred = conn .lookup (PROJECT , [key_pb ])
@@ -379,11 +362,9 @@ def test_lookup_multiple_keys_empty_response(self):
379
362
conn = self ._makeOne ()
380
363
URI = '/' .join ([
381
364
conn .api_base_url ,
382
- 'datastore' ,
383
365
conn .API_VERSION ,
384
- 'datasets' ,
385
- PROJECT ,
386
- 'lookup' ,
366
+ 'projects' ,
367
+ PROJECT + ':lookup' ,
387
368
])
388
369
http = conn ._http = Http ({'status' : '200' }, rsp_pb .SerializeToString ())
389
370
found , missing , deferred = conn .lookup (PROJECT , [key_pb1 , key_pb2 ])
@@ -414,11 +395,9 @@ def test_lookup_multiple_keys_w_missing(self):
414
395
conn = self ._makeOne ()
415
396
URI = '/' .join ([
416
397
conn .api_base_url ,
417
- 'datastore' ,
418
398
conn .API_VERSION ,
419
- 'datasets' ,
420
- PROJECT ,
421
- 'lookup' ,
399
+ 'projects' ,
400
+ PROJECT + ':lookup' ,
422
401
])
423
402
http = conn ._http = Http ({'status' : '200' }, rsp_pb .SerializeToString ())
424
403
result , missing , deferred = conn .lookup (PROJECT , [key_pb1 , key_pb2 ])
@@ -448,11 +427,9 @@ def test_lookup_multiple_keys_w_deferred(self):
448
427
conn = self ._makeOne ()
449
428
URI = '/' .join ([
450
429
conn .api_base_url ,
451
- 'datastore' ,
452
430
conn .API_VERSION ,
453
- 'datasets' ,
454
- PROJECT ,
455
- 'lookup' ,
431
+ 'projects' ,
432
+ PROJECT + ':lookup' ,
456
433
])
457
434
http = conn ._http = Http ({'status' : '200' }, rsp_pb .SerializeToString ())
458
435
result , missing , deferred = conn .lookup (PROJECT , [key_pb1 , key_pb2 ])
@@ -490,11 +467,9 @@ def test_run_query_w_eventual_no_transaction(self):
490
467
conn = self ._makeOne ()
491
468
URI = '/' .join ([
492
469
conn .api_base_url ,
493
- 'datastore' ,
494
470
conn .API_VERSION ,
495
- 'datasets' ,
496
- PROJECT ,
497
- 'runQuery' ,
471
+ 'projects' ,
472
+ PROJECT + ':runQuery' ,
498
473
])
499
474
http = conn ._http = Http ({'status' : '200' }, rsp_pb .SerializeToString ())
500
475
pbs , end , more , skipped = conn .run_query (PROJECT , q_pb ,
@@ -531,11 +506,9 @@ def test_run_query_wo_eventual_w_transaction(self):
531
506
conn = self ._makeOne ()
532
507
URI = '/' .join ([
533
508
conn .api_base_url ,
534
- 'datastore' ,
535
509
conn .API_VERSION ,
536
- 'datasets' ,
537
- PROJECT ,
538
- 'runQuery' ,
510
+ 'projects' ,
511
+ PROJECT + ':runQuery' ,
539
512
])
540
513
http = conn ._http = Http ({'status' : '200' }, rsp_pb .SerializeToString ())
541
514
pbs , end , more , skipped = conn .run_query (
@@ -589,11 +562,9 @@ def test_run_query_wo_namespace_empty_result(self):
589
562
conn = self ._makeOne ()
590
563
URI = '/' .join ([
591
564
conn .api_base_url ,
592
- 'datastore' ,
593
565
conn .API_VERSION ,
594
- 'datasets' ,
595
- PROJECT ,
596
- 'runQuery' ,
566
+ 'projects' ,
567
+ PROJECT + ':runQuery' ,
597
568
])
598
569
http = conn ._http = Http ({'status' : '200' }, rsp_pb .SerializeToString ())
599
570
pbs , end , more , skipped = conn .run_query (PROJECT , q_pb )
@@ -624,11 +595,9 @@ def test_run_query_w_namespace_nonempty_result(self):
624
595
conn = self ._makeOne ()
625
596
URI = '/' .join ([
626
597
conn .api_base_url ,
627
- 'datastore' ,
628
598
conn .API_VERSION ,
629
- 'datasets' ,
630
- PROJECT ,
631
- 'runQuery' ,
599
+ 'projects' ,
600
+ PROJECT + ':runQuery' ,
632
601
])
633
602
http = conn ._http = Http ({'status' : '200' }, rsp_pb .SerializeToString ())
634
603
pbs = conn .run_query (PROJECT , q_pb , 'NS' )[0 ]
@@ -651,11 +620,9 @@ def test_begin_transaction(self):
651
620
conn = self ._makeOne ()
652
621
URI = '/' .join ([
653
622
conn .api_base_url ,
654
- 'datastore' ,
655
623
conn .API_VERSION ,
656
- 'datasets' ,
657
- PROJECT ,
658
- 'beginTransaction' ,
624
+ 'projects' ,
625
+ PROJECT + ':beginTransaction' ,
659
626
])
660
627
http = conn ._http = Http ({'status' : '200' }, rsp_pb .SerializeToString ())
661
628
self .assertEqual (conn .begin_transaction (PROJECT ), TRANSACTION )
@@ -684,11 +651,9 @@ def test_commit_wo_transaction(self):
684
651
conn = self ._makeOne ()
685
652
URI = '/' .join ([
686
653
conn .api_base_url ,
687
- 'datastore' ,
688
654
conn .API_VERSION ,
689
- 'datasets' ,
690
- PROJECT ,
691
- 'commit' ,
655
+ 'projects' ,
656
+ PROJECT + ':commit' ,
692
657
])
693
658
http = conn ._http = Http ({'status' : '200' }, rsp_pb .SerializeToString ())
694
659
@@ -732,11 +697,9 @@ def test_commit_w_transaction(self):
732
697
conn = self ._makeOne ()
733
698
URI = '/' .join ([
734
699
conn .api_base_url ,
735
- 'datastore' ,
736
700
conn .API_VERSION ,
737
- 'datasets' ,
738
- PROJECT ,
739
- 'commit' ,
701
+ 'projects' ,
702
+ PROJECT + ':commit' ,
740
703
])
741
704
http = conn ._http = Http ({'status' : '200' }, rsp_pb .SerializeToString ())
742
705
@@ -771,11 +734,9 @@ def test_rollback_ok(self):
771
734
conn = self ._makeOne ()
772
735
URI = '/' .join ([
773
736
conn .api_base_url ,
774
- 'datastore' ,
775
737
conn .API_VERSION ,
776
- 'datasets' ,
777
- PROJECT ,
778
- 'rollback' ,
738
+ 'projects' ,
739
+ PROJECT + ':rollback' ,
779
740
])
780
741
http = conn ._http = Http ({'status' : '200' }, rsp_pb .SerializeToString ())
781
742
self .assertEqual (conn .rollback (PROJECT , TRANSACTION ), None )
@@ -794,11 +755,9 @@ def test_allocate_ids_empty(self):
794
755
conn = self ._makeOne ()
795
756
URI = '/' .join ([
796
757
conn .api_base_url ,
797
- 'datastore' ,
798
758
conn .API_VERSION ,
799
- 'datasets' ,
800
- PROJECT ,
801
- 'allocateIds' ,
759
+ 'projects' ,
760
+ PROJECT + ':allocateIds' ,
802
761
])
803
762
http = conn ._http = Http ({'status' : '200' }, rsp_pb .SerializeToString ())
804
763
self .assertEqual (conn .allocate_ids (PROJECT , []), [])
@@ -827,11 +786,9 @@ def test_allocate_ids_non_empty(self):
827
786
conn = self ._makeOne ()
828
787
URI = '/' .join ([
829
788
conn .api_base_url ,
830
- 'datastore' ,
831
789
conn .API_VERSION ,
832
- 'datasets' ,
833
- PROJECT ,
834
- 'allocateIds' ,
790
+ 'projects' ,
791
+ PROJECT + ':allocateIds' ,
835
792
])
836
793
http = conn ._http = Http ({'status' : '200' }, rsp_pb .SerializeToString ())
837
794
self .assertEqual (conn .allocate_ids (PROJECT , before_key_pbs ),
0 commit comments