|
|
|
@ -5,23 +5,19 @@
|
|
|
|
|
|
|
|
|
|
"""Unit tests for rietveld.py."""
|
|
|
|
|
|
|
|
|
|
import httplib
|
|
|
|
|
import logging
|
|
|
|
|
import os
|
|
|
|
|
import socket
|
|
|
|
|
import ssl
|
|
|
|
|
import StringIO
|
|
|
|
|
import sys
|
|
|
|
|
import time
|
|
|
|
|
import traceback
|
|
|
|
|
import unittest
|
|
|
|
|
import urllib2
|
|
|
|
|
|
|
|
|
|
sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
|
|
|
|
|
|
|
|
|
|
from testing_support.patches_data import GIT, RAW
|
|
|
|
|
from testing_support import auto_stub
|
|
|
|
|
from third_party import httplib2
|
|
|
|
|
|
|
|
|
|
import patch
|
|
|
|
|
import rietveld
|
|
|
|
@ -494,30 +490,6 @@ class DefaultTimeoutTest(auto_stub.TestCase):
|
|
|
|
|
self.rietveld.post('/api/1234', [('key', 'data')])
|
|
|
|
|
self.assertNotEqual(self.sleep_time, 0)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class OAuthRpcServerTest(auto_stub.TestCase):
|
|
|
|
|
def setUp(self):
|
|
|
|
|
super(OAuthRpcServerTest, self).setUp()
|
|
|
|
|
self.rpc_server = rietveld.OAuthRpcServer(
|
|
|
|
|
'http://www.example.com', 'foo', 'bar')
|
|
|
|
|
|
|
|
|
|
def set_mock_response(self, status):
|
|
|
|
|
def MockHttpRequest(*args, **kwargs):
|
|
|
|
|
return (httplib2.Response({'status': status}), 'body')
|
|
|
|
|
self.mock(self.rpc_server._http, 'request', MockHttpRequest)
|
|
|
|
|
|
|
|
|
|
def test_404(self):
|
|
|
|
|
self.set_mock_response(404)
|
|
|
|
|
with self.assertRaises(urllib2.HTTPError) as ctx:
|
|
|
|
|
self.rpc_server.Send('/foo')
|
|
|
|
|
self.assertEquals(404, ctx.exception.code)
|
|
|
|
|
|
|
|
|
|
def test_200(self):
|
|
|
|
|
self.set_mock_response(200)
|
|
|
|
|
ret = self.rpc_server.Send('/foo')
|
|
|
|
|
self.assertEquals('body', ret)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
|
|
|
logging.basicConfig(level=[
|
|
|
|
|
logging.ERROR, logging.INFO, logging.DEBUG][min(2, sys.argv.count('-v'))])
|
|
|
|
|