test_setup_for_attract.py
b7cad397a824master

test_setup_for_attract.py
View Options

			import pillar.tests.common_test_data as ctd
			from abstract_attract_test import AbstractAttractTest, MOCK_SVN_URL


			class TaskWorkflowTest(AbstractAttractTest):
			def setUp(self, **kwargs):
			AbstractAttractTest.setUp(self, **kwargs)
			self.create_user(user_id=ctd.EXAMPLE_PROJECT_OWNER_ID,
			groups=[ctd.EXAMPLE_ADMIN_GROUP_ID])
			self.create_valid_auth_token(ctd.EXAMPLE_PROJECT_OWNER_ID, 'token')
			self.project_id, _ = self.ensure_project_exists()

			def test_custom_properties(self):
			"""Projects should get their properties dict."""

			with self.app.test_request_context():
			proj_coll = self.app.data.driver.db['projects']
			project = proj_coll.find_one({'_id': self.project_id})
			aprops = project['extension_props']['attract']
			self.assertIsInstance(aprops, dict)
			self.assertEqual(MOCK_SVN_URL, aprops['svn_url'])

			def test_saving_api(self):
			"""Ensures that Eve accepts an Attract project as valid."""

			import pillar.api.utils

			url = '/api/projects/%s' % self.project_id
			resp = self.get(url)
			proj = resp.json()

			put_proj = pillar.api.utils.remove_private_keys(proj)

			self.put(url,
			json=put_proj,
			auth_token='token',
			headers={'If-Match': proj['_etag']})