Files
attract/tests/test_shortcode.py

41 lines
1.4 KiB
Python
Raw Permalink Normal View History

import bson
2016-11-01 12:33:03 +01:00
from abstract_attract_test import AbstractAttractTest
SVN_SERVER_URL = 'svn://biserver/agent327'
class ShortcodeTest(AbstractAttractTest):
def setUp(self, **kwargs):
AbstractAttractTest.setUp(self, **kwargs)
self.mngr = self.attract.task_manager
2016-11-01 12:33:03 +01:00
self.proj_id, self.project = self.ensure_project_exists()
def test_increment_simple(self):
from attract import shortcodes
with self.app.test_request_context():
2017-03-03 15:08:48 +01:00
code = shortcodes.generate_shortcode(self.proj_id, 'jemoeder', 'ø')
self.assertEqual('ø1', code)
2016-11-01 12:33:03 +01:00
with self.app.test_request_context():
2017-03-03 15:08:48 +01:00
code = shortcodes.generate_shortcode(self.proj_id, 'jemoeder', 'č')
self.assertEqual('č2', code)
def test_multiple_projects(self):
from attract import shortcodes
proj_id2, project2 = self.ensure_project_exists(project_overrides={
'_id': bson.ObjectId(24 * 'f'),
'url': 'proj2',
})
with self.app.app_context():
code1 = shortcodes.generate_shortcode(self.proj_id, 'jemoeder', 'č')
code2 = shortcodes.generate_shortcode(proj_id2, 'jemoeder', 'č')
code3 = shortcodes.generate_shortcode(proj_id2, 'jemoeder', 'č')
self.assertEqual('č1', code1)
self.assertEqual('č1', code2)
self.assertEqual('č2', code3)