@@ -341,7 +341,7 @@ def test_list_metrics(self):
341
341
self .assertFalse (metric .exists ())
342
342
before_metrics = list (Config .CLIENT .list_metrics ())
343
343
before_names = set (before .name for before in before_metrics )
344
- self .failIf (metric .name in before_names )
344
+ self .assertFalse (metric .name in before_names )
345
345
metric .create ()
346
346
self .to_delete .append (metric )
347
347
self .assertTrue (metric .exists ())
@@ -406,9 +406,10 @@ def test_create_sink_storage_bucket(self):
406
406
uri = self ._init_storage_bucket ()
407
407
SINK_NAME = 'test-create-sink-bucket%s' % (_RESOURCE_ID ,)
408
408
409
+ retry = RetryErrors ((Conflict , ServiceUnavailable ), max_tries = 10 )
409
410
sink = Config .CLIENT .sink (SINK_NAME , DEFAULT_FILTER , uri )
410
411
self .assertFalse (sink .exists ())
411
- sink .create ()
412
+ retry ( sink .create ) ()
412
413
self .to_delete .append (sink )
413
414
self .assertTrue (sink .exists ())
414
415
@@ -434,9 +435,10 @@ def test_create_sink_pubsub_topic(self):
434
435
435
436
TOPIC_URI = 'pubsub.googleapis.com/%s' % (topic_path ,)
436
437
438
+ retry = RetryErrors ((Conflict , ServiceUnavailable ), max_tries = 10 )
437
439
sink = Config .CLIENT .sink (SINK_NAME , DEFAULT_FILTER , TOPIC_URI )
438
440
self .assertFalse (sink .exists ())
439
- sink .create ()
441
+ retry ( sink .create ) ()
440
442
self .to_delete .append (sink )
441
443
self .assertTrue (sink .exists ())
442
444
@@ -465,22 +467,24 @@ def _init_bigquery_dataset(self):
465
467
466
468
def test_create_sink_bigquery_dataset (self ):
467
469
SINK_NAME = 'test-create-sink-dataset%s' % (_RESOURCE_ID ,)
470
+ retry = RetryErrors ((Conflict , ServiceUnavailable ), max_tries = 10 )
468
471
uri = self ._init_bigquery_dataset ()
469
472
sink = Config .CLIENT .sink (SINK_NAME , DEFAULT_FILTER , uri )
470
473
self .assertFalse (sink .exists ())
471
- sink .create ()
474
+ retry ( sink .create ) ()
472
475
self .to_delete .append (sink )
473
476
self .assertTrue (sink .exists ())
474
477
475
478
def test_list_sinks (self ):
476
479
SINK_NAME = 'test-list-sinks%s' % (_RESOURCE_ID ,)
477
480
uri = self ._init_storage_bucket ()
481
+ retry = RetryErrors ((Conflict , ServiceUnavailable ), max_tries = 10 )
478
482
sink = Config .CLIENT .sink (SINK_NAME , DEFAULT_FILTER , uri )
479
483
self .assertFalse (sink .exists ())
480
484
before_sinks = list (Config .CLIENT .list_sinks ())
481
485
before_names = set (before .name for before in before_sinks )
482
- self .failIf (sink .name in before_names )
483
- sink .create ()
486
+ self .assertFalse (sink .name in before_names )
487
+ retry ( sink .create ) ()
484
488
self .to_delete .append (sink )
485
489
self .assertTrue (sink .exists ())
486
490
after_sinks = list (Config .CLIENT .list_sinks ())
@@ -489,7 +493,7 @@ def test_list_sinks(self):
489
493
490
494
def test_reload_sink (self ):
491
495
SINK_NAME = 'test-reload-sink%s' % (_RESOURCE_ID ,)
492
- retry = RetryErrors (Conflict )
496
+ retry = RetryErrors (( Conflict , ServiceUnavailable ), max_tries = 10 )
493
497
uri = self ._init_bigquery_dataset ()
494
498
sink = Config .CLIENT .sink (SINK_NAME , DEFAULT_FILTER , uri )
495
499
self .assertFalse (sink .exists ())
@@ -503,7 +507,7 @@ def test_reload_sink(self):
503
507
504
508
def test_update_sink (self ):
505
509
SINK_NAME = 'test-update-sink%s' % (_RESOURCE_ID ,)
506
- retry = RetryErrors (Conflict , max_tries = 10 )
510
+ retry = RetryErrors (( Conflict , ServiceUnavailable ) , max_tries = 10 )
507
511
bucket_uri = self ._init_storage_bucket ()
508
512
dataset_uri = self ._init_bigquery_dataset ()
509
513
UPDATED_FILTER = 'logName:syslog'
0 commit comments