def get_data(self): test_data = None api = ReachabilityTestAPI() with bsn_api.Session.begin(subtransactions=True): test = api.getQuickTest(self.request.user.project_id, bsn_api.Session) result = api.getLastReachabilityQuickTestResult( self.request.user.project_id, bsn_api.Session) test_data = ReachabilityTestData(test, result) return test_data
def handle(self, request, data): testargs = { 'tenant_id': request.user.project_id, 'src_tenant_id': data['tenant_source'].encode('ascii'), 'src_segment_id': data['segment_source'].encode('ascii'), 'src_ip': data['ip_source'].encode('ascii', 'ignore'), 'dst_ip': data['ip_destination'].encode('ascii', 'ignore'), 'expected_result': data['expected_connection'].encode('ascii', 'ignore') } api = reachability_test_api.ReachabilityTestAPI() with bsn_api.Session.begin(subtransactions=True): test = reachability_test_db.ReachabilityQuickTest(**testargs) api.addQuickTest(test, bsn_api.Session) api.runQuickTest(request.user.project_id, bsn_api.Session) test = api.getQuickTest(request.user.project_id, bsn_api.Session) messages.success(request, _('Successfully ran quick test.')) return test