Ticket #701: test_catwalk_701.py

import unittest
from turbogears import testutil
from turbogears.toolbox.catwalk import CatWalk

from sqlobject import *
import cherrypy

# creating a fake model
sqlhub.processConnection = connectionForURI('sqlite:/:memory:')
class PetOwner(SQLObject):
    name = StringCol()
    pets = MultipleJoin('Pet')

class Pet(SQLObject):
    name = StringCol()
    # Note that SQLObject 0.7.x uses the MixedCaseStyle.
    # (http://www.sqlobject.org/SQLObject.html#changing-the-naming-style)
    petOwner = ForeignKey('PetOwner', default=None)


# Setting up the testing environment
import new
model = new.module('model')
model.PetOwner = PetOwner
model.Pet = Pet
cherrypy.root = CatWalk(model)

# testcase for ticket 701
class TestTicket701(testutil.DBTest):
    model=model

    def test_model(self):
        assert model.PetOwner.select().count() == 0
        model.PetOwner(name='Tim')
        assert model.PetOwner.select().count() == 1
        assert model.PetOwner.select()[0].name == 'Tim'
        assert model.PetOwner.select()[0].pets == []
        
    def test_ticket_701(self):
        # /browse works when there are no SecLevel instances
        assert model.PetOwner.select().count() == 0
        testutil.createRequest("/browse/?object_name=PetOwner")
        assert cherrypy.response.status == '200 OK', cherrypy.response.status
        # so we create one
        testutil.createRequest("/add?objectName=PetOwner&name=Tim")
        assert cherrypy.response.status == '200 OK', cherrypy.response.status
        assert model.PetOwner.select().count() == 1
        # request /browse wille now bomb out
        testutil.createRequest("/browse/?object_name=PetOwner")
        assert cherrypy.response.status == '200 OK', cherrypy.response.status