2 files changed, 59 insertions, 0 deletions
diff --git a/test/interactive-helper/CMakeLists.txt b/test/interactive-helper/CMakeLists.txt
index 8e8764353..bfca6c8f6 100644
--- a/test/interactive-helper/CMakeLists.txt
+++ b/test/interactive-helper/CMakeLists.txt
@@ -14,6 +14,8 @@ add_executable(mthdcat mthdcat.cc)
 target_link_libraries(mthdcat ${APTPKG_LIB})
 add_executable(testdeb testdeb.cc)
 target_link_libraries(testdeb ${APTPKG_LIB})
+add_executable(testkeep testkeep.cc)
+target_link_libraries(testkeep ${APTPKG_LIB})
 add_executable(extract-control extract-control.cc)
 target_link_libraries(extract-control ${APTPKG_LIB})
 add_executable(aptwebserver aptwebserver.cc)
diff --git a/test/interactive-helper/testkeep.cc b/test/interactive-helper/testkeep.cc
new file mode 100644
index 000000000..f65f43203
--- /dev/null
+++ b/test/interactive-helper/testkeep.cc
@@ -0,0 +1,57 @@
+#include <config.h>
+
+#include <apt-pkg/init.h>
+#include <apt-pkg/algorithms.h>
+#include <apt-pkg/cachefile.h>
+#include <apt-pkg/error.h>
+#include <apt-pkg/pkgsystem.h>
+#include <apt-pkg/prettyprinters.h>
+
+#include <iostream>
+#include <string>
+
+static bool test(const char *name, bool hold)
+{
+   pkgCacheFile cache;
+   if (not cache.Open())
+      return false;
+
+   for (auto Pkg = cache->PkgBegin(); Pkg.end() == false; ++Pkg)
+      std::cout << "A: " << APT::PrettyPkg(cache, Pkg) << std::endl;
+
+   cache->MarkDelete(cache->FindPkg(name));
+   if (hold)
+      cache->MarkProtected(cache->FindPkg(name));
+
+   for (auto Pkg = cache->PkgBegin(); Pkg.end() == false; ++Pkg)
+      std::cout << "B: " << APT::PrettyPkg(cache, Pkg) << std::endl;
+
+   pkgProblemResolver resolve(cache);
+   bool res = resolve.ResolveByKeep();
+
+   for (auto Pkg = cache->PkgBegin(); Pkg.end() == false; ++Pkg)
+      std::cout << "C: " << APT::PrettyPkg(cache, Pkg) << std::endl;
+
+   return res;
+}
+
+int main(int argc, const char *argv[])
+{
+   if (argc != 2 && argc != 3)
+   {
+      return _error->Error("Expected one argument");
+      goto err;
+   }
+   if (not pkgInitConfig(*_config))
+      goto err;
+   if (not pkgInitSystem(*_config, _system))
+      goto err;
+   if (not test(argv[1], argc > 2 && strcmp(argv[2], "--hold") == 0))
+      goto err;
+
+   return 0;
+err:
+   _error->DumpErrors();
+   return 1;
+}
+