adding unit tests

This commit is contained in:
David Goetz 2011-08-30 12:07:32 -07:00
parent 4eecac0a22
commit bed607b07a
2 changed files with 35 additions and 0 deletions

View File

@ -536,6 +536,27 @@ log_name = yarr'''
os.unlink('/tmp/test') os.unlink('/tmp/test')
self.assertRaises(SystemExit, utils.readconf, '/tmp/test') self.assertRaises(SystemExit, utils.readconf, '/tmp/test')
def test_readconf_raw(self):
conf = '''[section1]
foo = bar
[section2]
log_name = %(yarr)s'''
# setup a real file
with open('/tmp/test', 'wb') as f:
f.write(conf)
make_filename = lambda: '/tmp/test'
# setup a file stream
make_fp = lambda: StringIO(conf)
for conf_object_maker in (make_filename, make_fp):
result = utils.readconf(conf_object_maker(), raw=True)
expected = {'log_name': None,
'section1': {'foo': 'bar'},
'section2': {'log_name': '%(yarr)s'}}
self.assertEquals(result, expected)
os.unlink('/tmp/test')
self.assertRaises(SystemExit, utils.readconf, '/tmp/test')
def test_drop_privileges(self): def test_drop_privileges(self):
user = getuser() user = getuser()
# over-ride os with mock # over-ride os with mock

View File

@ -28,6 +28,7 @@ from StringIO import StringIO
from collections import defaultdict from collections import defaultdict
from eventlet import sleep from eventlet import sleep
from webob import Request
from swift.common import wsgi from swift.common import wsgi
@ -173,6 +174,19 @@ class TestWSGI(unittest.TestCase):
wsgi.sleep = old_sleep wsgi.sleep = old_sleep
wsgi.time = old_time wsgi.time = old_time
def test_pre_auth_req(self):
class FakeReq(object):
@classmethod
def fake_blank(cls, path, environ={}, body='', headers={}):
self.assertEquals(environ['swift.authorize']('test'), None)
self.assertEquals(environ['HTTP_X_TRANS_ID'], '1234')
was_blank = Request.blank
Request.blank = FakeReq.fake_blank
wsgi.make_pre_authed_request({'HTTP_X_TRANS_ID': '1234'},
'PUT', '/', body='tester', headers={})
wsgi.make_pre_authed_request({'HTTP_X_TRANS_ID': '1234'},
'PUT', '/', headers={})
Request.blank = was_blank
if __name__ == '__main__': if __name__ == '__main__':
unittest.main() unittest.main()