def test_get_tickets_for_all_milestones_custom_field(self):
tickets = get_tickets_for_all_milestones(self.env, field='project')
milestone1 = [{'id': 9, 'status': 'new', 'project': 'baz'},
{'id': 1, 'status': 'new', 'project': 'foo'},
{'id': 5, 'status': 'new', 'project': 'foo'}]
milestone2 = [{'id': 6, 'status': 'new', 'project': 'bar'},
{'id': 10, 'status': 'new', 'project': 'baz'},
{'id': 2, 'status': 'new', 'project': 'foo'}]
self.assertEqual(milestone1, tickets['milestone1'])
self.assertEqual(milestone1,
get_tickets_for_milestone(self.env,
milestone='milestone1',
field='project'))
self.assertEqual(milestone2, tickets['milestone2'])
self.assertEqual(milestone2,
get_tickets_for_milestone(self.env,
milestone='milestone2',
field='project'))
self.assertEqual(['milestone1', 'milestone2'], sorted(tickets))
def test_get_tickets_for_all_milestones(self):
tickets = get_tickets_for_all_milestones(self.env, field='owner')
milestone1 = [{'id': 9, 'status': 'new', 'owner': 'blah'},
{'id': 1, 'status': 'new', 'owner': 'joe'},
{'id': 5, 'status': 'new', 'owner': 'john'}]
milestone2 = [{'id': 6, 'status': 'new', 'owner': 'blah'},
{'id': 10, 'status': 'new', 'owner': 'blah'},
{'id': 2, 'status': 'new', 'owner': 'joe'}]
self.assertEqual(milestone1, tickets['milestone1'])
self.assertEqual(milestone1,
get_tickets_for_milestone(self.env,
milestone='milestone1',
field='owner'))
self.assertEqual(milestone2, tickets['milestone2'])
self.assertEqual(milestone2,
get_tickets_for_milestone(self.env,
milestone='milestone2',
field='owner'))
self.assertEqual(['milestone1', 'milestone2'], sorted(tickets))
