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.')
