Mercurial > hg > orthanc
annotate UnitTests/ServerIndex.cpp @ 191:bff0b77b02fa
refactoring getalluuids
author | Sebastien Jodogne <s.jodogne@gmail.com> |
---|---|
date | Tue, 13 Nov 2012 11:22:33 +0100 |
parents | b6cef9d45cc3 |
children | a1b9d1e1497b |
rev | line source |
---|---|
181 | 1 #include "gtest/gtest.h" |
2 | |
183 | 3 #include "../OrthancServer/DatabaseWrapper.h" |
4 | |
181 | 5 #include <ctype.h> |
6 #include <glog/logging.h> | |
7 | |
8 | |
183 | 9 using namespace Orthanc; |
181 | 10 |
183 | 11 namespace |
12 { | |
181 | 13 class ServerIndexListener : public IServerIndexListener |
14 { | |
15 public: | |
183 | 16 std::set<std::string> deletedFiles_; |
17 std::string ancestorId_; | |
18 ResourceType ancestorType_; | |
19 | |
20 void Reset() | |
181 | 21 { |
183 | 22 ancestorId_ = ""; |
23 deletedFiles_.clear(); | |
24 } | |
25 | |
26 virtual void SignalRemainingAncestor(ResourceType type, | |
27 const std::string& publicId) | |
28 { | |
29 ancestorId_ = publicId; | |
30 ancestorType_ = type; | |
181 | 31 } |
32 | |
33 virtual void SignalFileDeleted(const std::string& fileUuid) | |
34 { | |
183 | 35 deletedFiles_.insert(fileUuid); |
181 | 36 LOG(INFO) << "A file must be removed: " << fileUuid; |
37 } | |
38 }; | |
39 } | |
40 | |
41 | |
183 | 42 TEST(DatabaseWrapper, Simple) |
181 | 43 { |
44 ServerIndexListener listener; | |
183 | 45 DatabaseWrapper index(listener); |
181 | 46 |
47 int64_t a[] = { | |
182 | 48 index.CreateResource("a", ResourceType_Patient), // 0 |
49 index.CreateResource("b", ResourceType_Study), // 1 | |
50 index.CreateResource("c", ResourceType_Series), // 2 | |
51 index.CreateResource("d", ResourceType_Instance), // 3 | |
52 index.CreateResource("e", ResourceType_Instance), // 4 | |
53 index.CreateResource("f", ResourceType_Instance), // 5 | |
54 index.CreateResource("g", ResourceType_Study) // 6 | |
181 | 55 }; |
56 | |
190 | 57 { |
58 Json::Value t; | |
59 index.GetAllPublicIds(t, ResourceType_Patient); | |
60 | |
61 ASSERT_EQ(1, t.size()); | |
62 ASSERT_EQ("a", t[0u].asString()); | |
63 | |
64 index.GetAllPublicIds(t, ResourceType_Series); | |
65 ASSERT_EQ(1, t.size()); | |
66 ASSERT_EQ("c", t[0u].asString()); | |
67 | |
68 index.GetAllPublicIds(t, ResourceType_Study); | |
69 ASSERT_EQ(2, t.size()); | |
70 | |
71 index.GetAllPublicIds(t, ResourceType_Instance); | |
72 ASSERT_EQ(3, t.size()); | |
73 } | |
74 | |
181 | 75 index.SetGlobalProperty("Hello", "World"); |
76 | |
77 index.AttachChild(a[0], a[1]); | |
78 index.AttachChild(a[1], a[2]); | |
79 index.AttachChild(a[2], a[3]); | |
80 index.AttachChild(a[2], a[4]); | |
81 index.AttachChild(a[6], a[5]); | |
182 | 82 |
83 std::string s; | |
84 | |
85 ASSERT_FALSE(index.GetParentPublicId(s, a[0])); | |
86 ASSERT_FALSE(index.GetParentPublicId(s, a[6])); | |
87 ASSERT_TRUE(index.GetParentPublicId(s, a[1])); ASSERT_EQ("a", s); | |
88 ASSERT_TRUE(index.GetParentPublicId(s, a[2])); ASSERT_EQ("b", s); | |
89 ASSERT_TRUE(index.GetParentPublicId(s, a[3])); ASSERT_EQ("c", s); | |
90 ASSERT_TRUE(index.GetParentPublicId(s, a[4])); ASSERT_EQ("c", s); | |
91 ASSERT_TRUE(index.GetParentPublicId(s, a[5])); ASSERT_EQ("g", s); | |
92 | |
185 | 93 std::list<std::string> l; |
182 | 94 index.GetChildrenPublicId(l, a[0]); ASSERT_EQ(1, l.size()); ASSERT_EQ("b", l.front()); |
95 index.GetChildrenPublicId(l, a[1]); ASSERT_EQ(1, l.size()); ASSERT_EQ("c", l.front()); | |
96 index.GetChildrenPublicId(l, a[3]); ASSERT_EQ(0, l.size()); | |
97 index.GetChildrenPublicId(l, a[4]); ASSERT_EQ(0, l.size()); | |
98 index.GetChildrenPublicId(l, a[5]); ASSERT_EQ(0, l.size()); | |
99 index.GetChildrenPublicId(l, a[6]); ASSERT_EQ(1, l.size()); ASSERT_EQ("f", l.front()); | |
100 | |
101 index.GetChildrenPublicId(l, a[2]); ASSERT_EQ(2, l.size()); | |
102 if (l.front() == "d") | |
103 { | |
104 ASSERT_EQ("e", l.back()); | |
105 } | |
106 else | |
107 { | |
108 ASSERT_EQ("d", l.back()); | |
109 ASSERT_EQ("e", l.front()); | |
110 } | |
111 | |
183 | 112 index.AttachFile(a[4], "_json", "my json file", 21, 42, CompressionType_Zlib); |
113 index.AttachFile(a[4], "_dicom", "my dicom file", 42); | |
114 index.AttachFile(a[6], "_hello", "world", 44); | |
115 index.SetMetadata(a[4], MetadataType_Instance_RemoteAet, "PINNACLE"); | |
182 | 116 |
183 | 117 ASSERT_EQ(21 + 42 + 44, index.GetTotalCompressedSize()); |
118 ASSERT_EQ(42 + 42 + 44, index.GetTotalUncompressedSize()); | |
181 | 119 |
120 DicomMap m; | |
121 m.SetValue(0x0010, 0x0010, "PatientName"); | |
122 index.SetMainDicomTags(a[3], m); | |
123 | |
124 int64_t b; | |
125 ResourceType t; | |
188
090cefdab1d1
fix because of Windows macros
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
185
diff
changeset
|
126 ASSERT_TRUE(index.LookupResource("g", b, t)); |
181 | 127 ASSERT_EQ(7, b); |
128 ASSERT_EQ(ResourceType_Study, t); | |
129 | |
188
090cefdab1d1
fix because of Windows macros
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
185
diff
changeset
|
130 ASSERT_TRUE(index.LookupMetadata(s, a[4], MetadataType_Instance_RemoteAet)); |
090cefdab1d1
fix because of Windows macros
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
185
diff
changeset
|
131 ASSERT_FALSE(index.LookupMetadata(s, a[4], MetadataType_Instance_IndexInSeries)); |
181 | 132 ASSERT_EQ("PINNACLE", s); |
133 ASSERT_EQ("PINNACLE", index.GetMetadata(a[4], MetadataType_Instance_RemoteAet)); | |
134 ASSERT_EQ("None", index.GetMetadata(a[4], MetadataType_Instance_IndexInSeries, "None")); | |
135 | |
188
090cefdab1d1
fix because of Windows macros
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
185
diff
changeset
|
136 ASSERT_TRUE(index.LookupGlobalProperty(s, "Hello")); |
090cefdab1d1
fix because of Windows macros
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
185
diff
changeset
|
137 ASSERT_FALSE(index.LookupGlobalProperty(s, "Hello2")); |
181 | 138 ASSERT_EQ("World", s); |
139 ASSERT_EQ("World", index.GetGlobalProperty("Hello")); | |
140 ASSERT_EQ("None", index.GetGlobalProperty("Hello2", "None")); | |
141 | |
188
090cefdab1d1
fix because of Windows macros
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
185
diff
changeset
|
142 uint64_t us, cs; |
181 | 143 CompressionType ct; |
188
090cefdab1d1
fix because of Windows macros
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
185
diff
changeset
|
144 ASSERT_TRUE(index.LookupFile(a[4], "_json", s, cs, us, ct)); |
181 | 145 ASSERT_EQ("my json file", s); |
183 | 146 ASSERT_EQ(21, cs); |
181 | 147 ASSERT_EQ(42, us); |
148 ASSERT_EQ(CompressionType_Zlib, ct); | |
149 | |
188
090cefdab1d1
fix because of Windows macros
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
185
diff
changeset
|
150 ASSERT_EQ(0u, listener.deletedFiles_.size()); |
181 | 151 ASSERT_EQ(7, index.GetTableRecordCount("Resources")); |
183 | 152 ASSERT_EQ(3, index.GetTableRecordCount("AttachedFiles")); |
181 | 153 ASSERT_EQ(1, index.GetTableRecordCount("Metadata")); |
154 ASSERT_EQ(1, index.GetTableRecordCount("MainDicomTags")); | |
155 index.DeleteResource(a[0]); | |
183 | 156 |
157 ASSERT_EQ(2, listener.deletedFiles_.size()); | |
185 | 158 ASSERT_FALSE(listener.deletedFiles_.find("my json file") == listener.deletedFiles_.end()); |
159 ASSERT_FALSE(listener.deletedFiles_.find("my dicom file") == listener.deletedFiles_.end()); | |
183 | 160 |
181 | 161 ASSERT_EQ(2, index.GetTableRecordCount("Resources")); |
162 ASSERT_EQ(0, index.GetTableRecordCount("Metadata")); | |
183 | 163 ASSERT_EQ(1, index.GetTableRecordCount("AttachedFiles")); |
164 ASSERT_EQ(0, index.GetTableRecordCount("MainDicomTags")); | |
165 index.DeleteResource(a[5]); | |
166 ASSERT_EQ(0, index.GetTableRecordCount("Resources")); | |
181 | 167 ASSERT_EQ(0, index.GetTableRecordCount("AttachedFiles")); |
183 | 168 ASSERT_EQ(1, index.GetTableRecordCount("GlobalProperties")); |
169 | |
170 ASSERT_EQ(3, listener.deletedFiles_.size()); | |
185 | 171 ASSERT_FALSE(listener.deletedFiles_.find("world") == listener.deletedFiles_.end()); |
183 | 172 } |
173 | |
174 | |
175 | |
176 | |
177 TEST(DatabaseWrapper, Upward) | |
178 { | |
179 ServerIndexListener listener; | |
180 DatabaseWrapper index(listener); | |
181 | |
182 int64_t a[] = { | |
183 index.CreateResource("a", ResourceType_Patient), // 0 | |
184 index.CreateResource("b", ResourceType_Study), // 1 | |
185 index.CreateResource("c", ResourceType_Series), // 2 | |
186 index.CreateResource("d", ResourceType_Instance), // 3 | |
187 index.CreateResource("e", ResourceType_Instance), // 4 | |
188 index.CreateResource("f", ResourceType_Study), // 5 | |
189 index.CreateResource("g", ResourceType_Series), // 6 | |
190 index.CreateResource("h", ResourceType_Series) // 7 | |
191 }; | |
192 | |
193 index.AttachChild(a[0], a[1]); | |
194 index.AttachChild(a[1], a[2]); | |
195 index.AttachChild(a[2], a[3]); | |
196 index.AttachChild(a[2], a[4]); | |
197 index.AttachChild(a[1], a[6]); | |
198 index.AttachChild(a[0], a[5]); | |
199 index.AttachChild(a[5], a[7]); | |
200 | |
201 listener.Reset(); | |
202 index.DeleteResource(a[3]); | |
203 ASSERT_EQ("c", listener.ancestorId_); | |
204 ASSERT_EQ(ResourceType_Series, listener.ancestorType_); | |
205 | |
206 listener.Reset(); | |
207 index.DeleteResource(a[4]); | |
208 ASSERT_EQ("b", listener.ancestorId_); | |
209 ASSERT_EQ(ResourceType_Study, listener.ancestorType_); | |
210 | |
211 listener.Reset(); | |
212 index.DeleteResource(a[7]); | |
213 ASSERT_EQ("a", listener.ancestorId_); | |
214 ASSERT_EQ(ResourceType_Patient, listener.ancestorType_); | |
215 | |
216 listener.Reset(); | |
181 | 217 index.DeleteResource(a[6]); |
183 | 218 ASSERT_EQ("", listener.ancestorId_); // No more ancestor |
181 | 219 } |