def test_restoring_session_sets_logged_in(self):
fs = familysearch.FamilySearch(self.agent, self.key)
self.assertFalse(fs.logged_in, 'should not be logged in by default')
fs = familysearch.FamilySearch(self.agent,
self.key,
session=self.session)
self.assertTrue(fs.logged_in,
'should be logged in after restoring session')
def test_username_and_password_set_logged_in(self):
add_request_intercept(sample_login)
fs = familysearch.FamilySearch(self.agent, self.key, self.username,
self.password)
self.assertTrue(
fs.logged_in,
'should be logged in after providing username and password')
def test_pickle_restores_oauth_temporary_credentials(self):
add_request_intercept(sample_identity_properties)
add_request_intercept(sample_request_token,
port=1,
headers={'Content-type': 'text/plain'})
fs_oauth = familysearch.FamilySearch(self.agent, self.key)
fs_oauth.request_token()
fs_oauth_restored = pickle.loads(pickle.dumps(fs_oauth))
self.assertEqual(fs_oauth.agent, fs_oauth_restored.agent,
'user agent not restored properly')
self.assertEqual(fs_oauth.key, fs_oauth_restored.key,
'developer key not restored properly')
self.assertEqual(fs_oauth.session_id, fs_oauth_restored.session_id,
'session ID not restored properly')
self.assertEqual(fs_oauth_restored.logged_in, False,
'logged-in flag not restored properly')
self.assertEqual(fs_oauth.base, fs_oauth_restored.base,
'base URL not restored properly')
self.assertIn(
self.oauth_temp_token, fs_oauth_restored.oauth_secrets,
'OAuth temporary credentials identifier not restored properly')
self.assertEqual(
fs_oauth_restored.oauth_secrets[self.oauth_temp_token],
self.oauth_secret,
'OAuth temporary credentials shared-secret not restored properly')
def test_changes_base(self):
add_request_intercept(sample_person1, host='www.dev.usys.org', port=80)
add_request_intercept(sample_person2,
host='api.familysearch.org',
port=443)
fs_dev = familysearch.FamilySearch(self.agent, self.key)
fs_prod = familysearch.FamilySearch(
self.agent, self.key, base='https://api.familysearch.org')
person1 = fs_dev.person()
person2 = fs_prod.person()
self.assertNotEqual(person1, person2,
'base argument failed to change base URL')
self.assertEqual(person1['id'],
json.loads(sample_person1)['persons'][0]['id'],
'wrong person returned from default base')
self.assertEqual(person2['id'],
json.loads(sample_person2)['persons'][0]['id'],
'wrong person returned from production base')
def test_base_fs_creation(self):
fs = familysearch.FamilySearch(self.agent, self.devkey)
self.assertTrue(fs.base == 'https://sandbox.familysearch.org')
print("Base is correct.")
self.assertTrue(fs.dev_key == self.devkey)
print("Key is correct.")
self.assertTrue(isinstance(fs.opener, request.OpenerDirector))
print("HTTP opener works.")
self.assertIsNone(fs.access_token)
print("Access token works.")
def test_includes_user_agent(self):
request_environ = add_request_intercept(sample_person1)
fs = familysearch.FamilySearch(self.agent, self.key)
fs.person()
self.assertIn(self.agent, fs.agent,
'user agent not included in internal user agent')
self.assertIn('HTTP_USER_AGENT', request_environ,
'user agent header not included in request')
self.assertIn(self.agent, request_environ['HTTP_USER_AGENT'],
'user agent not included in user agent header')
def test_not_logged_in_if_error_401(self):
add_request_intercept('', status='401 Unauthorized')
fs = familysearch.FamilySearch(self.agent,
self.key,
session=self.session)
self.assertTrue(fs.logged_in,
'should be logged in after restoring session')
self.assertRaises(urllib2.HTTPError, fs.person)
self.assertFalse(fs.logged_in,
'should not be logged in after receiving error 401')
def setUp(self):
self.longMessage = True
self.agent = 'TEST_USER_AGENT'
self.key = 'FAKE_DEV_KEY'
self.session = 'FAKE_SESSION_ID'
self.id = 'FAKE_PERSON_ID'
self.id2 = 'FAKE_PERSON_ID_2'
wsgi_intercept.httplib_intercept.install()
self.fs = familysearch.FamilySearch(self.agent,
self.key,
session=self.session)
def test_requests_json_format(self):
request_environ = add_request_intercept(sample_person1)
fs = familysearch.FamilySearch(self.agent,
self.key,
session=self.session)
fs.person()
self.assertIn('QUERY_STRING', request_environ,
'query string not included in request')
self.assertIn('dataFormat=application%2Fjson',
request_environ['QUERY_STRING'],
'dataFormat not included in query string')
def setUp(self):
self.longMessage = True
self.agent = 'TEST_USER_AGENT'
self.key = 'FAKE_DEV_KEY'
self.session = 'FAKE_SESSION_ID'
self.username = '******'
self.password = '******'
self.callback = 'FAKE_CALLBACK'
self.temp_token = 'FAKE_TEMP_TOKEN'
self.secret = 'FAKE_SECRET'
self.verifier = 'FAKE_VERIFIER'
self.real_token = 'FAKE_TOKEN'
wsgi_intercept.httplib_intercept.install()
self.fs = familysearch.FamilySearch(self.agent, self.key)
def test_pickle_restores_logged_out_session(self):
fs_logged_out = familysearch.FamilySearch(
self.agent, self.key, base='https://api.familysearch.org')
fs_logged_out_restored = pickle.loads(pickle.dumps(fs_logged_out))
self.assertEqual(fs_logged_out.agent, fs_logged_out_restored.agent,
'user agent not restored properly')
self.assertEqual(fs_logged_out.key, fs_logged_out_restored.key,
'developer key not restored properly')
self.assertEqual(fs_logged_out.session_id,
fs_logged_out_restored.session_id,
'session ID not restored properly')
self.assertEqual(fs_logged_out_restored.logged_in, False,
'logged-in flag not restored properly')
self.assertEqual(fs_logged_out.base, fs_logged_out_restored.base,
'base URL not restored properly')
def test_passes_cookies_back(self):
fs = familysearch.FamilySearch(self.agent, self.key)
# First request sets a cookie
headers = default_headers.copy()
headers['Set-Cookie'] = self.cookie + '; Path=/'
add_request_intercept(sample_login, headers=headers)
fs.login(self.username, self.password)
# Second request should receive the cookie back
request_environ = add_request_intercept(sample_person1)
fs.person()
self.assertIn('HTTP_COOKIE', request_environ,
'cookie header not included in request')
self.assertIn(self.cookie, request_environ['HTTP_COOKIE'],
'previously-set cookie not included in cookie header')
def create_proxy():
return familysearch.FamilySearch(user_agent,
developer_key,
base=familysearch_base_url)
def test_accepts_user_agent_and_dev_key(self):
familysearch.FamilySearch(agent=self.agent, key=self.key)
