Mercurial > hg > orthanc-webviewer
annotate UnitTestsSources/UnitTestsMain.cpp @ 305:af985076f647 OrthancWebViewer-2.8
OrthancWebViewer-2.8
| author | Sebastien Jodogne <s.jodogne@gmail.com> |
|---|---|
| date | Tue, 22 Feb 2022 08:33:59 +0100 |
| parents | fb7d62e3235e |
| children | 0a1ad7492050 |
| rev | line source |
|---|---|
| 0 | 1 /** |
| 2 * Orthanc - A Lightweight, RESTful DICOM Store | |
| 122 | 3 * Copyright (C) 2012-2016 Sebastien Jodogne, Medical Physics |
| 0 | 4 * Department, University Hospital of Liege, Belgium |
| 303 | 5 * Copyright (C) 2017-2022 Osimis S.A., Belgium |
| 6 * Copyright (C) 2021-2022 Sebastien Jodogne, ICTEAM UCLouvain, Belgium | |
| 0 | 7 * |
| 8 * This program is free software: you can redistribute it and/or | |
| 9 * modify it under the terms of the GNU Affero General Public License | |
| 10 * as published by the Free Software Foundation, either version 3 of | |
| 11 * the License, or (at your option) any later version. | |
| 12 * | |
| 13 * This program is distributed in the hope that it will be useful, but | |
| 14 * WITHOUT ANY WARRANTY; without even the implied warranty of | |
| 15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | |
| 16 * Affero General Public License for more details. | |
| 17 * | |
| 18 * You should have received a copy of the GNU Affero General Public License | |
| 19 * along with this program. If not, see <http://www.gnu.org/licenses/>. | |
| 20 **/ | |
| 21 | |
| 22 | |
| 23 #include <gtest/gtest.h> | |
| 24 #include <boost/lexical_cast.hpp> | |
| 25 | |
| 26 static int argc_; | |
| 27 static char** argv_; | |
| 28 | |
| 29 #include "../Plugin/Cache/CacheManager.h" | |
| 30 #include "../Plugin/Cache/CacheScheduler.h" | |
| 31 #include "../Plugin/Cache/ICacheFactory.h" | |
| 32 #include "../Plugin/Cache/ICacheFactory.h" | |
| 33 | |
|
266
4e9d30c19b4b
linking against orthanc framework library
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
226
diff
changeset
|
34 #include <Compatibility.h> |
|
4e9d30c19b4b
linking against orthanc framework library
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
226
diff
changeset
|
35 #include <Logging.h> |
|
4e9d30c19b4b
linking against orthanc framework library
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
226
diff
changeset
|
36 #include <OrthancException.h> |
|
4e9d30c19b4b
linking against orthanc framework library
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
226
diff
changeset
|
37 #include <SystemToolbox.h> |
|
191
993dd140bd30
Resort to Orthanc framework
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
174
diff
changeset
|
38 |
| 0 | 39 using namespace OrthancPlugins; |
| 40 | |
| 41 | |
| 42 class CacheManagerTest : public testing::Test | |
| 43 { | |
| 44 private: | |
|
266
4e9d30c19b4b
linking against orthanc framework library
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
226
diff
changeset
|
45 std::unique_ptr<Orthanc::FilesystemStorage> storage_; |
|
4e9d30c19b4b
linking against orthanc framework library
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
226
diff
changeset
|
46 std::unique_ptr<Orthanc::SQLite::Connection> db_; |
|
4e9d30c19b4b
linking against orthanc framework library
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
226
diff
changeset
|
47 std::unique_ptr<CacheManager> cache_; |
| 0 | 48 |
| 49 public: | |
| 50 virtual void SetUp() | |
| 51 { | |
| 52 storage_.reset(new Orthanc::FilesystemStorage("UnitTestsResults")); | |
| 53 storage_->Clear(); | |
| 150 | 54 Orthanc::SystemToolbox::RemoveFile("UnitTestsResults/cache.db"); |
| 0 | 55 |
| 56 db_.reset(new Orthanc::SQLite::Connection()); | |
| 57 db_->Open("UnitTestsResults/cache.db"); | |
| 58 | |
|
147
70d1fe6d6309
Avoid hard crash if not enough memory
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
122
diff
changeset
|
59 cache_.reset(new CacheManager(NULL, *db_, *storage_)); |
| 0 | 60 cache_->SetSanityCheckEnabled(true); |
| 61 } | |
| 62 | |
| 63 virtual void TearDown() | |
| 64 { | |
| 65 cache_.reset(NULL); | |
| 66 db_.reset(NULL); | |
| 67 storage_.reset(NULL); | |
| 68 } | |
| 69 | |
| 70 CacheManager& GetCache() | |
| 71 { | |
| 72 return *cache_; | |
| 73 } | |
| 74 | |
| 75 Orthanc::FilesystemStorage& GetStorage() | |
| 76 { | |
| 77 return *storage_; | |
| 78 } | |
| 79 }; | |
| 80 | |
| 81 | |
| 82 | |
| 83 class TestF : public ICacheFactory | |
| 84 { | |
| 85 private: | |
| 86 int bundle_; | |
| 87 | |
| 88 public: | |
| 197 | 89 explicit TestF(int bundle) : bundle_(bundle) |
| 0 | 90 { |
| 91 } | |
| 92 | |
| 93 virtual bool Create(std::string& content, | |
| 289 | 94 const std::string& key) ORTHANC_OVERRIDE |
| 0 | 95 { |
| 96 content = "Bundle " + boost::lexical_cast<std::string>(bundle_) + ", item " + key; | |
| 97 return true; | |
| 98 } | |
| 99 }; | |
| 100 | |
| 101 | |
| 102 TEST_F(CacheManagerTest, DefaultQuota) | |
| 103 { | |
| 104 std::set<std::string> f; | |
| 105 GetStorage().ListAllFiles(f); | |
| 66 | 106 ASSERT_EQ(0u, f.size()); |
| 0 | 107 |
| 108 GetCache().SetDefaultQuota(10, 0); | |
| 66 | 109 for (unsigned int i = 0; i < 30; i++) |
| 0 | 110 { |
| 111 GetStorage().ListAllFiles(f); | |
| 66 | 112 ASSERT_EQ(i >= 10 ? 10u : i, f.size()); |
| 0 | 113 std::string s = boost::lexical_cast<std::string>(i); |
| 114 GetCache().Store(0, s, "Test " + s); | |
| 115 } | |
| 116 | |
| 117 GetStorage().ListAllFiles(f); | |
| 66 | 118 ASSERT_EQ(10u, f.size()); |
| 0 | 119 |
| 120 for (int i = 0; i < 30; i++) | |
| 121 { | |
| 122 ASSERT_EQ(i >= 20, GetCache().IsCached(0, boost::lexical_cast<std::string>(i))); | |
| 123 } | |
| 124 | |
| 125 GetCache().SetDefaultQuota(5, 0); | |
| 126 GetStorage().ListAllFiles(f); | |
| 66 | 127 ASSERT_EQ(5u, f.size()); |
| 0 | 128 for (int i = 0; i < 30; i++) |
| 129 { | |
| 130 ASSERT_EQ(i >= 25, GetCache().IsCached(0, boost::lexical_cast<std::string>(i))); | |
| 131 } | |
| 132 | |
| 133 for (int i = 0; i < 15; i++) | |
| 134 { | |
| 135 std::string s = boost::lexical_cast<std::string>(i); | |
| 136 GetCache().Store(0, s, "Test " + s); | |
| 137 } | |
| 138 | |
| 139 GetStorage().ListAllFiles(f); | |
| 66 | 140 ASSERT_EQ(5u, f.size()); |
| 0 | 141 |
| 142 for (int i = 0; i < 50; i++) | |
| 143 { | |
| 144 std::string s = boost::lexical_cast<std::string>(i); | |
| 145 if (i >= 10 && i < 15) | |
| 146 { | |
| 147 std::string tmp; | |
| 148 ASSERT_TRUE(GetCache().IsCached(0, s)); | |
| 149 ASSERT_TRUE(GetCache().Access(tmp, 0, s)); | |
| 150 ASSERT_EQ("Test " + s, tmp); | |
| 151 } | |
| 152 else | |
| 153 { | |
| 154 ASSERT_FALSE(GetCache().IsCached(0, s)); | |
| 155 } | |
| 156 } | |
| 157 } | |
| 158 | |
| 159 | |
| 160 | |
| 161 TEST_F(CacheManagerTest, Invalidate) | |
| 162 { | |
| 163 GetCache().SetDefaultQuota(10, 0); | |
| 164 for (int i = 0; i < 30; i++) | |
| 165 { | |
| 166 std::string s = boost::lexical_cast<std::string>(i); | |
| 167 GetCache().Store(0, s, "Test " + s); | |
| 168 } | |
| 169 | |
| 170 std::set<std::string> f; | |
| 171 GetStorage().ListAllFiles(f); | |
| 66 | 172 ASSERT_EQ(10u, f.size()); |
| 0 | 173 |
| 174 GetCache().Invalidate(0, "25"); | |
| 175 GetStorage().ListAllFiles(f); | |
| 66 | 176 ASSERT_EQ(9u, f.size()); |
| 0 | 177 for (int i = 0; i < 50; i++) |
| 178 { | |
| 179 std::string s = boost::lexical_cast<std::string>(i); | |
| 180 ASSERT_EQ((i >= 20 && i < 30 && i != 25), GetCache().IsCached(0, s)); | |
| 181 } | |
| 182 | |
| 183 for (int i = 0; i < 50; i++) | |
| 184 { | |
| 185 GetCache().Invalidate(0, boost::lexical_cast<std::string>(i)); | |
| 186 } | |
| 187 | |
| 188 GetStorage().ListAllFiles(f); | |
| 66 | 189 ASSERT_EQ(0u, f.size()); |
| 0 | 190 } |
| 191 | |
| 192 | |
| 193 | |
| 194 int main(int argc, char **argv) | |
| 195 { | |
| 196 argc_ = argc; | |
| 197 argv_ = argv; | |
| 198 | |
| 199 ::testing::InitGoogleTest(&argc, argv); | |
| 200 | |
| 195 | 201 /*Orthanc::Logging::Initialize(); |
| 202 Orthanc::Logging::EnableInfoLevel(true);*/ | |
|
191
993dd140bd30
Resort to Orthanc framework
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
174
diff
changeset
|
203 |
| 0 | 204 return RUN_ALL_TESTS(); |
| 205 } |