Fix using non-existent mock method

Mock 1.1.0 throws an error when using assert* methods that don't
exist.

Change-Id: I3fff6df417a29882f1e8672c309edc6d4201cdc2
This commit is contained in:
Josh Gachnang
2015-07-10 10:03:25 -07:00
parent a0f7485460
commit d6afe62e94

View File

@@ -183,7 +183,7 @@ class TestBaseAgent(test_base.BaseTestCase):
listen_addr[1],
self.agent.api,
server_class=simple_server.WSGIServer)
wsgi_server.serve_forever.assert_called_once()
wsgi_server.serve_forever.assert_called_once_with()
self.agent.heartbeater.start.assert_called_once_with()
@@ -337,7 +337,7 @@ class TestAgentStandalone(test_base.BaseTestCase):
listen_addr[1],
self.agent.api,
server_class=simple_server.WSGIServer)
wsgi_server.serve_forever.assert_called_once()
wsgi_server.serve_forever.assert_called_once_with()
self.assertFalse(self.agent.heartbeater.called)
self.assertFalse(self.agent.api_client.lookup_node.called)