def test_company_delete(self):
company_route = '/_admin/company'
url = self.random()
Company(name = self.random(), url = url, description = 'desc').put()
company_data = dict(action='delete')
self.admin_app.post(company_route+'/'+url, company_data)
self.assertFalse(Company.get_by_url(url))
def update(self, company_url= None):
company = Company.get_by_url(company_url)
if company:
company.name = self.read('name')
company.url = self.read('url')
company.description = self.read('description')
company.put()
self.redirect('/_admin/company')
def get(self, company_url, show_url):
company = Company.get_by_url(company_url)
if company: show = company.get_show_by_url(show_url)
else:
self.redirect('/', False)
return
if not show:
self.redirect('/' + company_url, False)
return
performances = show.get_performances().fetch(50)
self.render('public/show.html', dict(show=show, performances=performances, auth = utils.authdetails(show.get_route())))
def test_company_creation(self):
company_route = '/_admin/company'
self.admin_app.post(company_route, {'action':'create'}, status=403) # tells blank posts to bugger off
name = self.random()
url = self.random()
company_data = dict(action='create', name=name, url=url)
self.admin_app.post(company_route, company_data)
self.admin_app.post(company_route, company_data, status=403) # create the city only once
company = Company.get_by_url(url)
self.assertTrue(company)
self.assertEqual(name, company.name)
self.assertEqual(url, company.url)
def test_company_editing(self):
company_route = '/_admin/company'
url = self.random()
Company(name = self.random(), url = url, description = 'desc').put()
new_name = self.random()
new_url = self.random()
new_desc = self.random()
company_data = dict(action='update', name=new_name, url=new_url, description = new_desc)
self.admin_app.post(company_route+'/'+url, company_data)
company = Company.get_by_url(new_url)
self.assertEqual(new_name, company.name)
self.assertEqual(new_url, company.url)
self.assertEqual(new_desc,company.description)
def get(self, url):
city = City.get_by_url(url)
if not city: company = Company.get_by_url(url)
if not city and not company:
self.redirect('/', False)
return
data_type = 'city' if city else 'company'
data = city or company
start_after = self.read('start_after')
performances = data.get_performances(start_after = start_after).fetch(PAGE+1)
start_after = performances[-2].time_sort if len(performances) == PAGE+1 else None
render_data = dict(performances=performances[:PAGE],
start_after = start_after,
auth = utils.authdetails(data.get_route()))
render_data[data_type] = data
self.render('public/' + data_type + '.html', render_data)
def check_company(self):
company = Company.get_by_url(self.read('company'))
if not company: raise Exception("That company doesn't exist.")
return company
def get(self, company_url=None):
company = Company.get_by_url(company_url)
self.render('admin/company.html', dict(companies=Company.all(), current_company = company))
def delete(self, company_url=None):
company = Company.get_by_url(company_url)
if company: company.delete()
self.get()
