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