add '-m' option to test the solution callback
[platform/upstream/libsolv.git] / tools / testsolv.c
index be99975..34f393b 100644 (file)
@@ -5,9 +5,22 @@
 #include "pool.h"
 #include "repo.h"
 #include "solver.h"
+#include "selection.h"
 #include "solverdebug.h"
 #include "testcase.h"
 
+static struct resultflags2str {
+  Id flag;
+  const char *str;
+} resultflags2str[] = {
+  { TESTCASE_RESULT_TRANSACTION,        "transaction" },
+  { TESTCASE_RESULT_PROBLEMS,           "problems" },
+  { TESTCASE_RESULT_ORPHANED,           "orphaned" },
+  { TESTCASE_RESULT_RECOMMENDED,        "recommended" },
+  { TESTCASE_RESULT_UNNEEDED,           "unneeded" },
+  { 0, 0 }
+};
+
 static void
 usage(ex)
 {
@@ -15,28 +28,82 @@ usage(ex)
   exit(ex);
 }
 
+struct reportsolutiondata {
+  int count;
+  char *result;
+};
+
+static int
+reportsolutioncb(Solver *solv, void *cbdata)
+{
+  struct reportsolutiondata *sd = cbdata;
+  char *res;
+
+  sd->count++;
+  res = testcase_solverresult(solv, TESTCASE_RESULT_TRANSACTION);
+  if (*res)
+    {
+      char prefix[64];
+      char *p2, *p = res;
+      sprintf(prefix, "callback%d:", sd->count);
+      while ((p2 = strchr(p, '\n')) != 0)
+       {
+         char c = p2[1];
+         p2[1] = 0;
+         sd->result = solv_dupappend(sd->result, prefix, p);
+         p2[1] = c;
+         p = p2 + 1;
+       }
+    }
+  solv_free(res);
+  return 0;
+}
+
 int
 main(int argc, char **argv)
 {
   Pool *pool;
   Queue job;
+  Queue solq;
   Solver *solv;
   char *result = 0;
   int resultflags = 0;
   int debuglevel = 0;
+  int writeresult = 0;
+  int multijob = 0;
+  int rescallback = 0;
   int c;
   int ex = 0;
+  const char *list = 0;
+  FILE *fp;
+  const char *p;
 
-  while ((c = getopt(argc, argv, "vh")) >= 0)
+  queue_init(&solq);
+  while ((c = getopt(argc, argv, "vmrhl:s:")) >= 0)
     {
       switch (c)
       {
         case 'v':
           debuglevel++;
           break;
+        case 'r':
+          writeresult++;
+          break;
+        case 'm':
+          rescallback = 1;
+          break;
         case 'h':
          usage(0);
           break;
+        case 'l':
+         list = optarg;
+          break;
+        case 's':
+         if ((p = strchr(optarg, ':')))
+           queue_push2(&solq, atoi(optarg), atoi(p + 1));
+         else
+           queue_push2(&solq, 1, atoi(optarg));
+          break;
         default:
          usage(1);
           break;
@@ -48,61 +115,176 @@ main(int argc, char **argv)
     {
       pool = pool_create();
       pool_setdebuglevel(pool, debuglevel);
-      queue_init(&job);
-      solv = testcase_read(pool, 0, argv[optind], &job, &result, &resultflags);
-      if (!solv)
+
+      fp = fopen(argv[optind], "r");
+      if (!fp)
        {
-         pool_free(pool);
-         exit(1);
+         perror(argv[optind]);
+         exit(0);
        }
-
-      if (result)
+      while (!feof(fp))
        {
-         char *myresult, *resultdiff;
-         solver_solve(solv, &job);
-         myresult = testcase_solverresult(solv, resultflags);
-         resultdiff = testcase_resultdiff(result, myresult);
-         if (resultdiff)
+         queue_init(&job);
+         result = 0;
+         resultflags = 0;
+         solv = testcase_read(pool, fp, argv[optind], &job, &result, &resultflags);
+         if (!solv)
            {
-             printf("Results differ:\n%s", resultdiff);
-             ex = 1;
-             solv_free(resultdiff);
+             pool_free(pool);
+             exit(1);
            }
-         solv_free(result);
-         solv_free(myresult);
-       }
-      else
-       {
-         if (solver_solve(solv, &job))
+
+         if (!multijob && !feof(fp))
+           multijob = 1;
+
+         if (multijob)
+           printf("test %d:\n", multijob++);
+         if (list)
            {
-             int problem, solution, pcnt, scnt;
-             pcnt = solver_problem_count(solv);
-             printf("Found %d problems:\n", pcnt);
-             for (problem = 1; problem <= pcnt; problem++)
+             queue_empty(&job);
+             selection_make(pool, &job, list, SELECTION_NAME|SELECTION_PROVIDES|SELECTION_FILELIST|SELECTION_CANON|SELECTION_DOTARCH|SELECTION_REL|SELECTION_GLOB|SELECTION_FLAT);
+             if (!job.elements)
+               printf("No match\n");
+             else
                {
-                 printf("Problem %d:\n", problem);
-                 solver_printprobleminfo(solv, problem);
-                 printf("\n");
-                 scnt = solver_solution_count(solv, problem);
-                 for (solution = 1; solution <= scnt; solution++)
+                 Queue q;
+                 int i;
+                 queue_init(&q);
+                 selection_solvables(pool, &job, &q);
+                 for (i = 0; i < q.count; i++)
+                   printf("  - %s\n", testcase_solvid2str(pool, q.elements[i]));
+                 queue_free(&q);
+               }
+           }
+         else if (result || writeresult)
+           {
+             char *myresult, *resultdiff;
+             struct reportsolutiondata reportsolutiondata;
+             memset(&reportsolutiondata, 0, sizeof(reportsolutiondata));
+             if (rescallback)
+               {
+                 solv->solution_callback = reportsolutioncb;
+                 solv->solution_callback_data = &reportsolutiondata;
+               }
+             solver_solve(solv, &job);
+             solv->solution_callback = 0;
+             solv->solution_callback_data = 0;
+             if (!resultflags)
+               resultflags = TESTCASE_RESULT_TRANSACTION | TESTCASE_RESULT_PROBLEMS;
+             myresult = testcase_solverresult(solv, resultflags);
+             if (rescallback && reportsolutiondata.result)
+               {
+                 reportsolutiondata.result = solv_dupjoin(reportsolutiondata.result, myresult, 0);
+                 solv_free(myresult);
+                 myresult = reportsolutiondata.result;
+               }
+             if (writeresult)
+               {
+                 if (*myresult)
                    {
-                     printf("Solution %d:\n", solution);
-                     solver_printsolution(solv, problem, solution);
-                     printf("\n");
+                     if (writeresult > 1)
+                       {
+                         const char *p;
+                         int i;
+                         
+                         printf("result ");
+                         p = "%s";
+                         for (i = 0; resultflags2str[i].str; i++)
+                           if ((resultflags & resultflags2str[i].flag) != 0)
+                             {
+                               printf(p, resultflags2str[i].str);
+                               p = ",%s";
+                             }
+                         printf(" <inline>\n");
+                         p = myresult;
+                         while (*p)
+                           {
+                             const char *p2 = strchr(p, '\n');
+                             p2 = p2 ? p2 + 1 : p + strlen(p);
+                             printf("#>%.*s", (int)(p2 - p), p);
+                             p = p2;
+                           }
+                       }
+                     else
+                       printf("%s", myresult);
+                   }
+               }
+             else
+               {
+                 resultdiff = testcase_resultdiff(result, myresult);
+                 if (resultdiff)
+                   {
+                     printf("Results differ:\n%s", resultdiff);
+                     ex = 1;
+                     solv_free(resultdiff);
                    }
                }
+             solv_free(result);
+             solv_free(myresult);
            }
          else
            {
-             Transaction *trans = solver_create_transaction(solv);
-             printf("Transaction summary:\n\n");
-             transaction_print(trans);
-             transaction_free(trans);
+             int pcnt = solver_solve(solv, &job);
+             if (pcnt && solq.count)
+               {
+                 int i, taken = 0;
+                 for (i = 0; i < solq.count; i += 2)
+                   {
+                     if (solq.elements[i] > 0 && solq.elements[i] <= pcnt)
+                       if (solq.elements[i + 1] > 0 && solq.elements[i + 1] <=  solver_solution_count(solv, solq.elements[i]))
+                         {
+                           printf("problem %d: taking solution %d\n", solq.elements[i], solq.elements[i + 1]);
+                           solver_take_solution(solv, solq.elements[i], solq.elements[i + 1], &job);
+                           taken = 1;
+                         }
+                   }
+                 if (taken)
+                   pcnt = solver_solve(solv, &job);
+               }
+             if (pcnt)
+               {
+                 int problem, solution, scnt;
+                 printf("Found %d problems:\n", pcnt);
+                 for (problem = 1; problem <= pcnt; problem++)
+                   {
+                     printf("Problem %d:\n", problem);
+#if 1
+                     solver_printprobleminfo(solv, problem);
+#else
+                     {
+                       Queue pq;
+                       int j;
+                       queue_init(&pq);
+                       solver_findallproblemrules(solv, problem, &pq);
+                       for (j = 0; j < pq.count; j++)
+                         solver_printproblemruleinfo(solv, pq.elements[j]);
+                       queue_free(&pq);
+                     }
+#endif
+                     printf("\n");
+                     scnt = solver_solution_count(solv, problem);
+                     for (solution = 1; solution <= scnt; solution++)
+                       {
+                         printf("Solution %d:\n", solution);
+                         solver_printsolution(solv, problem, solution);
+                         printf("\n");
+                       }
+                   }
+               }
+             else
+               {
+                 Transaction *trans = solver_create_transaction(solv);
+                 printf("Transaction summary:\n\n");
+                 transaction_print(trans);
+                 transaction_free(trans);
+               }
            }
+         queue_free(&job);
+         solver_free(solv);
        }
-      queue_free(&job);
-      solver_free(solv);
       pool_free(pool);
+      fclose(fp);
     }
+  queue_free(&solq);
   exit(ex);
 }