openstack/ironic-python-agent
Code Issues Proposed changes

Merge pull request #59 from rackerlabs/jay/FixLeakyHeartbeatTest

Browse Source 
fix(test): Properly mock hwmgr for heartbeat test
This commit is contained in:
Jason Faulkner 2014-03-05 15:41:23 -08:00
commit 8b60c1c8b4
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
teeth_agent/tests/agent.py
View File

@ -26,7 +26,7 @@ from teeth_rest import encoding
from teeth_agent import agent from teeth_agent import agent
from teeth_agent import base from teeth_agent import base
from teeth_agent import errors from teeth_agent import errors
from teeth_agent import hardware
EXPECTED_ERROR = RuntimeError('command execution failed') EXPECTED_ERROR = RuntimeError('command execution failed')
@ -48,6 +48,8 @@ class TestHeartbeater(unittest.TestCase):
self.mock_agent = mock.Mock() self.mock_agent = mock.Mock()
self.heartbeater = agent.TeethAgentHeartbeater(self.mock_agent) self.heartbeater = agent.TeethAgentHeartbeater(self.mock_agent)
self.heartbeater.api = mock.Mock() self.heartbeater.api = mock.Mock()
self.heartbeater.hardware = mock.create_autospec(
hardware.HardwareManager)
self.heartbeater.stop_event = mock.Mock() self.heartbeater.stop_event = mock.Mock()
@mock.patch('time.time') @mock.patch('time.time')