def test_success(self, mock_push_cib, mock_get_topology, mock_remove): mock_get_topology.return_value = "topology el" lib_env = create_lib_env() lib.remove_all_levels(lib_env) mock_remove.assert_called_once_with("topology el") mock_get_topology.assert_called_once_with("mocked cib") mock_push_cib.assert_called_once_with("mocked cib")