def setUp(self): super(CollectionBaseTest, self).setUp() self.assertEqual(set(Boat.collection()), set()) self.boat1 = Boat(name="Pen Duick I", length=15.1, launched=1898) self.boat2 = Boat(name="Pen Duick II", length=13.6, launched=1964) self.boat3 = Boat(name="Pen Duick III", length=17.45, launched=1966) self.boat4 = Boat(name="Rainbow Warrior I", power="engine", length=40, launched=1955)
def addBoat(boat_input): # Get or create user user = getOrAddUser(boat_input) # Database session = dbconnect() boat = Boat() boat.name = boat_input["boat"]["name"] boat.started = boat_input["boat"]["started"] boat.boatType = boat_input["boat"]["boatType"] boat.desiredCourse = boat_input["boat"]["desiredCourse"] boat.isActive = boat_input["boat"]["isActive"] boat.boatFlags = boat_input["boat"]["boatFlags"] boat.race = addBoatRace(session, boat_input["boat"]["race"]) boat.user = user session.add(boat) session.commit() session.flush() # Core addBoatinCore(boat.name, boat_input["boat"]["race"]["startLat"], boat_input["boat"]["race"]["startLon"], boat.boatType, boat.boatFlags, boat.id) return boat.id