class ClientTest(unittest.TestCase): """Tests for the adspygoogle.common.Client module.""" def setUp(self): """Initialize a Client to test with.""" self.client = Client() def testLoadAuthCredentials(self): """Tests the _LoadAuthCredentials function.""" _, filename = tempfile.mkstemp() auth_credentials = { 'username': '******', 'password': '******' } with open(filename, 'w') as handle: pickle.dump(auth_credentials, handle) Client.auth_pkl = filename self.assertEqual(self.client._LoadAuthCredentials(), auth_credentials) Client.auth_pkl = '' def testLoadAuthCredentials_noPickle(self): """Tests the _LoadAuthCredentials function.""" try: self.client._LoadAuthCredentials() self.fail('Exception should have been thrown.') except ValidationError, e: self.assertEqual(str(e), 'Authentication data is missing.')
class ClientTest(unittest.TestCase): """Tests for the adspygoogle.common.Client module.""" def setUp(self): """Initialize a Client to test with.""" self.client = Client() def testLoadAuthCredentials_ClientLogin(self): """Tests the _LoadAuthCredentials function.""" _, filename = tempfile.mkstemp() auth_credentials = { 'username': '******', 'password': '******' } with open(filename, 'w') as handle: pickle.dump(auth_credentials, handle) try: Client.auth_pkl = filename self.assertEqual(self.client._LoadAuthCredentials(), auth_credentials) finally: Client.auth_pkl = '' def testLoadAuthCredentials_OAuth2(self): """Tests the _LoadAuthCredentials function.""" _, filename = tempfile.mkstemp() client_id = 'id1234id', client_secret = 'shhh,itsasecret', refresh_token = '1/not_a_refresh_token' auth_credentials = { 'clientId': client_id, 'clientSecret': client_secret, 'refreshToken': refresh_token } with open(filename, 'w') as handle: pickle.dump(auth_credentials, handle) try: Client.auth_pkl = filename read_credentials = self.client._LoadAuthCredentials() self.assertEqual(read_credentials['oauth2credentials'].refresh_token, refresh_token) self.assertEqual(read_credentials['oauth2credentials'].client_secret, client_secret) self.assertEqual(read_credentials['oauth2credentials'].client_id, client_id) self.assertTrue('clientId' not in read_credentials) self.assertTrue('clientSecret' not in read_credentials) self.assertTrue('refreshToken' not in read_credentials) finally: Client.auth_pkl = '' def testLoadAuthCredentials_noPickle(self): """Tests the _LoadAuthCredentials function.""" try: self.client._LoadAuthCredentials() self.fail('Exception should have been thrown.') except ValidationError, e: self.assertEqual(str(e), 'Authentication data is missing.')
class ClientTest(unittest.TestCase): """Tests for the adspygoogle.common.Client module.""" def setUp(self): """Initialize a Client to test with.""" self.client = Client() def testLoadAuthCredentials_ClientLogin(self): """Tests the _LoadAuthCredentials function.""" _, filename = tempfile.mkstemp() auth_credentials = {'username': '******', 'password': '******'} with open(filename, 'w') as handle: pickle.dump(auth_credentials, handle) try: Client.auth_pkl = filename self.assertEqual(self.client._LoadAuthCredentials(), auth_credentials) finally: Client.auth_pkl = '' def testLoadAuthCredentials_OAuth2(self): """Tests the _LoadAuthCredentials function.""" _, filename = tempfile.mkstemp() client_id = 'id1234id', client_secret = 'shhh,itsasecret', refresh_token = '1/not_a_refresh_token' auth_credentials = { 'clientId': client_id, 'clientSecret': client_secret, 'refreshToken': refresh_token } with open(filename, 'w') as handle: pickle.dump(auth_credentials, handle) try: Client.auth_pkl = filename read_credentials = self.client._LoadAuthCredentials() self.assertEqual( read_credentials['oauth2credentials'].refresh_token, refresh_token) self.assertEqual( read_credentials['oauth2credentials'].client_secret, client_secret) self.assertEqual(read_credentials['oauth2credentials'].client_id, client_id) self.assertTrue('clientId' not in read_credentials) self.assertTrue('clientSecret' not in read_credentials) self.assertTrue('refreshToken' not in read_credentials) finally: Client.auth_pkl = '' def testLoadAuthCredentials_noPickle(self): """Tests the _LoadAuthCredentials function.""" try: self.client._LoadAuthCredentials() self.fail('Exception should have been thrown.') except ValidationError, e: self.assertEqual(str(e), 'Authentication data is missing.')