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

Use a real node manager.

parent bfd3fe2f
......@@ -17,6 +17,7 @@
import unittest
from mock import Mock
from .. import NeoUnitTestBase
from neo.lib.node import NodeManager
from neo.lib.pt import PartitionTable
from neo.lib.protocol import NodeTypes, NodeStates
from neo.client.handlers.master import PrimaryBootstrapHandler
......@@ -29,7 +30,7 @@ class MasterHandlerTests(NeoUnitTestBase):
super(MasterHandlerTests, self).setUp()
self.db = Mock()
self.app = Mock({'getDB': self.db})
self.app.nm = Mock()
self.app.nm = NodeManager()
self.app.dispatcher = Mock()
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