src/mc/sosp/PageStore_test.cpp

   1 /* Copyright (c) 2015-2018. The SimGrid Team. All rights reserved.          */
   2
   3 /* This program is free software; you can redistribute it and/or modify it
   4  * under the terms of the license (GNU LGPL) which comes with this package. */
   5
   6 #define BOOST_TEST_MODULE PAGESTORE
   7 #define BOOST_TEST_DYN_LINK
   8 #include <boost/test/unit_test.hpp>
   9
  10 #include <cstdint>
  11 #include <cstring>
  12 #include <iostream>
  13
  14 #include <sys/mman.h>
  15 #include <unistd.h>
  16
  17 #include <memory>
  18
  19 #include "src/mc/sosp/PageStore.hpp"
  20
  21 using simgrid::mc::PageStore;
  22
  23 /***********************************/
  24 // a class to hold the variable used in the test cases
  25 class BOOST_tests {
  26 public:
  27   static std::size_t pagesize;
  28   static std::unique_ptr<PageStore> store;
  29   static void* data;
  30   static size_t pageno1, pageno2, pageno3, pageno4;
  31   static int value;
  32
  33 public: // member functions used by the test suite(s)
  34   static void Init();
  35   static void store_page_once();
  36   static void store_same_page();
  37   static void store_new_page();
  38   static void unref_pages();
  39   static void reallocate_page();
  40
  41   static void new_content(void* data, std::size_t size);
  42   static void* getpage();
  43 };
  44
  45 // static member datat initialization
  46 std::size_t BOOST_tests::pagesize             = 0;
  47 std::unique_ptr<PageStore> BOOST_tests::store = nullptr;
  48 void* BOOST_tests::data                       = nullptr;
  49 size_t BOOST_tests::pageno1                   = 0;
  50 size_t BOOST_tests::pageno2                   = 0;
  51 size_t BOOST_tests::pageno3                   = 0;
  52 size_t BOOST_tests::pageno4                   = 0;
  53 int BOOST_tests::value                        = 0;
  54
  55 void BOOST_tests::Init()
  56 {
  57   pagesize = (size_t)getpagesize();
  58   store    = std::unique_ptr<PageStore>(new simgrid::mc::PageStore(500));
  59   data     = getpage();
  60   BOOST_CHECK_MESSAGE(store->size() == 0, "Bad size");
  61 }
  62
  63 void BOOST_tests::store_page_once()
  64 {
  65   new_content(data, pagesize);
  66   pageno1 = store->store_page(data);
  67   BOOST_CHECK_MESSAGE(store->get_ref(pageno1) == 1, "Bad refcount");
  68   const void* copy = store->get_page(pageno1);
  69   BOOST_CHECK_MESSAGE(::memcmp(data, copy, pagesize) == 0, "Page data should be the same");
  70   BOOST_CHECK_MESSAGE(store->size() == 1, "Bad size");
  71 }
  72
  73 void BOOST_tests::store_same_page()
  74 {
  75   pageno2 = store->store_page(data);
  76   BOOST_CHECK_MESSAGE(pageno1 == pageno2, "Page should be the same");
  77   BOOST_CHECK_MESSAGE(store->get_ref(pageno1) == 2, "Bad refcount");
  78   BOOST_CHECK_MESSAGE(store->size() == 1, "Bad size");
  79 }
  80
  81 void BOOST_tests::store_new_page()
  82 {
  83   new_content(data, pagesize);
  84   pageno3 = store->store_page(data);
  85   BOOST_CHECK_MESSAGE(pageno1 != pageno3, "New page should be different");
  86   BOOST_CHECK_MESSAGE(store->size() == 2, "Bad size");
  87 }
  88
  89 void BOOST_tests::unref_pages()
  90 {
  91   store->unref_page(pageno1);
  92   BOOST_CHECK_MESSAGE(store->get_ref(pageno1) == 1, "Bad refcount");
  93   BOOST_CHECK_MESSAGE(store->size() == 2, "Bad size");
  94   store->unref_page(pageno2);
  95   BOOST_CHECK_MESSAGE(store->size() == 1, "Bad size");
  96 }
  97
  98 void BOOST_tests::reallocate_page()
  99 {
 100   new_content(data, pagesize);
 101   pageno4 = store->store_page(data);
 102   BOOST_CHECK_MESSAGE(pageno1 == pageno4, "Page was not reused");
 103   BOOST_CHECK_MESSAGE(store->get_ref(pageno4) == 1, "Bad refcount");
 104   BOOST_CHECK_MESSAGE(store->size() == 2, "Bad size");
 105 }
 106
 107 void BOOST_tests::new_content(void* data, std::size_t size)
 108 {
 109   ::memset(data, ++value, size);
 110 }
 111
 112 void* BOOST_tests::getpage()
 113 {
 114   return mmap(nullptr, getpagesize(), PROT_READ | PROT_WRITE, MAP_PRIVATE | MAP_ANONYMOUS, -1, 0);
 115 }
 116
 117 namespace utf = boost::unit_test; // for test case dependence
 118
 119 BOOST_AUTO_TEST_SUITE(PAGESTORE)
 120 BOOST_AUTO_TEST_CASE(Init)
 121 {
 122   BOOST_tests::Init();
 123 }
 124
 125 BOOST_AUTO_TEST_CASE(store_page_once, *utf::depends_on("PAGESTORE/Init"))
 126 {
 127   BOOST_tests::store_page_once();
 128 }
 129
 130 BOOST_AUTO_TEST_CASE(store_same_page, *utf::depends_on("PAGESTORE/store_page_once"))
 131 {
 132   BOOST_tests::store_same_page();
 133 }
 134
 135 BOOST_AUTO_TEST_CASE(store_new_page, *utf::depends_on("PAGESTORE/store_same_page"))
 136 {
 137   BOOST_tests::store_new_page();
 138 }
 139
 140 BOOST_AUTO_TEST_CASE(unref_pages, *utf::depends_on("PAGESTORE/store_new_page"))
 141 {
 142   BOOST_tests::unref_pages();
 143 }
 144
 145 BOOST_AUTO_TEST_CASE(reallocate_page, *utf::depends_on("PAGESTORE/unref_pages"))
 146 {
 147   BOOST_tests::reallocate_page();
 148 }
 149 BOOST_AUTO_TEST_SUITE_END()