END_TEST
START_TEST(test_goal_selector_upgrade)
{
HySelector sltr = hy_selector_create(test_globals.sack);
HyGoal goal = hy_goal_create(test_globals.sack);
fail_if(hy_selector_set(sltr, HY_PKG_NAME, HY_EQ, "dog"));
fail_if(hy_selector_set(sltr, HY_PKG_EVR, HY_EQ, "1-2"));
fail_if(hy_goal_upgrade_to_selector(goal, sltr));
fail_if(hy_goal_run(goal));
GPtrArray *plist = hy_goal_list_upgrades(goal, NULL);
fail_unless(plist->len == 1);
assert_nevra_eq(g_ptr_array_index(plist, 0), "dog-1-2.x86_64");
g_ptr_array_unref(plist);
hy_goal_free(goal);
hy_selector_free(sltr);
sltr = hy_selector_create(test_globals.sack);
goal = hy_goal_create(test_globals.sack);
fail_if(hy_selector_set(sltr, HY_PKG_NAME, HY_EQ, "pilchard"));
fail_if(hy_goal_upgrade_to_selector(goal, sltr));
fail_if(hy_goal_run(goal));
assert_iueo(goal, 0, 2, 0, 0);
hy_goal_free(goal);
hy_selector_free(sltr);
}
END_TEST
START_TEST(test_goal_selector_upgrade)
{
HySelector sltr = hy_selector_create(test_globals.sack);
HyGoal goal = hy_goal_create(test_globals.sack);
fail_if(hy_selector_set(sltr, HY_PKG_NAME, HY_EQ, "dog"));
fail_if(hy_selector_set(sltr, HY_PKG_EVR, HY_EQ, "1-2"));
fail_if(hy_goal_upgrade_to_selector(goal, sltr));
fail_if(hy_goal_run(goal));
HyPackageList plist = hy_goal_list_upgrades(goal);
fail_unless(hy_packagelist_count(plist) == 1);
assert_nevra_eq(hy_packagelist_get(plist, 0), "dog-1-2.x86_64");
hy_packagelist_free(plist);
hy_goal_free(goal);
hy_selector_free(sltr);
sltr = hy_selector_create(test_globals.sack);
goal = hy_goal_create(test_globals.sack);
fail_if(hy_selector_set(sltr, HY_PKG_NAME, HY_EQ, "pilchard"));
fail_if(hy_goal_upgrade_to_selector(goal, sltr));
fail_if(hy_goal_run(goal));
assert_iueo(goal, 0, 2, 0, 0);
hy_goal_free(goal);
hy_selector_free(sltr);
}
static PyObject *
upgrade_to(_GoalObject *self, PyObject *args, PyObject *kwds)
{
HifPackage *pkg = NULL;
HySelector sltr = NULL;
int ret;
int flags = 0;
if (!args_pkg_sltr_parse(args, kwds, &pkg, &sltr,
&flags, HY_CHECK_INSTALLED))
return NULL;
if (sltr) {
ret = hy_goal_upgrade_to_selector(self->goal, sltr);
return op_ret2exc(ret);
}
ret = hy_goal_upgrade_to_flags(self->goal, pkg, flags);
return op_ret2exc(ret);
}
uint32_t
TDNFGoal(
PTDNF pTdnf,
HyPackageList hPkgList,
HySelector hSelector,
TDNF_ALTERTYPE nResolveFor,
PTDNF_SOLVED_PKG_INFO pInfo
)
{
uint32_t dwError = 0;
HyGoal hGoal = NULL;
HyPackage hPkg = NULL;
PTDNF_PKG_INFO pPkgsToInstall = NULL;
PTDNF_PKG_INFO pPkgsToUpgrade = NULL;
PTDNF_PKG_INFO pPkgsToDowngrade = NULL;
PTDNF_PKG_INFO pPkgsToRemove = NULL;
PTDNF_PKG_INFO pPkgsUnNeeded = NULL;
PTDNF_PKG_INFO pPkgsToReinstall = NULL;
int nFlags = 0;
int nRequirePkgList =
(nResolveFor != ALTER_UPGRADEALL &&
nResolveFor != ALTER_DISTRO_SYNC);
if(!pTdnf || !pInfo )
{
dwError = ERROR_TDNF_INVALID_PARAMETER;
BAIL_ON_TDNF_ERROR(dwError);
}
if(nResolveFor == ALTER_UPGRADE && !hSelector)
{
dwError = ERROR_TDNF_INVALID_PARAMETER;
BAIL_ON_TDNF_ERROR(dwError);
}
if(nRequirePkgList)
{
if(!hPkgList)
{
dwError = ERROR_TDNF_INVALID_PARAMETER;
BAIL_ON_TDNF_ERROR(dwError);
}
hPkg = hy_packagelist_get(hPkgList, 0);
if(!hPkg)
{
dwError = ERROR_TDNF_PACKAGELIST_EMPTY;
BAIL_ON_TDNF_ERROR(dwError);
}
}
hGoal = hy_goal_create(pTdnf->hSack);
if(!hGoal)
{
dwError = ERROR_TDNF_GOAL_CREATE;
BAIL_ON_TDNF_ERROR(dwError);
}
switch(nResolveFor)
{
case ALTER_DOWNGRADE:
dwError = hy_goal_downgrade_to(hGoal, hPkg);
BAIL_ON_TDNF_HAWKEY_ERROR(dwError);
break;
case ALTER_ERASE:
dwError = hy_goal_erase(hGoal, hPkg);
BAIL_ON_TDNF_HAWKEY_ERROR(dwError);
break;
case ALTER_REINSTALL:
dwError = hy_goal_install(hGoal, hPkg);
BAIL_ON_TDNF_HAWKEY_ERROR(dwError);
break;
case ALTER_INSTALL:
dwError = TDNFPackageGetLatest(hPkgList, &hPkg);
BAIL_ON_TDNF_ERROR(dwError);
dwError = hy_goal_install(hGoal, hPkg);
BAIL_ON_TDNF_HAWKEY_ERROR(dwError);
break;
case ALTER_UPGRADE:
dwError = hy_goal_upgrade_to_selector(hGoal, hSelector);
BAIL_ON_TDNF_HAWKEY_ERROR(dwError);
break;
case ALTER_UPGRADEALL:
dwError = hy_goal_upgrade_all(hGoal);
BAIL_ON_TDNF_HAWKEY_ERROR(dwError);
break;
case ALTER_DISTRO_SYNC:
dwError = hy_goal_distupgrade_all(hGoal);
BAIL_ON_TDNF_HAWKEY_ERROR(dwError);
break;
case ALTER_AUTOERASE:
dwError = TDNFGoalSetUserInstalled(hGoal, hPkgList);
BAIL_ON_TDNF_HAWKEY_ERROR(dwError);
break;
default:
dwError = ERROR_TDNF_INVALID_RESOLVE_ARG;
BAIL_ON_TDNF_ERROR(dwError);
}
if(pTdnf->pArgs->nBest)
{
nFlags = nFlags | HY_FORCE_BEST;
}
if(pTdnf->pArgs->nAllowErasing ||
nResolveFor == ALTER_ERASE ||
nResolveFor == ALTER_AUTOERASE)
{
nFlags = nFlags | HY_ALLOW_UNINSTALL;
}
dwError = hy_goal_run_flags(hGoal, nFlags);
if(pTdnf->pArgs->nDebugSolver)
{
hy_goal_write_debugdata(hGoal, "debugdata");
}
BAIL_ON_TDNF_HAWKEY_ERROR(dwError);
dwError = TDNFGoalGetResultsIgnoreNoData(
hy_goal_list_installs(hGoal),
&pPkgsToInstall);
BAIL_ON_TDNF_ERROR(dwError);
dwError = TDNFGoalGetResultsIgnoreNoData(
hy_goal_list_upgrades(hGoal),
&pPkgsToUpgrade);
BAIL_ON_TDNF_ERROR(dwError);
dwError = TDNFGoalGetResultsIgnoreNoData(
hy_goal_list_downgrades(hGoal),
&pPkgsToDowngrade);
BAIL_ON_TDNF_ERROR(dwError);
dwError = TDNFGoalGetResultsIgnoreNoData(
hy_goal_list_erasures(hGoal),
&pPkgsToRemove);
BAIL_ON_TDNF_ERROR(dwError);
if(nResolveFor == ALTER_AUTOERASE)
{
dwError = TDNFGoalGetResultsIgnoreNoData(
hy_goal_list_unneeded(hGoal),
&pPkgsUnNeeded);
BAIL_ON_TDNF_ERROR(dwError);
}
dwError = TDNFGoalGetResultsIgnoreNoData(
hy_goal_list_reinstalls(hGoal),
&pPkgsToReinstall);
BAIL_ON_TDNF_ERROR(dwError);
pInfo->pPkgsToInstall = pPkgsToInstall;
pInfo->pPkgsToUpgrade = pPkgsToUpgrade;
pInfo->pPkgsToDowngrade = pPkgsToDowngrade;
pInfo->pPkgsToRemove = pPkgsToRemove;
pInfo->pPkgsUnNeeded = pPkgsUnNeeded;
pInfo->pPkgsToReinstall = pPkgsToReinstall;
pTdnf->hGoal = hGoal;
cleanup:
return dwError;
error:
if(hGoal)
{
TDNFGoalReportProblems(hGoal);
hy_goal_free(hGoal);
}
if(pPkgsToInstall)
{
TDNFFreePackageInfo(pPkgsToInstall);
}
if(pPkgsToUpgrade)
{
TDNFFreePackageInfo(pPkgsToUpgrade);
}
if(pPkgsToDowngrade)
{
TDNFFreePackageInfo(pPkgsToDowngrade);
}
if(pPkgsToRemove)
{
TDNFFreePackageInfo(pPkgsToRemove);
}
if(pPkgsUnNeeded)
{
TDNFFreePackageInfo(pPkgsUnNeeded);
}
if(pPkgsToReinstall)
{
TDNFFreePackageInfo(pPkgsToReinstall);
}
goto cleanup;
}
