Mercurial > hg > orthanc
annotate UnitTests/ServerIndex.cpp @ 188:090cefdab1d1
fix because of Windows macros
author | Sebastien Jodogne <s.jodogne@gmail.com> |
---|---|
date | Mon, 12 Nov 2012 17:43:12 +0100 |
parents | 5185ae1d9af5 |
children | b6cef9d45cc3 |
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 | |
57 index.SetGlobalProperty("Hello", "World"); | |
58 | |
59 index.AttachChild(a[0], a[1]); | |
60 index.AttachChild(a[1], a[2]); | |
61 index.AttachChild(a[2], a[3]); | |
62 index.AttachChild(a[2], a[4]); | |
63 index.AttachChild(a[6], a[5]); | |
182 | 64 |
65 std::string s; | |
66 | |
67 ASSERT_FALSE(index.GetParentPublicId(s, a[0])); | |
68 ASSERT_FALSE(index.GetParentPublicId(s, a[6])); | |
69 ASSERT_TRUE(index.GetParentPublicId(s, a[1])); ASSERT_EQ("a", s); | |
70 ASSERT_TRUE(index.GetParentPublicId(s, a[2])); ASSERT_EQ("b", s); | |
71 ASSERT_TRUE(index.GetParentPublicId(s, a[3])); ASSERT_EQ("c", s); | |
72 ASSERT_TRUE(index.GetParentPublicId(s, a[4])); ASSERT_EQ("c", s); | |
73 ASSERT_TRUE(index.GetParentPublicId(s, a[5])); ASSERT_EQ("g", s); | |
74 | |
185 | 75 std::list<std::string> l; |
182 | 76 index.GetChildrenPublicId(l, a[0]); ASSERT_EQ(1, l.size()); ASSERT_EQ("b", l.front()); |
77 index.GetChildrenPublicId(l, a[1]); ASSERT_EQ(1, l.size()); ASSERT_EQ("c", l.front()); | |
78 index.GetChildrenPublicId(l, a[3]); ASSERT_EQ(0, l.size()); | |
79 index.GetChildrenPublicId(l, a[4]); ASSERT_EQ(0, l.size()); | |
80 index.GetChildrenPublicId(l, a[5]); ASSERT_EQ(0, l.size()); | |
81 index.GetChildrenPublicId(l, a[6]); ASSERT_EQ(1, l.size()); ASSERT_EQ("f", l.front()); | |
82 | |
83 index.GetChildrenPublicId(l, a[2]); ASSERT_EQ(2, l.size()); | |
84 if (l.front() == "d") | |
85 { | |
86 ASSERT_EQ("e", l.back()); | |
87 } | |
88 else | |
89 { | |
90 ASSERT_EQ("d", l.back()); | |
91 ASSERT_EQ("e", l.front()); | |
92 } | |
93 | |
183 | 94 index.AttachFile(a[4], "_json", "my json file", 21, 42, CompressionType_Zlib); |
95 index.AttachFile(a[4], "_dicom", "my dicom file", 42); | |
96 index.AttachFile(a[6], "_hello", "world", 44); | |
97 index.SetMetadata(a[4], MetadataType_Instance_RemoteAet, "PINNACLE"); | |
182 | 98 |
183 | 99 ASSERT_EQ(21 + 42 + 44, index.GetTotalCompressedSize()); |
100 ASSERT_EQ(42 + 42 + 44, index.GetTotalUncompressedSize()); | |
181 | 101 |
102 DicomMap m; | |
103 m.SetValue(0x0010, 0x0010, "PatientName"); | |
104 index.SetMainDicomTags(a[3], m); | |
105 | |
106 int64_t b; | |
107 ResourceType t; | |
188
090cefdab1d1
fix because of Windows macros
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
185
diff
changeset
|
108 ASSERT_TRUE(index.LookupResource("g", b, t)); |
181 | 109 ASSERT_EQ(7, b); |
110 ASSERT_EQ(ResourceType_Study, t); | |
111 | |
188
090cefdab1d1
fix because of Windows macros
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
185
diff
changeset
|
112 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
|
113 ASSERT_FALSE(index.LookupMetadata(s, a[4], MetadataType_Instance_IndexInSeries)); |
181 | 114 ASSERT_EQ("PINNACLE", s); |
115 ASSERT_EQ("PINNACLE", index.GetMetadata(a[4], MetadataType_Instance_RemoteAet)); | |
116 ASSERT_EQ("None", index.GetMetadata(a[4], MetadataType_Instance_IndexInSeries, "None")); | |
117 | |
188
090cefdab1d1
fix because of Windows macros
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
185
diff
changeset
|
118 ASSERT_TRUE(index.LookupGlobalProperty(s, "Hello")); |
090cefdab1d1
fix because of Windows macros
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
185
diff
changeset
|
119 ASSERT_FALSE(index.LookupGlobalProperty(s, "Hello2")); |
181 | 120 ASSERT_EQ("World", s); |
121 ASSERT_EQ("World", index.GetGlobalProperty("Hello")); | |
122 ASSERT_EQ("None", index.GetGlobalProperty("Hello2", "None")); | |
123 | |
188
090cefdab1d1
fix because of Windows macros
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
185
diff
changeset
|
124 uint64_t us, cs; |
181 | 125 CompressionType ct; |
188
090cefdab1d1
fix because of Windows macros
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
185
diff
changeset
|
126 ASSERT_TRUE(index.LookupFile(a[4], "_json", s, cs, us, ct)); |
181 | 127 ASSERT_EQ("my json file", s); |
183 | 128 ASSERT_EQ(21, cs); |
181 | 129 ASSERT_EQ(42, us); |
130 ASSERT_EQ(CompressionType_Zlib, ct); | |
131 | |
188
090cefdab1d1
fix because of Windows macros
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
185
diff
changeset
|
132 ASSERT_EQ(0u, listener.deletedFiles_.size()); |
181 | 133 ASSERT_EQ(7, index.GetTableRecordCount("Resources")); |
183 | 134 ASSERT_EQ(3, index.GetTableRecordCount("AttachedFiles")); |
181 | 135 ASSERT_EQ(1, index.GetTableRecordCount("Metadata")); |
136 ASSERT_EQ(1, index.GetTableRecordCount("MainDicomTags")); | |
137 index.DeleteResource(a[0]); | |
183 | 138 |
139 ASSERT_EQ(2, listener.deletedFiles_.size()); | |
185 | 140 ASSERT_FALSE(listener.deletedFiles_.find("my json file") == listener.deletedFiles_.end()); |
141 ASSERT_FALSE(listener.deletedFiles_.find("my dicom file") == listener.deletedFiles_.end()); | |
183 | 142 |
181 | 143 ASSERT_EQ(2, index.GetTableRecordCount("Resources")); |
144 ASSERT_EQ(0, index.GetTableRecordCount("Metadata")); | |
183 | 145 ASSERT_EQ(1, index.GetTableRecordCount("AttachedFiles")); |
146 ASSERT_EQ(0, index.GetTableRecordCount("MainDicomTags")); | |
147 index.DeleteResource(a[5]); | |
148 ASSERT_EQ(0, index.GetTableRecordCount("Resources")); | |
181 | 149 ASSERT_EQ(0, index.GetTableRecordCount("AttachedFiles")); |
183 | 150 ASSERT_EQ(1, index.GetTableRecordCount("GlobalProperties")); |
151 | |
152 ASSERT_EQ(3, listener.deletedFiles_.size()); | |
185 | 153 ASSERT_FALSE(listener.deletedFiles_.find("world") == listener.deletedFiles_.end()); |
183 | 154 } |
155 | |
156 | |
157 | |
158 | |
159 TEST(DatabaseWrapper, Upward) | |
160 { | |
161 ServerIndexListener listener; | |
162 DatabaseWrapper index(listener); | |
163 | |
164 int64_t a[] = { | |
165 index.CreateResource("a", ResourceType_Patient), // 0 | |
166 index.CreateResource("b", ResourceType_Study), // 1 | |
167 index.CreateResource("c", ResourceType_Series), // 2 | |
168 index.CreateResource("d", ResourceType_Instance), // 3 | |
169 index.CreateResource("e", ResourceType_Instance), // 4 | |
170 index.CreateResource("f", ResourceType_Study), // 5 | |
171 index.CreateResource("g", ResourceType_Series), // 6 | |
172 index.CreateResource("h", ResourceType_Series) // 7 | |
173 }; | |
174 | |
175 index.AttachChild(a[0], a[1]); | |
176 index.AttachChild(a[1], a[2]); | |
177 index.AttachChild(a[2], a[3]); | |
178 index.AttachChild(a[2], a[4]); | |
179 index.AttachChild(a[1], a[6]); | |
180 index.AttachChild(a[0], a[5]); | |
181 index.AttachChild(a[5], a[7]); | |
182 | |
183 listener.Reset(); | |
184 index.DeleteResource(a[3]); | |
185 ASSERT_EQ("c", listener.ancestorId_); | |
186 ASSERT_EQ(ResourceType_Series, listener.ancestorType_); | |
187 | |
188 listener.Reset(); | |
189 index.DeleteResource(a[4]); | |
190 ASSERT_EQ("b", listener.ancestorId_); | |
191 ASSERT_EQ(ResourceType_Study, listener.ancestorType_); | |
192 | |
193 listener.Reset(); | |
194 index.DeleteResource(a[7]); | |
195 ASSERT_EQ("a", listener.ancestorId_); | |
196 ASSERT_EQ(ResourceType_Patient, listener.ancestorType_); | |
197 | |
198 listener.Reset(); | |
181 | 199 index.DeleteResource(a[6]); |
183 | 200 ASSERT_EQ("", listener.ancestorId_); // No more ancestor |
181 | 201 } |