def get_cdns(context, request): page = request.params.get('page', 0) num_per_page = request.params.get('num_per_page', 5) if num_per_page <= 0: result = CDN.select() else: result = CDN.select().paginate(page, num_per_page) cdns = [serialize_cdn(cdn) for cdn in result] request.reply(cdns)
def cdn_list(self): """ List CDNs registered with CacheBrowser """ cdns = CDN.select() for cdn in cdns: self.send_line("%*s: %s" % (15, cdn.id, ' '.join(cdn.addresses)))
def test_database(self): initialize_database(":memory:", reset=True) cdn1 = CDN.create(id="cdn1", name="Sample CDN", edge_server="1.2.3.4") cdn2 = CDN.create(id="cdn2", name="Sample CDN2", edge_server="1.2.3.5") host1 = Host.create(hostname="first.host", cdn=cdn1) host2 = Host.create(hostname="second.host", cdn=cdn2) host3 = Host.create(hostname="third.host", cdn=cdn1) self.assertEqual([host1, host2, host3], list(Host.select())) self.assertEqual([host1, host3], Host.select().where(Host.cdn == cdn1)) self.assertEqual([host1, host3], Host.select().join(CDN).where(CDN.id == 'cdn1')) self.assertEqual(host2, Host.get(Host.cdn == cdn2)) self.assertEqual([], CDN.select().where(CDN.id == "doesntexist")) with self.assertRaises(DoesNotExist): CDN.get(CDN.id == "doesntexist")
def test_bootstrap(self): self.bootstrapper.bootstrap('the.host') self.assertTrue(Host.select().where(Host.hostname == 'the.host').exists()) self.assertTrue(CDN.select().where(CDN.id == 'thecdn').exists())