def setUp(self, *args, **kwargs):
yield super(TestSERVICEWeb, self).setUp(*args, **kwargs)
self.service = SERVICEWeb(
datastore=self.app.db,
)
class TestSERVICEWeb(TestSERVICE):
@defer.inlineCallbacks
def setUp(self, *args, **kwargs):
yield super(TestSERVICEWeb, self).setUp(*args, **kwargs)
self.service = SERVICEWeb(
datastore=self.app.db,
)
def submit_login_form(self, uri, args):
login_form = requests.get(
uri,
headers=self.headers(),
)
self.assertNotEqual(login_form.status_code, 404)
# Get the request to be the actual request we need for the CSRF bypass
# in the post() below.
login_form = self.follow_redirect_until_not(
login_form,
r'twitter.com/',
)
soup = BeautifulSoup(login_form.content)
arguments = self.extract_form_fields(soup.form)
arguments['session[username_or_email]'] = args['username']
arguments['session[password]'] = args['password']
del arguments['cancel']
return requests.post(
soup.form['action'],
headers=self.headers(),
data=arguments,
cookies=login_form.cookies,
# We need to trap the redirect back to localhost:5000
allow_redirects=False,
)
@defer.inlineCallbacks
def test_authorize(self):
redir_uri = 'https://lg-local.example.com/id/auth/callback/SERVICENAME'
# Just match a regex since we'll also get back a requestToken from
# this since it's an OAuth 1.0 Service.
expected_redirect = r'https://api.twitter.com/oauth/authorize\?oauth_token='
redirect_uri = yield self.service.start_authorization(
client_name='testing',
args={'redirect_uri': redir_uri},
)
self.assertRegexpMatches(redirect_uri, expected_redirect)
response = self.submit_login_form(
uri=redirect_uri,
args=dict(
username=AUTHN_TEST_USER['email'],
password=AUTHN_TEST_USER['password'],
),
)
self.assertNotEqual(response.status_code, 404)
response = self.follow_redirect_until_not(response, r'twitter.com')
# So, Twitter's super cool and doesn't 302 us.
# Instead, we get back a 200 page with a meta http-equiv refresh set
# to our callback page. Here, we parse this page and grab the args.
# This is pretty brittle.
self.assertEqual(response.status_code, 200)
soup = BeautifulSoup(response.content)
meta_redir = soup.findAll('meta', {'http-equiv': 'refresh'})[0]
# Strip off the URL and get the QS
meta_qs = meta_redir['content'][6:].split('?')[1]
args = dict(parse_qsl(meta_qs))
try:
auth = yield self.service.finish_authorization(
client_name='testing',
args=args,
)
except CommandContextException as exc:
print exc.context().next_command.context['msg']
raise
oauth_token = auth.token()
user_id = auth.user_id()
oauth_secret = auth.secret()
self.assertEqual(AUTHN_TEST_USER['id'], user_id)
yield self.set_authorization(
user_id=user_id,
token=oauth_token,
secret=oauth_secret
)
yield self.service.daemon_object().run(
authorization=self.authorization,
timestamp=int(time.time()),
)
profile = yield self.call_method('profile')
expected_profile = dict(
name=AUTHN_TEST_USER['name'],
username=AUTHN_TEST_USER['username'],
bio=AUTHN_TEST_USER['bio'],
profile_link=AUTHN_TEST_USER['profile_link'],
profile_picture_link=AUTHN_TEST_USER['profile_picture_url'],
)
self._test_method_result_keys('profile', profile, expected_profile)
num_friends = yield self.call_method('num_friends')
self.assertEqual(
num_friends, {'num': AUTHN_TEST_USER['num_friends']}
)
def test_methods(self):
self.assertEqual(self.service.methods(), {
'most_recent_activity': "The most recent activity on this service's feed",
'profile': "Returns this user's profile.",
'num_friends': 'Number of friends',
'num_friends_interval': 'Number of friends over time',
})
@defer.inlineCallbacks
def test_num_friends_and_interval(self, **kwargs):
yield self.set_authorization()
yield self.write_value(method='num_friends', data='10', timestamp=100)
rv = yield self.call_method('num_friends')
self.assertEqual(rv, {'num': 10})
rv = yield self.call_method('num_friends_interval',
arguments=dict(start=90, end=110),
)
self.assertEqual(rv, {'data': [
{'timestamp': '100', 'num': 10},
]})
yield self.write_value(method='num_friends', data='12', timestamp=105)
rv = yield self.call_method('num_friends')
self.assertEqual(rv, {'num': 12})
rv = yield self.call_method('num_friends_interval',
arguments=dict(start=90, end=110),
)
self.assertEqual(rv, {'data': [
{'timestamp': '100', 'num': 10},
{'timestamp': '105', 'num': 12},
]})
