11import netrc , os , unittest , sys , textwrap
2-
3- from test import support
42from contextlib import ExitStack
3+ from test import support
4+ from test .support import os_helper
55
66try :
77 import pwd
@@ -36,24 +36,15 @@ def __exit__(self, *ignore_exc):
3636 self .stack .close ()
3737
3838 def generate_netrc (
39- self ,
40- content ,
41- filename = ".netrc" ,
42- mode = 0o600 ,
43- encoding = "utf-8" ,
39+ self , content , filename = ".netrc" , mode = 0o600 , encoding = None ,
4440 ):
4541 """Create and return the path to a temporary `.netrc` file."""
46- write_mode = "w"
47- if sys .platform != "cygwin" :
48- write_mode += "t"
49-
5042 netrc_file = os .path .join (self .tmpdir , filename )
43+ write_mode = "w" if sys .platform != "cygwin" else "wt"
5144 with open (netrc_file , mode = write_mode , encoding = encoding ) as fp :
5245 fp .write (textwrap .dedent (content ))
53-
5446 if support .os_helper .can_chmod ():
5547 os .chmod (netrc_file , mode = mode )
56-
5748 return netrc_file
5849
5950
@@ -70,7 +61,6 @@ def use_default_netrc_in_home(*args, **kwargs):
7061 with NetrcEnvironment () as helper :
7162 helper .environ .unset ("NETRC" )
7263 helper .environ .set ("HOME" , helper .tmpdir )
73-
7464 helper .generate_netrc (* args , ** kwargs )
7565 return netrc .netrc ()
7666
@@ -82,7 +72,6 @@ def use_netrc_envvar(*args, **kwargs):
8272 with NetrcEnvironment () as helper :
8373 netrc_file = helper .generate_netrc (* args , ** kwargs )
8474 helper .environ .set ("NETRC" , netrc_file )
85-
8675 return netrc .netrc ()
8776
8877 @staticmethod
@@ -93,7 +82,6 @@ def use_file_argument(*args, **kwargs):
9382 # Just to stress a bit more the test scenario, the NETRC envvar
9483 # will contain rubish information which shouldn't be used
9584 helper .environ .set ("NETRC" , "not-a-file.netrc" )
96-
9785 netrc_file = helper .generate_netrc (* args , ** kwargs )
9886 return netrc .netrc (netrc_file )
9987
0 commit comments