added grpc for a node command and fixed grpc unit tests

This commit is contained in:
Blake Harnden 2019-05-31 11:46:47 -07:00
parent e063fcd4fe
commit 1890db5991
5 changed files with 52 additions and 2 deletions

View file

@ -1,5 +1,4 @@
import time
from Queue import Queue
import grpc
@ -8,6 +7,7 @@ import pytest
from core.conf import ConfigShim
from core.data import EventData
from core.emane.ieee80211abg import EmaneIeee80211abgModel
from core.emulator.emudata import NodeOptions
from core.enumerations import NodeTypes, EventTypes, ConfigFlags, ExceptionLevels
from core.grpc import core_pb2
from core.grpc.client import CoreGrpcClient
@ -239,6 +239,24 @@ class TestGrpc:
with pytest.raises(KeyError):
assert session.get_object(node.objid)
def test_node_command(self, grpc_server):
# given
client = CoreGrpcClient()
session = grpc_server.coreemu.create_session()
session.set_state(EventTypes.CONFIGURATION_STATE)
node_options = NodeOptions(model="Host")
node = session.add_node(node_options=node_options)
session.instantiate()
output = "hello world"
# then
command = "echo %s" % output
with client.context_connect():
response = client.node_command(session.id, node.objid, command)
# then
assert response.output == output
def test_get_hooks(self, grpc_server):
# given
client = CoreGrpcClient()