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:
@@ -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)
|
||||
|
Reference in New Issue
Block a user