Python test_common.lineno函数代码示例

 def check_bgp_asn(self, fq_name, asn):
     router = self._vnc_lib.bgp_router_read(fq_name)
     params = router.get_bgp_router_parameters()
     if not params:
         print "retrying ... ", test_common.lineno()
         raise Exception('bgp params is None for %s' % fq_name)
     self.assertEqual(params.get_autonomous_system(), asn)

    def test_policy_in_policy(self):
        vn1_name = 'vn1'
        vn2_name = 'vn2'
        vn3_name = 'vn3'
        vn1_obj = VirtualNetwork(vn1_name)
        vn2_obj = VirtualNetwork(vn2_name)

        np1 = self.create_network_policy(vn1_obj, vn2_obj)
        np2 = self.create_network_policy(vn2_obj, vn1_obj)

        np1.network_policy_entries.policy_rule[0].dst_addresses[0].virtual_network = None
        np1.network_policy_entries.policy_rule[0].dst_addresses[0].network_policy = np2.get_fq_name_str()
        np1.set_network_policy_entries(np1.network_policy_entries)
        self._vnc_lib.network_policy_update(np1)

        np2.network_policy_entries.policy_rule[0].src_addresses[0].virtual_network = 'local'
        np2.set_network_policy_entries(np1.network_policy_entries)
        self._vnc_lib.network_policy_update(np2)
        seq = SequenceType(1, 1)
        vnp = VirtualNetworkPolicyType(seq)
        vn1_obj.set_network_policy(np1, vnp)
        vn2_obj.set_network_policy(np2, vnp)
        vn1_uuid = self._vnc_lib.virtual_network_create(vn1_obj)
        vn2_uuid = self._vnc_lib.virtual_network_create(vn2_obj)

        try:
            self.check_ri_state_vn_policy(fq_name=[u'default-domain', u'default-project', 'vn1', 'vn1'],
                                       to_fq_name=[u'default-domain', u'default-project', u'vn2', u'vn2'])
        except NoIdError, e:
            print "failed : routing instance state is not correct... ", test_common.lineno()
            self.assertTrue(False)

 def check_lr_asn(self, fq_name, rt_target):
     router = self._vnc_lib.logical_router_read(fq_name)
     rt_refs = router.get_route_target_refs()
     if not rt_refs:
         print "retrying ... ", test_common.lineno()
         raise Exception('ri_refs is None for %s' % fq_name)
     self.assertEqual(rt_refs[0]['to'][0], rt_target)

    def test_basic_policy(self):
        vn1_name = 'vn1'
        vn2_name = 'vn2'
        vn1_obj = VirtualNetwork(vn1_name)
        vn2_obj = VirtualNetwork(vn2_name)

        np = self.create_network_policy(vn1_obj, vn2_obj)
        seq = SequenceType(1, 1)
        vnp = VirtualNetworkPolicyType(seq)
        vn1_obj.set_network_policy(np, vnp)
        vn2_obj.set_network_policy(np, vnp)
        vn1_uuid = self._vnc_lib.virtual_network_create(vn1_obj)
        vn2_uuid = self._vnc_lib.virtual_network_create(vn2_obj)

        #import pdb; pdb.set_trace()
        for obj in [vn1_obj, vn2_obj]:
            ident_name = self.get_obj_imid(obj)
            gevent.sleep(2)
            ifmap_ident = self.assertThat(FakeIfmapClient._graph, Contains(ident_name))

        try:
            self.check_ri_state_vn_policy(fq_name=[u'default-domain', u'default-project', 'vn1', 'vn1'],
                                       to_fq_name=[u'default-domain', u'default-project', u'vn2', u'vn2'])
        except NoIdError, e:
            print "failed : routing instance state is not correct... ", test_common.lineno()
            self.assertTrue(False)

 def check_lr_is_deleted(self, uuid):
     try:
         self._vnc_lib.logical_router_read(id=uuid)
         print "retrying ... ", test_common.lineno()
         raise Exception('logical router %s still exists' % uuid)
     except NoIdError:
         print 'lr deleted'

 def check_ri_is_deleted(self, fq_name):
     try:
         self._vnc_lib.routing_instance_read(fq_name)
         print "retrying ... ", test_common.lineno()
         raise Exception('routing instance %s still exists' % fq_name)
     except NoIdError:
         print 'ri deleted'

 def check_service_chain_prefix_match(self, fq_name, prefix):
     ri = self._vnc_lib.routing_instance_read(fq_name)
     sci = ri.get_service_chain_information()
     if sci is None:
         print "retrying ... ", test_common.lineno()
         raise Exception('Service chain info not found for %s' % fq_name)
     self.assertEqual(sci.prefix[0], prefix)

 def check_ri_state_vn_policy(self, fq_name, to_fq_name):
     ri = self._vnc_lib.routing_instance_read(fq_name)
     ri_refs = ri.get_routing_instance_refs()
     if not ri_refs:
         print "retrying ... ", test_common.lineno()
         raise Exception('ri_refs is None for %s' % fq_name)
     self.assertEqual(ri_refs[0]['to'], to_fq_name)

    def test_policy_with_cidr(self):
        vn1 = self.create_virtual_network("vn1", "10.1.1.0/24")
        vn2 = self.create_virtual_network("vn2", "10.2.1.0/24")
        rules = []
        rule1 = { "protocol": "icmp",
                  "direction": "<>",
                  "src-port": "any",
                  "src": {"type": "vn", "value": vn1},
                  "dst": {"type": "cidr", "value": "10.2.1.2/32"},
                  "dst-port": "any",
                  "action": "deny"
                 }
        rules.append(rule1)

        np = self.create_network_policy_with_multiple_rules(rules)
        seq = SequenceType(1, 1)
        vnp = VirtualNetworkPolicyType(seq)
        vn1.set_network_policy(np, vnp)
        self._vnc_lib.virtual_network_update(vn1)

        for obj in [vn1]:
            ident_name = self.get_obj_imid(obj)
            ifmap_ident = self.assertThat(FakeIfmapClient._graph, Contains(ident_name))

        try:
            self.check_vn_ri_state(fq_name=[u'default-domain', u'default-project', 'vn1', 'vn1'])

        except NoIdError, e:
            print "failed : Routing instance state is not correct... ", test_common.lineno()
            self.assertTrue(False)

 def check_vn_is_deleted(self, uuid):
     try:
         self._vnc_lib.virtual_network_read(id=uuid)
         print "retrying ... ", test_common.lineno()
         raise Exception('virtual network %s still exists' % uuid)
     except NoIdError:
         print 'vn deleted'

 def check_rt_is_deleted(self, name):
     try:
         rt_obj = self._vnc_lib.route_target_read(fq_name=[name])
         print "retrying ... ", test_common.lineno()
         raise Exception(
             'rt %s still exists: RI backrefs %s LR backrefs %s' % (
                 name, rt_obj.get_routing_instance_back_refs(),
                 rt_obj.get_logical_router_back_refs()))
     except NoIdError:
         print 'rt deleted'

 def check_ri_asn(self, fq_name, rt_target):
     ri = self._vnc_lib.routing_instance_read(fq_name)
     rt_refs = ri.get_route_target_refs()
     if not rt_refs:
         print "retrying ... ", test_common.lineno()
         raise Exception('ri_refs is None for %s' % fq_name)
     for rt_ref in rt_refs:
         if rt_ref['to'][0] == rt_target:
             return
     raise Exception('rt_target %s not found in ri %s' % (rt_target, fq_name))

    def test_vn_delete(self):
        vn = self.create_virtual_network("vn", "10.1.1.0/24")
        gevent.sleep(2)
        for obj in [vn]:
            ident_name = self.get_obj_imid(obj)
            ifmap_ident = self.assertThat(FakeIfmapClient._graph, Contains(ident_name))

        try:
            self.check_vn_ri_state(fq_name=[u'default-domain', u'default-project', 'vn', 'vn'])

        except NoIdError, e:
            print "failed : routing instance state is not created ... ", test_common.lineno()
            self.assertTrue(False)

    def check_ri_rt_state_vn_policy(self, fq_name, to_fq_name, expect_to_find):
        ri = self._vnc_lib.routing_instance_read(fq_name)
        rt_refs = ri.get_route_target_refs()
        if not rt_refs:
            print "retrying ... ", test_common.lineno()
            raise Exception('ri_refs is None for %s' % fq_name)

        found = False
        for rt_ref in rt_refs:
            rt_obj = self._vnc_lib.route_target_read(id=rt_ref['uuid'])
            ri_refs = rt_obj.get_routing_instance_back_refs()
            for ri_ref in ri_refs:
                if ri_ref['to'] == to_fq_name:
                    found = True
                    break
            if found == True:
                break
        self.assertTrue(found == expect_to_find)

    def test_multiple_policy(self):
        vn1_name = 'vn1'
        vn2_name = 'vn2'
        vn1_obj = VirtualNetwork(vn1_name)
        vn2_obj = VirtualNetwork(vn2_name)

        np1 = self.create_network_policy(vn1_obj, vn2_obj)
        np2 = self.create_network_policy(vn2_obj, vn1_obj)
        seq = SequenceType(1, 1)
        vnp = VirtualNetworkPolicyType(seq)
        vn1_obj.set_network_policy(np1, vnp)
        vn2_obj.set_network_policy(np2, vnp)
        vn1_uuid = self._vnc_lib.virtual_network_create(vn1_obj)
        vn2_uuid = self._vnc_lib.virtual_network_create(vn2_obj)

        try:
            self.check_ri_state_vn_policy(fq_name=[u'default-domain', u'default-project', 'vn1', 'vn1'],
                                       to_fq_name=[u'default-domain', u'default-project', u'vn2', u'vn2'])
        except NoIdError, e:
            print "failed : routing instance state is not correct... ", test_common.lineno()
            self.assertTrue(False)