+# Django imports
+import django.test.client as client
+import django.test as test
+from django.utils.http import urlencode
+
+# Piston imports
+from piston import oauth
+from piston.models import Consumer, Token
+
+# 3rd/Python party imports
+import httplib2, urllib, cgi
+
+URLENCODED_FORM_CONTENT = 'application/x-www-form-urlencoded'
+
+class OAuthClient(client.Client):
+ def __init__(self, consumer, token):
+ self.token = oauth.OAuthToken(token.key, token.secret)
+ self.consumer = oauth.OAuthConsumer(consumer.key, consumer.secret)
+ self.signature = oauth.OAuthSignatureMethod_HMAC_SHA1()
+
+ super(OAuthClient, self).__init__()
+
+ def request(self, **request):
+ # Figure out parameters from request['QUERY_STRING'] and FakePayload
+ params = {}
+ if request['REQUEST_METHOD'] in ('POST', 'PUT'):
+ if request['CONTENT_TYPE'] == URLENCODED_FORM_CONTENT:
+ payload = request['wsgi.input'].read()
+ request['wsgi.input'] = client.FakePayload(payload)
+ params = cgi.parse_qs(payload)
+
+ url = "http://testserver" + request['PATH_INFO']
+
+ req = oauth.OAuthRequest.from_consumer_and_token(
+ self.consumer, token=self.token,
+ http_method=request['REQUEST_METHOD'], http_url=url,
+ parameters=params
+ )
+
+ req.sign_request(self.signature, self.consumer, self.token)
+ headers = req.to_header()
+ request['HTTP_AUTHORIZATION'] = headers['Authorization']
+
+ return super(OAuthClient, self).request(**request)
+
+ def post(self, path, data={}, content_type=None, follow=False, **extra):
+ if content_type is None:
+ content_type = URLENCODED_FORM_CONTENT
+
+ if isinstance(data, dict):
+ data = urlencode(data)
+
+ return super(OAuthClient, self).post(path, data, content_type, follow, **extra)
+
+class TestCase(test.TestCase):
+ pass
+
+class OAuthTestCase(TestCase):
+ @property
+ def oauth(self):
+ return OAuthClient(self.consumer, self.token)
+