|
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;
|
|
|
108 ASSERT_TRUE(index.FindResource("g", b, t));
|
|
|
109 ASSERT_EQ(7, b);
|
|
|
110 ASSERT_EQ(ResourceType_Study, t);
|
|
|
111
|
|
|
112 ASSERT_TRUE(index.FindMetadata(s, a[4], MetadataType_Instance_RemoteAet));
|
|
|
113 ASSERT_FALSE(index.FindMetadata(s, a[4], MetadataType_Instance_IndexInSeries));
|
|
|
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
|
|
|
118 ASSERT_TRUE(index.FindGlobalProperty(s, "Hello"));
|
|
|
119 ASSERT_FALSE(index.FindGlobalProperty(s, "Hello2"));
|
|
|
120 ASSERT_EQ("World", s);
|
|
|
121 ASSERT_EQ("World", index.GetGlobalProperty("Hello"));
|
|
|
122 ASSERT_EQ("None", index.GetGlobalProperty("Hello2", "None"));
|
|
|
123
|
|
183
|
124 size_t us, cs;
|
|
181
|
125 CompressionType ct;
|
|
183
|
126 ASSERT_TRUE(index.FindFile(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
|
|
183
|
132 ASSERT_EQ(0, 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 }
|
