Sybren A. Stüvel
390d687f61
This is mostly useful for the CLI interface, as the majority of our Pillar code actually needs more project information than just the ID.
30 lines
892 B
Python
30 lines
892 B
Python
# -*- encoding: utf-8 -*-
|
|
|
|
"""Unit tests for pillar.api.project.utils."""
|
|
|
|
import logging
|
|
|
|
from bson import ObjectId
|
|
from pillar.tests import AbstractPillarTest
|
|
|
|
log = logging.getLogger(__name__)
|
|
|
|
|
|
class ProjectUtilsTest(AbstractPillarTest):
|
|
def test_project_id_from_url(self):
|
|
self.enter_app_context()
|
|
|
|
self.ensure_project_exists({'_id': ObjectId(24 * 'a'), 'url': 'project1'})
|
|
self.ensure_project_exists({'_id': ObjectId(24 * 'b'), 'url': 'project2'})
|
|
self.ensure_project_exists({'_id': ObjectId(24 * 'c'), 'url': 'project3'})
|
|
|
|
from pillar.api.projects.utils import project_id
|
|
|
|
pid1 = project_id('project1')
|
|
pid2 = project_id('project2')
|
|
pid3 = project_id('project3')
|
|
|
|
self.assertEqual(ObjectId(24 * 'a'), pid1)
|
|
self.assertEqual(ObjectId(24 * 'b'), pid2)
|
|
self.assertEqual(ObjectId(24 * 'c'), pid3)
|