8
8
from django .views .generic .base import ContextMixin
9
9
from django_dynamic_fixture import get , new
10
10
11
- from readthedocs .builds .constants import EXTERNAL
11
+ from readthedocs .builds .constants import BUILD_STATE_FINISHED , EXTERNAL
12
12
from readthedocs .builds .models import Build , Version
13
13
from readthedocs .integrations .models import GenericAPIWebhook , GitHubWebhook
14
14
from readthedocs .oauth .models import RemoteRepository , RemoteRepositoryRelation
@@ -517,28 +517,56 @@ def test_badge_caching(self):
517
517
self .assertTrue ('no-cache' in res ['Cache-Control' ])
518
518
519
519
def test_passing_badge (self ):
520
- get (Build , project = self .project , version = self .version , success = True )
521
- res = self .client .get (self .badge_url , {'version' : self .version .slug })
522
- self .assertContains (res , 'passing' )
523
- self .assertEqual (res ['Content-Type' ], 'image/svg+xml' )
520
+ get (
521
+ Build ,
522
+ project = self .project ,
523
+ version = self .version ,
524
+ success = True ,
525
+ state = BUILD_STATE_FINISHED ,
526
+ )
527
+ res = self .client .get (self .badge_url , {"version" : self .version .slug })
528
+ self .assertContains (res , "passing" )
529
+ self .assertEqual (res ["Content-Type" ], "image/svg+xml" )
524
530
525
531
def test_failing_badge (self ):
526
- get (Build , project = self .project , version = self .version , success = False )
527
- res = self .client .get (self .badge_url , {'version' : self .version .slug })
528
- self .assertContains (res , 'failing' )
532
+ get (
533
+ Build ,
534
+ project = self .project ,
535
+ version = self .version ,
536
+ success = False ,
537
+ state = BUILD_STATE_FINISHED ,
538
+ )
539
+ res = self .client .get (self .badge_url , {"version" : self .version .slug })
540
+ self .assertContains (res , "failing" )
529
541
530
542
def test_plastic_failing_badge (self ):
531
- get (Build , project = self .project , version = self .version , success = False )
532
- res = self .client .get (self .badge_url , {'version' : self .version .slug , 'style' : 'plastic' })
533
- self .assertContains (res , 'failing' )
543
+ get (
544
+ Build ,
545
+ project = self .project ,
546
+ version = self .version ,
547
+ success = False ,
548
+ state = BUILD_STATE_FINISHED ,
549
+ )
550
+ res = self .client .get (
551
+ self .badge_url , {"version" : self .version .slug , "style" : "plastic" }
552
+ )
553
+ self .assertContains (res , "failing" )
534
554
535
555
# The plastic badge has slightly more rounding
536
556
self .assertContains (res , 'rx="4"' )
537
557
538
558
def test_social_passing_badge (self ):
539
- get (Build , project = self .project , version = self .version , success = True )
540
- res = self .client .get (self .badge_url , {'version' : self .version .slug , 'style' : 'social' })
541
- self .assertContains (res , 'passing' )
559
+ get (
560
+ Build ,
561
+ project = self .project ,
562
+ version = self .version ,
563
+ success = True ,
564
+ state = BUILD_STATE_FINISHED ,
565
+ )
566
+ res = self .client .get (
567
+ self .badge_url , {"version" : self .version .slug , "style" : "social" }
568
+ )
569
+ self .assertContains (res , "passing" )
542
570
543
571
# The social badge (but not the other badges) has this element
544
572
self .assertContains (res , 'rlink' )
@@ -556,9 +584,15 @@ def test_private_version(self):
556
584
self .version .save ()
557
585
558
586
# Without a token, badge is unknown
559
- get (Build , project = self .project , version = self .version , success = True )
560
- res = self .client .get (self .badge_url , {'version' : self .version .slug })
561
- self .assertContains (res , 'unknown' )
587
+ get (
588
+ Build ,
589
+ project = self .project ,
590
+ version = self .version ,
591
+ success = True ,
592
+ state = BUILD_STATE_FINISHED ,
593
+ )
594
+ res = self .client .get (self .badge_url , {"version" : self .version .slug })
595
+ self .assertContains (res , "unknown" )
562
596
563
597
# With an invalid token, the badge is unknown
564
598
res = self .client .get (
0 commit comments