Exemplo n.º 1
0
    def test_dict_to_protobuf(self):
        pb = bess_msg.CreatePortResponse()
        pb.error.err = 1
        pb.error.errmsg = 'bar'
        pb.error.details = ''
        pb.name = 'foo'

        msg_dict = {
            'error': {
                'err': 1,
                'errmsg': 'bar',
            },
            'name': 'foo'
        }

        msg = proto_conv.dict_to_protobuf(msg_dict,
                                          bess_msg.CreatePortResponse)
        self.assertEqual(msg, pb)

        pb = bess_msg.CreateModuleRequest()
        pb.name = 'm1'
        pb.mclass = 'bpf'

        kv = {
            'name': 'm1',
            'mclass': 'bpf',
        }
        msg = proto_conv.dict_to_protobuf(kv, bess_msg.CreateModuleRequest)
        self.assertEqual(msg, pb)
Exemplo n.º 2
0
Arquivo: bess.py Projeto: vivi/bess
    def create_module(self, mclass, name=None, arg=None):
        arg = arg or {}

        request = bess_msg.CreateModuleRequest()
        request.name = name or ''
        request.mclass = mclass

        message_type = getattr(module_msg, mclass + 'Arg', bess_msg.EmptyArg)
        arg_msg = pb_conv.dict_to_protobuf(message_type, arg)
        request.arg.Pack(arg_msg)

        return self._request('CreateModule', request)