Commitfe84d77

committed

Add GitHub#temporary_basic_auth context manager

- Included: Tests!

1 parent42ce39e commitfe84d77Copy full SHA for fe84d77

File tree

+29

-0

lines changed

+29

-0

lines changed

Lines changed: 13 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -4,6 +4,7 @@`
`4`	`4`	`fromcollectionsimportCallable`
`5`	`5`	`fromgithub3import__version__`
`6`	`6`	`fromloggingimportgetLogger`
	`7`	`+fromcontextlibimportcontextmanager`
`7`	`8`
`8`	`9`	`__url_cache__= {}`
`9`	`10`	`__logs__=getLogger(__package__)`
`@@ -109,3 +110,15 @@ def token_auth(self, token):`
`109`	`110`	`})`
`110`	`111`	`# Unset username/password so we stop sending them`
`111`	`112`	`self.auth=None`
	`113`	`+`
	`114`	`+@contextmanager`
	`115`	`+deftemporary_basic_auth(self,*auth):`
	`116`	`+old_basic_auth=self.auth`
	`117`	`+old_token_auth=self.headers.get('Authorization')`
	`118`	`+`
	`119`	`+self.basic_auth(*auth)`
	`120`	`+yield`
	`121`	`+`
	`122`	`+self.auth=old_basic_auth`
	`123`	`+ifold_token_auth:`
	`124`	`+self.headers['Authorization']=old_token_auth`

Lines changed: 16 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -184,3 +184,19 @@ def test_issubclass_of_requests_Session(self):`
`184`	`184`	`"""Test that GitHubSession is a subclass of requests.Session"""`
`185`	`185`	`assertissubclass(session.GitHubSession,`
`186`	`186`	`requests.Session)`
	`187`	`+`
	`188`	`+deftest_can_use_temporary_basic_auth(self):`
	`189`	`+"""Test that temporary_basic_auth resets old auth."""`
	`190`	`+s=self.build_session()`
	`191`	`+s.basic_auth('foo','bar')`
	`192`	`+withs.temporary_basic_auth('temp','pass'):`
	`193`	`+asserts.auth!= ('foo','bar')`
	`194`	`+`
	`195`	`+asserts.auth== ('foo','bar')`
	`196`	`+`
	`197`	`+deftest_temporary_basic_auth_replaces_auth(self):`
	`198`	`+"""Test that temporary_basic_auth sets the proper credentials."""`
	`199`	`+s=self.build_session()`
	`200`	`+s.basic_auth('foo','bar')`
	`201`	`+withs.temporary_basic_auth('temp','pass'):`
	`202`	`+asserts.auth== ('temp','pass')`

Comments

(0)