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']})