test/ToGraphvizWriterTests.cpp

   1 #include <fstream>
   2 #include <memory>
   3
   4 #include <UnitTest++/UnitTest++.h>
   5
   6 #include "Article.h"
   7 #include "ToGraphvizWriter.h"
   8
   9 // these test always pass, unless exceptions occur
  10 //! TODO: try to initialize some variables only in one place
  11
  12 SUITE(ArticleToGraphvizWriterTests)
  13 {
  14   using namespace WikiWalker;
  15   using namespace WikiWalker::CollectionUtils;
  16
  17   TEST(WriteArticleWithMultipleLinks)
  18   {
  19     std::ofstream outfile("_artlinks.dot",
  20                           std::ofstream::out | std::ofstream::trunc);
  21     ToGraphvizWriter atj;
  22     Article a("Farm");
  23
  24     auto al1 = std::make_shared<Article>("Animal"),
  25          al2 = std::make_shared<Article>("Pig"),
  26          al3 = std::make_shared<Article>("Equality");
  27
  28     a.addLink(al1);
  29     a.addLink(al2);
  30     a.addLink(al3);
  31     atj.output(&a, outfile);
  32     outfile.flush();
  33     outfile.close();
  34   }
  35
  36   TEST(WriteArticleCollectionDepthOne)
  37   {
  38     std::ofstream outfile("_artctree.dot",
  39                           std::ofstream::out | std::ofstream::trunc);
  40     ToGraphvizWriter atj;
  41     ArticleCollection ac;
  42
  43     auto farm = std::make_shared<Article>("Farm");
  44     CollectionUtils::add(ac, farm);
  45
  46     auto animal = std::make_shared<Article>("Animal");
  47     farm->addLink(animal);
  48     CollectionUtils::add(ac, animal);
  49
  50     auto pig = std::make_shared<Article>("Pig");
  51     farm->addLink(pig);
  52     CollectionUtils::add(ac, pig);
  53
  54     auto equa = std::make_shared<Article>("Equality");
  55     farm->addLink(equa);
  56     CollectionUtils::add(ac, equa);
  57
  58     atj.output(ac, outfile);
  59     outfile.flush();
  60     outfile.close();
  61   }
  62
  63   TEST(WriteArticleCollectionDepthOne_WithInterlinks)
  64   {
  65     std::ofstream outfile("_artctree_interlink.dot",
  66                           std::ofstream::out | std::ofstream::trunc);
  67     ToGraphvizWriter atj;
  68     ArticleCollection ac;
  69
  70     auto farm = std::make_shared<Article>("Farm");
  71     CollectionUtils::add(ac, farm);
  72
  73     auto animal = std::make_shared<Article>("Animal");
  74     farm->addLink(animal);
  75     CollectionUtils::add(ac, animal);
  76
  77     auto pig = std::make_shared<Article>("Pig");
  78     pig->marked(true);
  79     farm->addLink(pig);
  80     CollectionUtils::add(ac, pig);
  81
  82     pig->addLink(animal);
  83     animal->addLink(pig);
  84
  85     auto equa = std::make_shared<Article>("Equality");
  86     farm->addLink(equa);
  87     CollectionUtils::add(ac, equa);
  88
  89     atj.output(ac, outfile);
  90     outfile.flush();
  91     outfile.close();
  92   }
  93
  94   TEST(WriteArticleCollectionDepthTwo)
  95   {
  96     std::ofstream outfile("_artc_deeptree.dot",
  97                           std::ofstream::out | std::ofstream::trunc);
  98     ToGraphvizWriter atj;
  99     ArticleCollection ac;
 100
 101     auto farm = std::make_shared<Article>("Farm");
 102     CollectionUtils::add(ac, farm);
 103
 104     auto animal = std::make_shared<Article>("Animal");
 105     farm->addLink(animal);
 106     CollectionUtils::add(ac, animal);
 107
 108     auto pig = std::make_shared<Article>("Pig");
 109     farm->addLink(pig);
 110     CollectionUtils::add(ac, pig);
 111
 112     auto equa = std::make_shared<Article>("Equality");
 113     farm->addLink(equa);
 114     CollectionUtils::add(ac, equa);
 115
 116     auto rights = std::make_shared<Article>("Rights");
 117     equa->addLink(rights);
 118     CollectionUtils::add(ac, rights);
 119
 120     auto cat = std::make_shared<Article>("Cat");
 121     CollectionUtils::add(ac, cat);
 122     animal->addLink(cat);
 123     auto dog = std::make_shared<Article>("Dog");
 124     CollectionUtils::add(ac, dog);
 125     animal->addLink(dog);
 126
 127     atj.output(ac, outfile);
 128     outfile.flush();
 129     outfile.close();
 130   }
 131 }