grpc added initial tests and fixture to support them
This commit is contained in:
parent
0c0ff95fb6
commit
3498a59ed5
4 changed files with 74 additions and 9 deletions
|
@ -3,6 +3,7 @@ Unit test fixture module.
|
|||
"""
|
||||
|
||||
import os
|
||||
import threading
|
||||
|
||||
import pytest
|
||||
from mock.mock import MagicMock
|
||||
|
@ -26,6 +27,8 @@ from core.enumerations import LinkTypes
|
|||
from core.enumerations import MessageFlags
|
||||
from core.enumerations import NodeTlvs
|
||||
from core.enumerations import NodeTypes
|
||||
from core.grpc.client import CoreGrpcClient
|
||||
from core.grpc.server import CoreGrpcServer
|
||||
from core.misc import ipaddress
|
||||
from core.misc.ipaddress import MacAddress
|
||||
from core.service import ServiceManager
|
||||
|
@ -204,6 +207,18 @@ class CoreServerTest(object):
|
|||
self.server.server_close()
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def grpc_server():
|
||||
coremu = CoreEmu()
|
||||
grpc_server = CoreGrpcServer(coremu)
|
||||
thread = threading.Thread(target=grpc_server.listen)
|
||||
thread.daemon = True
|
||||
thread.start()
|
||||
yield grpc_server
|
||||
coremu.shutdown()
|
||||
grpc_server.server.stop(None)
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def session():
|
||||
# use coreemu and create a session
|
||||
|
@ -242,8 +257,6 @@ def cored():
|
|||
# cleanup
|
||||
server.shutdown()
|
||||
|
||||
#
|
||||
|
||||
# cleanup services
|
||||
ServiceManager.services.clear()
|
||||
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue