def test_load_from_storage(self): cloud_provider = MagicMock() cloud_provider.is_instance_running.return_value = True cloud_provider.get_ips.return_value = ('127.0.0.1', '127.0.0.1') setup_provider = MagicMock() storage = MagicMock() cluster = Cluster(config_cluster_name, config_cloud_name, cloud_provider, setup_provider, 1, 2, Configurator()) cluster._storage = storage cluster.load_from_storage() storage.load_cluster.assert_called_once_with(cluster)