From d37d27b2d6fd09b4b8ca111057f894a55725c579 Mon Sep 17 00:00:00 2001 From: Tang Chen Date: Wed, 2 Dec 2015 19:27:06 +0800 Subject: [PATCH] Enable setup_servers_mock() to take attributes param. setup_servers_mock() is now able to set methods to the fake servers. But it cannot set attributes. This patch enable it to do so. This will be useful in "server list" test cases. Change-Id: Ic30d750ebe4650244707a368cdd5d622a8f1b8ed Implements: blueprint osc-unit-test-framework-improvement --- openstackclient/tests/compute/v2/test_server.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/openstackclient/tests/compute/v2/test_server.py b/openstackclient/tests/compute/v2/test_server.py index a750ed5dde..6cc314e89c 100644 --- a/openstackclient/tests/compute/v2/test_server.py +++ b/openstackclient/tests/compute/v2/test_server.py @@ -53,11 +53,15 @@ class TestServer(compute_fakes.TestComputev2): self.volumes_mock = self.app.client_manager.volume.volumes self.volumes_mock.reset_mock() + # Set object attributes to be tested. Could be overwriten in subclass. + self.attrs = {} + # Set object methods to be tested. Could be overwriten in subclass. self.methods = {} def setup_servers_mock(self, count): - servers = compute_fakes.FakeServer.create_servers(methods=self.methods, + servers = compute_fakes.FakeServer.create_servers(attrs=self.attrs, + methods=self.methods, count=count) # This is the return value for utils.find_resource()