Mercurial > hg > orthanc
view UnitTests/ServerIndex.cpp @ 555:11fee43c5861
fix
author | Sebastien Jodogne <s.jodogne@gmail.com> |
---|---|
date | Tue, 17 Sep 2013 16:19:36 +0200 |
parents | 2c739f76d0bb |
children | f64e3838d6e1 |
line wrap: on
line source
#include "gtest/gtest.h" #include "../OrthancServer/DatabaseWrapper.h" #include "../Core/Uuid.h" #include <ctype.h> #include <glog/logging.h> #include <algorithm> using namespace Orthanc; namespace { class ServerIndexListener : public IServerIndexListener { public: std::vector<std::string> deletedFiles_; std::string ancestorId_; ResourceType ancestorType_; void Reset() { ancestorId_ = ""; deletedFiles_.clear(); } virtual void SignalRemainingAncestor(ResourceType type, const std::string& publicId) { ancestorId_ = publicId; ancestorType_ = type; } virtual void SignalFileDeleted(const FileInfo& info) { const std::string fileUuid = info.GetUuid(); deletedFiles_.push_back(fileUuid); LOG(INFO) << "A file must be removed: " << fileUuid; } }; } TEST(DatabaseWrapper, Simple) { ServerIndexListener listener; DatabaseWrapper index(listener); int64_t a[] = { index.CreateResource("a", ResourceType_Patient), // 0 index.CreateResource("b", ResourceType_Study), // 1 index.CreateResource("c", ResourceType_Series), // 2 index.CreateResource("d", ResourceType_Instance), // 3 index.CreateResource("e", ResourceType_Instance), // 4 index.CreateResource("f", ResourceType_Instance), // 5 index.CreateResource("g", ResourceType_Study) // 6 }; ASSERT_EQ("a", index.GetPublicId(a[0])); ASSERT_EQ("b", index.GetPublicId(a[1])); ASSERT_EQ("c", index.GetPublicId(a[2])); ASSERT_EQ("d", index.GetPublicId(a[3])); ASSERT_EQ("e", index.GetPublicId(a[4])); ASSERT_EQ("f", index.GetPublicId(a[5])); ASSERT_EQ("g", index.GetPublicId(a[6])); ASSERT_EQ(ResourceType_Patient, index.GetResourceType(a[0])); ASSERT_EQ(ResourceType_Study, index.GetResourceType(a[1])); ASSERT_EQ(ResourceType_Series, index.GetResourceType(a[2])); ASSERT_EQ(ResourceType_Instance, index.GetResourceType(a[3])); ASSERT_EQ(ResourceType_Instance, index.GetResourceType(a[4])); ASSERT_EQ(ResourceType_Instance, index.GetResourceType(a[5])); ASSERT_EQ(ResourceType_Study, index.GetResourceType(a[6])); { Json::Value t; index.GetAllPublicIds(t, ResourceType_Patient); ASSERT_EQ(1u, t.size()); ASSERT_EQ("a", t[0u].asString()); index.GetAllPublicIds(t, ResourceType_Series); ASSERT_EQ(1u, t.size()); ASSERT_EQ("c", t[0u].asString()); index.GetAllPublicIds(t, ResourceType_Study); ASSERT_EQ(2u, t.size()); index.GetAllPublicIds(t, ResourceType_Instance); ASSERT_EQ(3u, t.size()); } index.SetGlobalProperty(GlobalProperty_FlushSleep, "World"); index.AttachChild(a[0], a[1]); index.AttachChild(a[1], a[2]); index.AttachChild(a[2], a[3]); index.AttachChild(a[2], a[4]); index.AttachChild(a[6], a[5]); int64_t parent; ASSERT_FALSE(index.LookupParent(parent, a[0])); ASSERT_TRUE(index.LookupParent(parent, a[1])); ASSERT_EQ(a[0], parent); ASSERT_TRUE(index.LookupParent(parent, a[2])); ASSERT_EQ(a[1], parent); ASSERT_TRUE(index.LookupParent(parent, a[3])); ASSERT_EQ(a[2], parent); ASSERT_TRUE(index.LookupParent(parent, a[4])); ASSERT_EQ(a[2], parent); ASSERT_TRUE(index.LookupParent(parent, a[5])); ASSERT_EQ(a[6], parent); ASSERT_FALSE(index.LookupParent(parent, a[6])); std::string s; ASSERT_FALSE(index.GetParentPublicId(s, a[0])); ASSERT_FALSE(index.GetParentPublicId(s, a[6])); ASSERT_TRUE(index.GetParentPublicId(s, a[1])); ASSERT_EQ("a", s); ASSERT_TRUE(index.GetParentPublicId(s, a[2])); ASSERT_EQ("b", s); ASSERT_TRUE(index.GetParentPublicId(s, a[3])); ASSERT_EQ("c", s); ASSERT_TRUE(index.GetParentPublicId(s, a[4])); ASSERT_EQ("c", s); ASSERT_TRUE(index.GetParentPublicId(s, a[5])); ASSERT_EQ("g", s); std::list<std::string> l; index.GetChildrenPublicId(l, a[0]); ASSERT_EQ(1u, l.size()); ASSERT_EQ("b", l.front()); index.GetChildrenPublicId(l, a[1]); ASSERT_EQ(1u, l.size()); ASSERT_EQ("c", l.front()); index.GetChildrenPublicId(l, a[3]); ASSERT_EQ(0u, l.size()); index.GetChildrenPublicId(l, a[4]); ASSERT_EQ(0u, l.size()); index.GetChildrenPublicId(l, a[5]); ASSERT_EQ(0u, l.size()); index.GetChildrenPublicId(l, a[6]); ASSERT_EQ(1u, l.size()); ASSERT_EQ("f", l.front()); index.GetChildrenPublicId(l, a[2]); ASSERT_EQ(2u, l.size()); if (l.front() == "d") { ASSERT_EQ("e", l.back()); } else { ASSERT_EQ("d", l.back()); ASSERT_EQ("e", l.front()); } std::list<MetadataType> md; index.ListAvailableMetadata(md, a[4]); ASSERT_EQ(0u, md.size()); index.AddAttachment(a[4], FileInfo("my json file", FileContentType_Json, 42, CompressionType_Zlib, 21)); index.AddAttachment(a[4], FileInfo("my dicom file", FileContentType_Dicom, 42)); index.AddAttachment(a[6], FileInfo("world", FileContentType_Dicom, 44)); index.SetMetadata(a[4], MetadataType_Instance_RemoteAet, "PINNACLE"); index.ListAvailableMetadata(md, a[4]); ASSERT_EQ(1u, md.size()); ASSERT_EQ(MetadataType_Instance_RemoteAet, md.front()); index.SetMetadata(a[4], MetadataType_ModifiedFrom, "TUTU"); index.ListAvailableMetadata(md, a[4]); ASSERT_EQ(2u, md.size()); index.DeleteMetadata(a[4], MetadataType_ModifiedFrom); index.ListAvailableMetadata(md, a[4]); ASSERT_EQ(1u, md.size()); ASSERT_EQ(MetadataType_Instance_RemoteAet, md.front()); ASSERT_EQ(21u + 42u + 44u, index.GetTotalCompressedSize()); ASSERT_EQ(42u + 42u + 44u, index.GetTotalUncompressedSize()); DicomMap m; m.SetValue(0x0010, 0x0010, "PatientName"); index.SetMainDicomTags(a[3], m); int64_t b; ResourceType t; ASSERT_TRUE(index.LookupResource("g", b, t)); ASSERT_EQ(7, b); ASSERT_EQ(ResourceType_Study, t); ASSERT_TRUE(index.LookupMetadata(s, a[4], MetadataType_Instance_RemoteAet)); ASSERT_FALSE(index.LookupMetadata(s, a[4], MetadataType_Instance_IndexInSeries)); ASSERT_EQ("PINNACLE", s); ASSERT_EQ("PINNACLE", index.GetMetadata(a[4], MetadataType_Instance_RemoteAet)); ASSERT_EQ("None", index.GetMetadata(a[4], MetadataType_Instance_IndexInSeries, "None")); ASSERT_TRUE(index.LookupGlobalProperty(s, GlobalProperty_FlushSleep)); ASSERT_FALSE(index.LookupGlobalProperty(s, static_cast<GlobalProperty>(42))); ASSERT_EQ("World", s); ASSERT_EQ("World", index.GetGlobalProperty(GlobalProperty_FlushSleep)); ASSERT_EQ("None", index.GetGlobalProperty(static_cast<GlobalProperty>(42), "None")); FileInfo att; ASSERT_TRUE(index.LookupAttachment(att, a[4], FileContentType_Json)); ASSERT_EQ("my json file", att.GetUuid()); ASSERT_EQ(21u, att.GetCompressedSize()); ASSERT_EQ(42u, att.GetUncompressedSize()); ASSERT_EQ(CompressionType_Zlib, att.GetCompressionType()); ASSERT_EQ(0u, listener.deletedFiles_.size()); ASSERT_EQ(7u, index.GetTableRecordCount("Resources")); ASSERT_EQ(3u, index.GetTableRecordCount("AttachedFiles")); ASSERT_EQ(1u, index.GetTableRecordCount("Metadata")); ASSERT_EQ(1u, index.GetTableRecordCount("MainDicomTags")); index.DeleteResource(a[0]); ASSERT_EQ(2u, listener.deletedFiles_.size()); ASSERT_FALSE(std::find(listener.deletedFiles_.begin(), listener.deletedFiles_.end(), "my json file") == listener.deletedFiles_.end()); ASSERT_FALSE(std::find(listener.deletedFiles_.begin(), listener.deletedFiles_.end(), "my dicom file") == listener.deletedFiles_.end()); ASSERT_EQ(2u, index.GetTableRecordCount("Resources")); ASSERT_EQ(0u, index.GetTableRecordCount("Metadata")); ASSERT_EQ(1u, index.GetTableRecordCount("AttachedFiles")); ASSERT_EQ(0u, index.GetTableRecordCount("MainDicomTags")); index.DeleteResource(a[5]); ASSERT_EQ(0u, index.GetTableRecordCount("Resources")); ASSERT_EQ(0u, index.GetTableRecordCount("AttachedFiles")); ASSERT_EQ(2u, index.GetTableRecordCount("GlobalProperties")); ASSERT_EQ(3u, listener.deletedFiles_.size()); ASSERT_FALSE(std::find(listener.deletedFiles_.begin(), listener.deletedFiles_.end(), "world") == listener.deletedFiles_.end()); } TEST(DatabaseWrapper, Upward) { ServerIndexListener listener; DatabaseWrapper index(listener); int64_t a[] = { index.CreateResource("a", ResourceType_Patient), // 0 index.CreateResource("b", ResourceType_Study), // 1 index.CreateResource("c", ResourceType_Series), // 2 index.CreateResource("d", ResourceType_Instance), // 3 index.CreateResource("e", ResourceType_Instance), // 4 index.CreateResource("f", ResourceType_Study), // 5 index.CreateResource("g", ResourceType_Series), // 6 index.CreateResource("h", ResourceType_Series) // 7 }; index.AttachChild(a[0], a[1]); index.AttachChild(a[1], a[2]); index.AttachChild(a[2], a[3]); index.AttachChild(a[2], a[4]); index.AttachChild(a[1], a[6]); index.AttachChild(a[0], a[5]); index.AttachChild(a[5], a[7]); { Json::Value j; index.GetChildren(j, a[0]); ASSERT_EQ(2u, j.size()); ASSERT_TRUE((j[0u] == "b" && j[1u] == "f") || (j[1u] == "b" && j[0u] == "f")); index.GetChildren(j, a[1]); ASSERT_EQ(2u, j.size()); ASSERT_TRUE((j[0u] == "c" && j[1u] == "g") || (j[1u] == "c" && j[0u] == "g")); index.GetChildren(j, a[2]); ASSERT_EQ(2u, j.size()); ASSERT_TRUE((j[0u] == "d" && j[1u] == "e") || (j[1u] == "d" && j[0u] == "e")); index.GetChildren(j, a[3]); ASSERT_EQ(0u, j.size()); index.GetChildren(j, a[4]); ASSERT_EQ(0u, j.size()); index.GetChildren(j, a[5]); ASSERT_EQ(1u, j.size()); ASSERT_EQ("h", j[0u].asString()); index.GetChildren(j, a[6]); ASSERT_EQ(0u, j.size()); index.GetChildren(j, a[7]); ASSERT_EQ(0u, j.size()); } listener.Reset(); index.DeleteResource(a[3]); ASSERT_EQ("c", listener.ancestorId_); ASSERT_EQ(ResourceType_Series, listener.ancestorType_); listener.Reset(); index.DeleteResource(a[4]); ASSERT_EQ("b", listener.ancestorId_); ASSERT_EQ(ResourceType_Study, listener.ancestorType_); listener.Reset(); index.DeleteResource(a[7]); ASSERT_EQ("a", listener.ancestorId_); ASSERT_EQ(ResourceType_Patient, listener.ancestorType_); listener.Reset(); index.DeleteResource(a[6]); ASSERT_EQ("", listener.ancestorId_); // No more ancestor } TEST(DatabaseWrapper, PatientRecycling) { ServerIndexListener listener; DatabaseWrapper index(listener); std::vector<int64_t> patients; for (int i = 0; i < 10; i++) { std::string p = "Patient " + boost::lexical_cast<std::string>(i); patients.push_back(index.CreateResource(p, ResourceType_Patient)); index.AddAttachment(patients[i], FileInfo(p, FileContentType_Dicom, i + 10)); ASSERT_FALSE(index.IsProtectedPatient(patients[i])); } ASSERT_EQ(10u, index.GetTableRecordCount("Resources")); ASSERT_EQ(10u, index.GetTableRecordCount("PatientRecyclingOrder")); listener.Reset(); index.DeleteResource(patients[5]); index.DeleteResource(patients[0]); ASSERT_EQ(8u, index.GetTableRecordCount("Resources")); ASSERT_EQ(8u, index.GetTableRecordCount("PatientRecyclingOrder")); ASSERT_EQ(2u, listener.deletedFiles_.size()); ASSERT_EQ("Patient 5", listener.deletedFiles_[0]); ASSERT_EQ("Patient 0", listener.deletedFiles_[1]); int64_t p; ASSERT_TRUE(index.SelectPatientToRecycle(p)); ASSERT_EQ(p, patients[1]); index.DeleteResource(p); ASSERT_TRUE(index.SelectPatientToRecycle(p)); ASSERT_EQ(p, patients[2]); index.DeleteResource(p); ASSERT_TRUE(index.SelectPatientToRecycle(p)); ASSERT_EQ(p, patients[3]); index.DeleteResource(p); ASSERT_TRUE(index.SelectPatientToRecycle(p)); ASSERT_EQ(p, patients[4]); index.DeleteResource(p); ASSERT_TRUE(index.SelectPatientToRecycle(p)); ASSERT_EQ(p, patients[6]); index.DeleteResource(p); index.DeleteResource(patients[8]); ASSERT_TRUE(index.SelectPatientToRecycle(p)); ASSERT_EQ(p, patients[7]); index.DeleteResource(p); ASSERT_TRUE(index.SelectPatientToRecycle(p)); ASSERT_EQ(p, patients[9]); index.DeleteResource(p); ASSERT_FALSE(index.SelectPatientToRecycle(p)); ASSERT_EQ(10u, listener.deletedFiles_.size()); ASSERT_EQ(0u, index.GetTableRecordCount("Resources")); ASSERT_EQ(0u, index.GetTableRecordCount("PatientRecyclingOrder")); } TEST(DatabaseWrapper, PatientProtection) { ServerIndexListener listener; DatabaseWrapper index(listener); std::vector<int64_t> patients; for (int i = 0; i < 5; i++) { std::string p = "Patient " + boost::lexical_cast<std::string>(i); patients.push_back(index.CreateResource(p, ResourceType_Patient)); index.AddAttachment(patients[i], FileInfo(p, FileContentType_Dicom, i + 10)); ASSERT_FALSE(index.IsProtectedPatient(patients[i])); } ASSERT_EQ(5u, index.GetTableRecordCount("Resources")); ASSERT_EQ(5u, index.GetTableRecordCount("PatientRecyclingOrder")); ASSERT_FALSE(index.IsProtectedPatient(patients[2])); index.SetProtectedPatient(patients[2], true); ASSERT_TRUE(index.IsProtectedPatient(patients[2])); ASSERT_EQ(4u, index.GetTableRecordCount("PatientRecyclingOrder")); ASSERT_EQ(5u, index.GetTableRecordCount("Resources")); index.SetProtectedPatient(patients[2], true); ASSERT_TRUE(index.IsProtectedPatient(patients[2])); ASSERT_EQ(4u, index.GetTableRecordCount("PatientRecyclingOrder")); index.SetProtectedPatient(patients[2], false); ASSERT_FALSE(index.IsProtectedPatient(patients[2])); ASSERT_EQ(5u, index.GetTableRecordCount("PatientRecyclingOrder")); index.SetProtectedPatient(patients[2], false); ASSERT_FALSE(index.IsProtectedPatient(patients[2])); ASSERT_EQ(5u, index.GetTableRecordCount("PatientRecyclingOrder")); ASSERT_EQ(5u, index.GetTableRecordCount("Resources")); ASSERT_EQ(5u, index.GetTableRecordCount("PatientRecyclingOrder")); index.SetProtectedPatient(patients[2], true); ASSERT_TRUE(index.IsProtectedPatient(patients[2])); ASSERT_EQ(4u, index.GetTableRecordCount("PatientRecyclingOrder")); index.SetProtectedPatient(patients[2], false); ASSERT_FALSE(index.IsProtectedPatient(patients[2])); ASSERT_EQ(5u, index.GetTableRecordCount("PatientRecyclingOrder")); index.SetProtectedPatient(patients[3], true); ASSERT_TRUE(index.IsProtectedPatient(patients[3])); ASSERT_EQ(4u, index.GetTableRecordCount("PatientRecyclingOrder")); ASSERT_EQ(5u, index.GetTableRecordCount("Resources")); ASSERT_EQ(0u, listener.deletedFiles_.size()); // Unprotecting a patient puts it at the last position in the recycling queue int64_t p; ASSERT_TRUE(index.SelectPatientToRecycle(p)); ASSERT_EQ(p, patients[0]); index.DeleteResource(p); ASSERT_TRUE(index.SelectPatientToRecycle(p, patients[1])); ASSERT_EQ(p, patients[4]); ASSERT_TRUE(index.SelectPatientToRecycle(p)); ASSERT_EQ(p, patients[1]); index.DeleteResource(p); ASSERT_TRUE(index.SelectPatientToRecycle(p)); ASSERT_EQ(p, patients[4]); index.DeleteResource(p); ASSERT_FALSE(index.SelectPatientToRecycle(p, patients[2])); ASSERT_TRUE(index.SelectPatientToRecycle(p)); ASSERT_EQ(p, patients[2]); index.DeleteResource(p); // "patients[3]" is still protected ASSERT_FALSE(index.SelectPatientToRecycle(p)); ASSERT_EQ(4u, listener.deletedFiles_.size()); ASSERT_EQ(1u, index.GetTableRecordCount("Resources")); ASSERT_EQ(0u, index.GetTableRecordCount("PatientRecyclingOrder")); index.SetProtectedPatient(patients[3], false); ASSERT_EQ(1u, index.GetTableRecordCount("PatientRecyclingOrder")); ASSERT_FALSE(index.SelectPatientToRecycle(p, patients[3])); ASSERT_TRUE(index.SelectPatientToRecycle(p, patients[2])); ASSERT_TRUE(index.SelectPatientToRecycle(p)); ASSERT_EQ(p, patients[3]); index.DeleteResource(p); ASSERT_EQ(5u, listener.deletedFiles_.size()); ASSERT_EQ(0u, index.GetTableRecordCount("Resources")); ASSERT_EQ(0u, index.GetTableRecordCount("PatientRecyclingOrder")); } TEST(DatabaseWrapper, Sequence) { ServerIndexListener listener; DatabaseWrapper index(listener); ASSERT_EQ(1u, index.IncrementGlobalSequence(GlobalProperty_AnonymizationSequence)); ASSERT_EQ(2u, index.IncrementGlobalSequence(GlobalProperty_AnonymizationSequence)); ASSERT_EQ(3u, index.IncrementGlobalSequence(GlobalProperty_AnonymizationSequence)); ASSERT_EQ(4u, index.IncrementGlobalSequence(GlobalProperty_AnonymizationSequence)); } TEST(DatabaseWrapper, LookupTagValue) { ServerIndexListener listener; DatabaseWrapper index(listener); int64_t a[] = { index.CreateResource("a", ResourceType_Study), // 0 index.CreateResource("b", ResourceType_Study), // 1 index.CreateResource("c", ResourceType_Study), // 2 index.CreateResource("d", ResourceType_Series) // 3 }; DicomMap m; m.Clear(); m.SetValue(DICOM_TAG_STUDY_INSTANCE_UID, "0"); index.SetMainDicomTags(a[0], m); m.Clear(); m.SetValue(DICOM_TAG_STUDY_INSTANCE_UID, "1"); index.SetMainDicomTags(a[1], m); m.Clear(); m.SetValue(DICOM_TAG_STUDY_INSTANCE_UID, "0"); index.SetMainDicomTags(a[2], m); m.Clear(); m.SetValue(DICOM_TAG_SERIES_INSTANCE_UID, "0"); index.SetMainDicomTags(a[3], m); std::list<int64_t> s; index.LookupTagValue(s, DICOM_TAG_STUDY_INSTANCE_UID, "0"); ASSERT_EQ(2u, s.size()); ASSERT_TRUE(std::find(s.begin(), s.end(), a[0]) != s.end()); ASSERT_TRUE(std::find(s.begin(), s.end(), a[2]) != s.end()); index.LookupTagValue(s, "0"); ASSERT_EQ(3u, s.size()); ASSERT_TRUE(std::find(s.begin(), s.end(), a[0]) != s.end()); ASSERT_TRUE(std::find(s.begin(), s.end(), a[2]) != s.end()); ASSERT_TRUE(std::find(s.begin(), s.end(), a[3]) != s.end()); index.LookupTagValue(s, DICOM_TAG_STUDY_INSTANCE_UID, "1"); ASSERT_EQ(1u, s.size()); ASSERT_TRUE(std::find(s.begin(), s.end(), a[1]) != s.end()); index.LookupTagValue(s, "1"); ASSERT_EQ(1u, s.size()); ASSERT_TRUE(std::find(s.begin(), s.end(), a[1]) != s.end()); /*{ std::list<std::string> s; context.GetIndex().LookupTagValue(s, DICOM_TAG_STUDY_INSTANCE_UID, "1.2.250.1.74.20130819132500.29000036381059"); for (std::list<std::string>::iterator i = s.begin(); i != s.end(); i++) { std::cout << "*** " << *i << std::endl;; } }*/ }