Mark Gius b5ed7916d3 Cleans up test code a bit.
Remove unused imports, unused variables, and some unneeded helper
classes.  Move view-specific testing code into the view test base class.
2011-07-22 14:52:18 -07:00

67 lines
2.6 KiB
Python

# vim: tabstop=4 shiftwidth=4 softtabstop=4
from django import http
from django import test
import mox
from django_openstack.middleware import keystone
class TestCase(test.TestCase):
TEST_STAFF_USER = 'staffUser'
TEST_TENANT = 'aTenant'
TEST_TOKEN = 'aToken'
TEST_USER = 'test'
TEST_SERVICE_CATALOG = \
{'cdn':
[{'adminURL': 'http://cdn.admin-nets.local/v1.1/1234',
'region': 'RegionOne',
'internalURL': 'http://127.0.0.1:7777/v1.1/1234',
'publicURL': 'http://cdn.publicinternets.com/v1.1/1234'}],
'nova_compat':
[{'adminURL': 'http://127.0.0.1:8774/v1.0',
'region': 'RegionOne',
'internalURL': 'http://localhost:8774/v1.0',
'publicURL': 'http://nova.publicinternets.com/v1.0/'}],
'nova':
[{'adminURL': 'http://nova/novapi/admin',
'region':'RegionOne',
'internalURL': 'http://nova/novapi/internal',
'publicURL': 'http://nova/novapi/public'}],
'keystone':
[{'adminURL': 'http://127.0.0.1:8081/v2.0',
'region': 'RegionOne',
'internalURL': 'http://127.0.0.1:8080/v2.0',
'publicURL': 'http://keystone.publicinternets.com/v2.0'}],
'glance':
[{'adminURL': 'http://glance/glanceapi/admin',
'region':'RegionOne',
'internalURL': 'http://glance/glanceapi/internal',
'publicURL': 'http://glance/glanceapi/public'}],
'swift':
[{'adminURL': 'http://swift.admin-nets.local:8080/',
'region': 'RegionOne',
'internalURL': 'http://127.0.0.1:8080/v1/AUTH_1234',
'publicURL': 'http://swift.publicinternets.com/v1/AUTH_1234'}],
}
def setUp(self):
self.mox = mox.Mox()
self._real_get_user_from_request = keystone.get_user_from_request
self.setActiveUser(self.TEST_TOKEN, self.TEST_USER, self.TEST_TENANT,
True, self.TEST_SERVICE_CATALOG)
self.request = http.HttpRequest()
keystone.AuthenticationMiddleware().process_request(self.request)
def tearDown(self):
self.mox.UnsetStubs()
keystone.get_user_from_request = self._real_get_user_from_request
def setActiveUser(self, token, username,
tenant, is_admin, service_catalog):
keystone.get_user_from_request = \
lambda x: keystone.User(token, username, tenant,
is_admin, service_catalog)