Commit 3042a1f2 authored by Vincent Pelletier's avatar Vincent Pelletier

Use a real node manager.

parent bfd3fe2f
...@@ -17,6 +17,7 @@ ...@@ -17,6 +17,7 @@
import unittest import unittest
from mock import Mock from mock import Mock
from .. import NeoUnitTestBase from .. import NeoUnitTestBase
from neo.lib.node import NodeManager
from neo.lib.pt import PartitionTable from neo.lib.pt import PartitionTable
from neo.lib.protocol import NodeTypes, NodeStates from neo.lib.protocol import NodeTypes, NodeStates
from neo.client.handlers.master import PrimaryBootstrapHandler from neo.client.handlers.master import PrimaryBootstrapHandler
...@@ -29,7 +30,7 @@ class MasterHandlerTests(NeoUnitTestBase): ...@@ -29,7 +30,7 @@ class MasterHandlerTests(NeoUnitTestBase):
super(MasterHandlerTests, self).setUp() super(MasterHandlerTests, self).setUp()
self.db = Mock() self.db = Mock()
self.app = Mock({'getDB': self.db}) self.app = Mock({'getDB': self.db})
self.app.nm = Mock() self.app.nm = NodeManager()
self.app.dispatcher = Mock() self.app.dispatcher = Mock()
class MasterBootstrapHandlerTests(MasterHandlerTests): class MasterBootstrapHandlerTests(MasterHandlerTests):
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment