Fixed unittests for disabled AB-testing

This commit is contained in:
Sybren A. Stüvel 2016-06-29 16:43:41 +02:00
parent 1e404351b0
commit 3bb0e588d8

View File

@ -50,7 +50,7 @@ class HomeProjectTest(AbstractPillarTest):
resp = self.client.get(endpoint) resp = self.client.get(endpoint)
# While we're still AB-testing, unauthenticated users should get a 404. # While we're still AB-testing, unauthenticated users should get a 404.
# When that's over, it should result in a 403. # When that's over, it should result in a 403.
self.assertEqual(404, resp.status_code) self.assertEqual(403, resp.status_code)
resp = self.client.get(endpoint, headers={'Authorization': self.make_header('token')}) resp = self.client.get(endpoint, headers={'Authorization': self.make_header('token')})
self.assertEqual(200, resp.status_code) self.assertEqual(200, resp.status_code)
@ -87,27 +87,27 @@ class HomeProjectTest(AbstractPillarTest):
}) })
self.assertIsNotNone(node) self.assertIsNotNone(node)
@responses.activate # @responses.activate
def test_home_project_ab_testing(self): # def test_home_project_ab_testing(self):
self.mock_blenderid_validate_happy() # self.mock_blenderid_validate_happy()
resp = self.client.get('/users/me', headers={'Authorization': self.make_header('token')}) # resp = self.client.get('/users/me', headers={'Authorization': self.make_header('token')})
self.assertEqual(200, resp.status_code, resp) # self.assertEqual(200, resp.status_code, resp)
#
# Grant subscriber and but NOT homeproject role, and fetch the home project. # # Grant subscriber and but NOT homeproject role, and fetch the home project.
self.badger(TEST_EMAIL_ADDRESS, {'subscriber'}, 'grant') # self.badger(TEST_EMAIL_ADDRESS, {'subscriber'}, 'grant')
#
resp = self.client.get('/bcloud/home-project', # resp = self.client.get('/bcloud/home-project',
headers={'Authorization': self.make_header('token')}) # headers={'Authorization': self.make_header('token')})
self.assertEqual(404, resp.status_code) # self.assertEqual(404, resp.status_code)
#
resp = self.client.get('/users/me', # resp = self.client.get('/users/me',
headers={'Authorization': self.make_header('token')}) # headers={'Authorization': self.make_header('token')})
self.assertEqual(200, resp.status_code) # self.assertEqual(200, resp.status_code)
me = json.loads(resp.data) # me = json.loads(resp.data)
#
with self.app.test_request_context(): # with self.app.test_request_context():
from application.modules.blender_cloud import home_project # from application.modules.blender_cloud import home_project
self.assertFalse(home_project.has_home_project(me['_id'])) # self.assertFalse(home_project.has_home_project(me['_id']))
@responses.activate @responses.activate
def test_autocreate_home_project_with_demo_role(self): def test_autocreate_home_project_with_demo_role(self):