def test_house_cannot_have_more_thant_50_students(self): h = House(1, 'Griffindor', 51) teaches_dark_arts = 0 repo = BaseHouseRepository() service = HouseService(repo) create_house = service.create_house(h.name, h.students, teaches_dark_arts) self.assertFalse(create_house)
def test_can_create_house_with_right_parameters(self): h = House(1, 'Griffindor', 50) teaches_dark_arts = 0 repo = BaseHouseRepository() service = HouseService(repo) create_house = service.create_house(h.name, h.students, teaches_dark_arts) self.assertTrue(create_house)
def test_cannot_create_house_if_teaches_dark_arts(self): h = House(1, 'Griffindor', 50) teaches_dark_arts = 1 repo = BaseHouseRepository() service = HouseService(repo) create_house = service.create_house(h.name, h.students, teaches_dark_arts) self.assertFalse(create_house)
def delete_house(): service = HouseService(HouseRepository()) houses = service.get_all_houses() if request.method == 'POST': service.delete_house(request.form['houses_id']) return render_template('house/houses.html', houses=houses) return render_template('house/delete_house.html', houses=houses)
def create_house(): if request.method == 'POST': name = request.form['house_name'] max_students = request.form['max_students'] teaches_dark_arts = request.form['teaches_dark_arts'] service = HouseService(HouseRepository()) success = service.create_house(name, max_students, teaches_dark_arts) if success: return render_template('house/house_success.html') return render_template('house/house_error.html') return render_template('house/new_house.html')
def update_house(house_id): service = HouseService(HouseRepository()) if request.method == 'POST': can_update_house = service.update_house( request.form['house_id'], request.form['house_name'], request.form['max_students'] ) if not can_update_house: return render_template('house/house_error.html') house = service.get_house_by_id(house_id) if house is None: return 'Not Found' return render_template('house/update_house.html', house=house)
def create_wizard(): h_service = HouseService(HouseRepository()) houses = h_service.get_all_houses() if request.method == 'POST': service = WizardService(WizardRepository()) new_wizard = service.create_wizard( request.form['wizard_name'], request.form['wizard_age'], request.form['has_received_letter'], request.form['voldemort_friend'], request.form['dark_wizard'], request.form['house_options'], ) if new_wizard: return render_template('wizard/wizard_success.html') return render_template('wizard/wizard_error.html') return render_template('wizard/new_wizard.html', houses=houses)
def update_wizard(wizard_id): w_service = WizardService(WizardRepository()) h_service = HouseService(HouseRepository()) houses = h_service.get_all_houses() if request.method == 'POST': can_update_wizard = w_service.update_wizard( request.form['wizard_id'], request.form['wizard_name'], request.form['wizard_age'], request.form['wizard_house']) if not can_update_wizard: return render_template('wizard/wizard_error.html') wizard = w_service.get_wizard_by_id(wizard_id) current_house = h_service.get_house_by_id(wizard.house_id) if wizard is not None: return render_template('wizard/update_wizard.html', wizard=wizard, houses=houses, current_house=current_house) else: return render_template('wizard/wizard_not_found.html')
def houses(): service = HouseService(HouseRepository()) houses = service.get_all_houses() return render_template('house/houses.html', houses=houses)
def house_by_id(house_id): service = HouseService(HouseRepository()) house = service.get_house_by_id(house_id) return render_template('house/house.html', house=house) if house else render_template('house/house_not_found.html')