from urllib import urlencode
from mock import Mock, patch
from tg import config
from nose.tools import assert_true
from ming.orm.base import session
from pyforge.lib import helpers as h
from pyforge.model import User
from pylons import c
from forgetracker.tests import TestController
from forgetracker.model import Ticket, Globals
class TestRootController(TestController):
def test_index(self):
response = self.app.get('/tickets/')
assert_true('ForgeTracker for' in response)
class TestWhenSearchingWithCustomFields(TestRootController):
def setUp(self):
super(TestRootController, self).setUp()
c.user = User.query.get(username='test_user')
h.set_context('test', 'bugs')
def test_that_fields_are_in_search_results(self):
response = self._search_results()
assert response.html.find(
lambda tag: (tag.name == 'td'
and 'Iteration 1' in str(tag)))
def test_that_field_names_are_in_table_head(self):
response = self._search_results()
assert response.html.find(
lambda tag: (tag.name == 'th'
and 'Iteration Number' in str(tag)))
@patch('forgetracker.tracker_main.search_artifact')
def _search_results(self, search_artifact):
set_tracker_custom_fields([dict(name='iteration_number',
label='Iteration Number')])
ticket = create_ticket(summary="colors are wrong",
custom_fields=dict(iteration_number='Iteration 1'))
make_search_return_ticket(search_artifact, ticket)
response = self.app.get('/bugs/search/?q=friends')
return response
def set_tracker_custom_fields(custom_fields):
tracker_globals = Globals.for_current_tracker()
tracker_globals.custom_fields = custom_fields
session(tracker_globals).flush()
def create_ticket(summary, custom_fields):
ticket = Ticket(app_config_id=c.app.config._id,
ticket_num=1,
summary=summary,
custom_fields=custom_fields)
session(ticket).flush()
return ticket
def make_search_return_ticket(search_artifact, ticket):
matches = Mock()
matches.docs = [dict(ticket_num_i=ticket.ticket_num)]
search_artifact.return_value = matches