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