19
19
import os
20
20
import sys
21
21
import unittest
22
+ from typing import Callable , List
22
23
23
24
from sauceclient import SauceClient
24
25
29
30
sauce = SauceClient (SAUCE_USERNAME , SAUCE_ACCESS_KEY )
30
31
31
32
32
- def on_platforms (platforms ) :
33
- def decorator (base_class ) :
33
+ def on_platforms (platforms : List [ str ]) -> Callable [[ type ], None ] :
34
+ def decorator (base_class : type ) -> None :
34
35
module = sys .modules [base_class .__module__ ].__dict__
35
36
for i , platform in enumerate (platforms ):
36
37
name = "%s_%s" % (base_class .__name__ , i + 1 )
@@ -40,16 +41,16 @@ def decorator(base_class):
40
41
41
42
42
43
class SauceTestCase (unittest .TestCase ):
43
- def setUp (self ):
44
- self .desired_capabilities ['name' ] = self .id ()
44
+ def setUp (self ) -> None :
45
+ self .desired_capabilities ['name' ] = self .id () # type: ignore
45
46
sauce_url = "http://%s:%s@ondemand.saucelabs.com:80/wd/hub"
46
47
self .driver = webdriver .Remote (
47
- desired_capabilities = self .desired_capabilities ,
48
+ desired_capabilities = self .desired_capabilities , # type: ignore
48
49
command_executor = sauce_url % (SAUCE_USERNAME , SAUCE_ACCESS_KEY )
49
50
)
50
51
self .driver .implicitly_wait (30 )
51
52
52
- def tearDown (self ):
53
+ def tearDown (self ) -> None :
53
54
print ("Link to your job: https://saucelabs.com/jobs/%s" % self .driver .session_id )
54
55
try :
55
56
if sys .exc_info () == (None , None , None ):
0 commit comments