def test_annotate(self):
importer = tracker.ForgeTrackerImporter()
user = mock.Mock(_id=1)
self.assertEqual(importer.annotate('foo', user, 'bar'), 'foo')
user._id = None
self.assertEqual(importer.annotate('foo', user, 'bar'),
'*Originally by:* bar\n\nfoo')
def test_import_tool_failure(self, h, M, ThreadLocalORMSession):
M.session.artifact_orm_session._get.side_effect = ValueError
project = mock.Mock()
user = mock.Mock()
tracker_json = {
'tracker_config': {
'_id': 'orig_id',
'options': {}
},
'open_status_names': 'os',
'closed_status_names': 'cs',
}
importer = tracker.ForgeTrackerImporter()
importer._load_json = mock.Mock(return_value=tracker_json)
self.assertRaises(ValueError,
importer.import_tool,
project,
user,
project_name='project_name',
mount_point='mount_point',
mount_label='mount_label')
h.make_app_admin_only.assert_called_once_with(
project.install_app.return_value)
def test_process_comments(self, c, File):
importer = tracker.ForgeTrackerImporter()
author = mock.Mock()
importer.get_user = mock.Mock(return_value=author)
importer.annotate = mock.Mock(side_effect=['at1', 'at2'])
ticket = mock.Mock()
add_post = ticket.discussion_thread.add_post
ama = add_post.return_value.add_multiple_attachments
File.side_effect = ['f1', 'f2', 'f3', 'f4']
comments = [
{
'author': 'a1',
'text': 't1',
'timestamp': '2013-09-01',
'attachments': [{
'url': 'u1'
}, {
'url': 'u2'
}],
},
{
'author': 'a2',
'text': 't2',
'timestamp': '2013-09-02',
'attachments': [{
'url': 'u3'
}, {
'url': 'u4'
}],
},
]
importer.process_comments(ticket, comments)
self.assertEqual(importer.get_user.call_args_list,
[mock.call('a1'), mock.call('a2')])
self.assertEqual(importer.annotate.call_args_list, [
mock.call('t1', author, 'a1'),
mock.call('t2', author, 'a2'),
])
self.assertEqual(add_post.call_args_list, [
mock.call(text='at1',
ignore_security=True,
timestamp=datetime(2013, 9, 1)),
mock.call(text='at2',
ignore_security=True,
timestamp=datetime(2013, 9, 2)),
])
self.assertEqual(File.call_args_list, [
mock.call('u1'),
mock.call('u2'),
mock.call('u3'),
mock.call('u4'),
])
self.assertEqual(ama.call_args_list, [
mock.call(['f1', 'f2']),
mock.call(['f3', 'f4']),
])
def test_process_bins(self, TM):
app = mock.Mock()
app.config._id = 1
importer = tracker.ForgeTrackerImporter()
importer.process_bins(app, [{'_id': 1, 'b': 1}, {'b': 2}])
TM.Bin.query.remove.assert_called_once_with({'app_config_id': 1})
self.assertEqual(TM.Bin.call_args_list, [
mock.call(app_config_id=1, b=1),
mock.call(app_config_id=1, b=2),
])
def test_annotate(self):
importer = tracker.ForgeTrackerImporter()
user = mock.Mock(_id=1)
user.is_anonymous.return_value = False
self.assertEqual(importer.annotate('foo', user, 'bar'), 'foo')
user.is_anonymous.return_value = True
self.assertEqual(importer.annotate('foo', user, 'bar'),
'*Originally by:* bar\n\nfoo')
self.assertEqual(importer.annotate('foo', user, 'nobody'), 'foo')
self.assertEqual(importer.annotate('foo', user, None), 'foo')
def test_get_user(self, M):
importer = tracker.ForgeTrackerImporter()
M.User.anonymous.return_value = 'anon'
M.User.by_username.return_value = 'bar'
self.assertEqual(importer.get_user('foo'), 'bar')
self.assertEqual(M.User.anonymous.call_count, 0)
M.User.by_username.return_value = None
self.assertEqual(importer.get_user('foo'), 'anon')
self.assertEqual(M.User.anonymous.call_count, 1)
def test_import_tool(self, TM, M, session, tlos, c, g, mao):
importer = tracker.ForgeTrackerImporter()
importer._load_json = mock.Mock(
return_value={
'tracker_config': {
'_id': 'orig_id',
'options': {
'foo': 'bar',
},
},
'open_status_names':
'open statuses',
'closed_status_names':
'closed statuses',
'custom_fields':
'fields',
'saved_bins':
'bins',
'tickets': [
{
'reported_by': 'rb1',
'ticket_num': 1,
'description': 'd1',
'created_date': '2013-09-01',
'mod_date': '2013-09-02',
'summary': 's1',
'custom_fields': 'cf1',
'status': 'st1',
'labels': 'l1',
'votes_down': 1,
'votes_up': 2,
'private': False,
'discussion_thread': {
'posts': 'comments1'
},
},
{
'reported_by': 'rb2',
'ticket_num': 100,
'description': 'd2',
'created_date': '2013-09-03',
'mod_date': '2013-09-04',
'summary': 's2',
'custom_fields': 'cf2',
'status': 'st2',
'labels': 'l2',
'votes_down': 3,
'votes_up': 5,
'private': True,
'discussion_thread': {
'posts': 'comments2'
},
},
],
})
reporter = mock.Mock()
importer.get_user = mock.Mock(return_value=reporter)
importer.annotate = mock.Mock(side_effect=['ad1', 'ad2'])
importer.process_comments = mock.Mock()
importer.process_bins = mock.Mock()
project, user = mock.Mock(), mock.Mock()
app = project.install_app.return_value
app.config.options.mount_point = 'mount_point'
app.config.options.import_id = {
'source': 'Allura',
'app_config_id': 'orig_id',
}
app.config.options.get = lambda *a: getattr(app.config.options, *a)
app.url = 'foo'
tickets = TM.Ticket.side_effect = [mock.Mock(), mock.Mock()]
importer.import_tool(project,
user,
mount_point='mount_point',
mount_label='mount_label')
project.install_app.assert_called_once_with(
'tickets',
'mount_point',
'mount_label',
open_status_names='open statuses',
closed_status_names='closed statuses',
import_id={
'source': 'Allura',
'app_config_id': 'orig_id',
},
foo='bar',
)
self.assertEqual(importer.annotate.call_args_list, [
mock.call('d1', reporter, 'rb1'),
mock.call('d2', reporter, 'rb2'),
])
self.assertEqual(TM.Ticket.call_args_list, [
mock.call(
app_config_id=app.config._id,
import_id={
'source': 'Allura',
'app_config_id': 'orig_id',
'source_id': 1,
},
description='ad1',
created_date=datetime(2013, 9, 1),
mod_date=datetime(2013, 9, 2),
ticket_num=1,
summary='s1',
custom_fields='cf1',
status='st1',
labels='l1',
votes_down=1,
votes_up=2,
votes=1,
),
mock.call(
app_config_id=app.config._id,
import_id={
'source': 'Allura',
'app_config_id': 'orig_id',
'source_id': 100,
},
description='ad2',
created_date=datetime(2013, 9, 3),
mod_date=datetime(2013, 9, 4),
ticket_num=100,
summary='s2',
custom_fields='cf2',
status='st2',
labels='l2',
votes_down=3,
votes_up=5,
votes=2,
),
])
self.assertEqual(tickets[0].private, False)
self.assertEqual(tickets[1].private, True)
self.assertEqual(importer.process_comments.call_args_list, [
mock.call(tickets[0], 'comments1'),
mock.call(tickets[1], 'comments2'),
])
self.assertEqual(tlos.flush_all.call_args_list, [
mock.call(),
mock.call(),
])
self.assertEqual(session.return_value.flush.call_args_list, [
mock.call(tickets[0]),
mock.call(tickets[1]),
])
self.assertEqual(session.return_value.expunge.call_args_list, [
mock.call(tickets[0]),
mock.call(tickets[1]),
])
self.assertEqual(app.globals.custom_fields, 'fields')
importer.process_bins.assert_called_once_with(app, 'bins')
self.assertEqual(app.globals.last_ticket_num, 100)
M.AuditLog.log.assert_called_once_with(
'import tool mount_point from exported Allura JSON',
project=project,
user=user,
url='foo')
g.post_event.assert_called_once_with('project_updated')
app.globals.invalidate_bin_counts.assert_called_once_with()
def setUp(self):
"""Mount Allura importer on the Tracker admin controller"""
super(TestForgeTrackerImportController, self).setUp()
from forgetracker.tracker_main import TrackerAdminController
TrackerAdminController._importer = \
tracker.ForgeTrackerImportController(tracker.ForgeTrackerImporter())
