def setup(self): cloud_provider = MagicMock() setup_provider = MagicMock() setup_provider.setup_cluster.return_value = True cluster = Cluster(config_cluster_name, config_cloud_name, cloud_provider, setup_provider, 1, 2, Configurator()) cluster.setup() setup_provider.setup_cluster.assert_called_once_with(cluster)