def testSetup(self): test_names = ["Tom", "Dick", "Harry"] GhostrEngine.setup(test_names) stored_ghosts = Ghost.query() for ghost in stored_ghosts: assert ghost.ghost_name in test_names
import webapp2 from google.appengine.api import users from src.ghostr_engine import GhostrEngine from google.appengine.ext.webapp import template import os import sys from google.appengine.api import urlfetch sys.path.append(os.getcwd()) GhostrEngine.setup() class MainPage(webapp2.RequestHandler): def get(self): user = users.get_current_user() template_values = { 'ghosts': GhostrEngine.list_all(), 'user': user, 'form_linktext': GhostrEngine.set_form_text(user), } path = os.path.join(os.path.dirname(__file__), 'templates/index.html') self.response.out.write(template.render(path, template_values)) class EntryPage(webapp2.RequestHandler): def get(self, template_values=dict()): if users.get_current_user(): auth_url = users.create_logout_url(self.request.uri) auth_linktext = 'Logout'