def test_get_sites(self): # Get Site objects array, [Site, Site, Site...] dao = SiteDAO() self._register_node(dao) sites = dao.get_sites() self.assertEqual(len(sites), 1) self.assertEqual(len(sites['site1'].nodes.keys()), 1)
def show_site(request): dao = SiteDAO() site = dao.get_site(request.matchdict['site']) if not site: msg = 'Unable to find site "{0}"'.format(request.matchdict['site']) raise HTTPNotFound(msg) return { 'site': site, 'site_json': dumps(site) }
def deploy_application(request): try: dao = SiteDAO() site = dao.get_site(request.POST['site']) app = site.applications[request.POST['application']] app.deploy_application(site) return dumps({ 'success': True, 'app': app }) except KeyError as e: msg = 'Unable to deploy application under "{0}"' msg = msg.format(request.POST['site'], request.POST['application']) return dumps({ 'success': False, 'msg': msg })
def tag_application(request): try: dao = SiteDAO() site = dao.get_site(request.POST['site']) app = site.applications[request.POST['application']] app.tag(request.POST['tag'], request.POST['msg']) return dumps({ 'success': True, 'app': app }) except KeyError as e: msg = 'Unable to tag site and application under "{0}" and "{1}"' msg = msg.format(request.POST['site'], request.POST['application']) return dumps({ 'success': False, 'msg': msg })
def test_all_site_keys(self): dao = SiteDAO() node = { 'name': 'node1', 'site': 'site1', 'url' : 'http://node1' } dao.register_node(node) keys = dao._all_site_keys() self.assertEqual(keys[0], 'site:site1')
def test_register_node(self): site = 'site1' node1 = { 'name': 'node1', 'site': site, 'url' : 'http://node1' } node2 = { 'name': 'node2', 'site': site, 'url' : 'http://node2' } dao = SiteDAO() dao.register_node(node1) dao.register_node(node2) self.assertEqual(len(dao.nodes(site).keys()), 2) node3 = { 'name': 'node3', 'site': site, 'url' : 'http://node3' } dao.register_node(node3) self.assertEqual(len(dao.nodes(site).keys()), 3)
def show_application(request): try: dao = SiteDAO() site = dao.get_site(request.matchdict['site']) app = site.applications[request.matchdict['application']] if take_note(request, app): app.add_note(request.POST['note']) except: if 'note' in request.POST: msg = 'Unable to add note to "{0}"' msg = msg.format(request.matchdict['application']) else: msg = 'Unable to find site and application under "{0}" and "{1}"' msg = msg.format(request.matchdict['site'], request.matchdict['application']) raise HTTPNotFound(msg) return { 'site': site, 'app': app }
def test_unregister_node(self): node1 = { 'name': 'node1', 'site': 'site1', 'url' : 'http://node1' } dao = SiteDAO() dao.register_node(node1) self.assertEqual(len(dao.nodes('site1').keys()), 1) dao.unregister_node(node1) self.assertEqual(len(dao.nodes('site1').keys()), 0) sites = dao.get_sites() self.assertEqual(len(sites.keys()), 0)
def show_sites(request): dao = SiteDAO() return { 'sites': dao.get_sites() }
def test_register_node_two(self): # Make sure we don't return an unknown site dao = SiteDAO() nodes = dao.nodes('unknown site') self.assertEqual(len(nodes), 0)