from turbogears import (testutil, database) import turbogears import cherrypy from naked_planning.controllers import Root from naked_planning.controllers.root import Mmf as MmfController from naked_planning.model.plan import Mmf as MmfModel from naked_planning import model from naked_planning.tests import util cherrypy.root = MmfController() class MmfTest(util.DBTest): model = model root = cherrypy.root def test_can_add_mmfs(self): testutil.createRequest(turbogears.url("do_add", name='abc')) assert MmfModel.query().count() == 1 testutil.createRequest(turbogears.url("do_add", name='xyz')) assert MmfModel.query().count() == 2 def test_cant_add_mmfs_with_same_name(self): testutil.createRequest(turbogears.url("do_add", name='abc')) testutil.createRequest(turbogears.url("do_add", name='abc')) assert "500 Internal error" in cherrypy.response.body[0] print cherrypy.response.status assert cherrypy.response.status == "500 Internal error" def test_cant_add_mmf_with_no_name(self): testutil.createRequest(turbogears.url("do_add", name='')) assert "Please enter a value" in cherrypy.response.body[0]