Mercurial > hg > orthanc-databases
annotate MySQL/UnitTests/UnitTestsMain.cpp @ 17:54ea251aed70
unit test
| author | Sebastien Jodogne <s.jodogne@gmail.com> |
|---|---|
| date | Tue, 10 Jul 2018 10:33:02 +0200 |
| parents | 9e419261f1c9 |
| children | 1e9bad493475 |
| rev | line source |
|---|---|
| 0 | 1 /** |
| 2 * Orthanc - A Lightweight, RESTful DICOM Store | |
| 3 * Copyright (C) 2012-2016 Sebastien Jodogne, Medical Physics | |
| 4 * Department, University Hospital of Liege, Belgium | |
| 5 * Copyright (C) 2017-2018 Osimis S.A., Belgium | |
| 6 * | |
| 7 * This program is free software: you can redistribute it and/or | |
| 8 * modify it under the terms of the GNU Affero General Public License | |
| 9 * as published by the Free Software Foundation, either version 3 of | |
| 10 * the License, or (at your option) any later version. | |
| 11 * | |
| 12 * This program is distributed in the hope that it will be useful, but | |
| 13 * WITHOUT ANY WARRANTY; without even the implied warranty of | |
| 14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | |
| 15 * Affero General Public License for more details. | |
| 16 * | |
| 17 * You should have received a copy of the GNU Affero General Public License | |
| 18 * along with this program. If not, see <http://www.gnu.org/licenses/>. | |
| 19 **/ | |
| 20 | |
| 21 | |
| 22 #include "../Plugins/MySQLIndex.h" | |
| 17 | 23 #include "../Plugins/MySQLStorageArea.h" |
| 0 | 24 |
| 25 OrthancDatabases::MySQLParameters globalParameters_; | |
| 26 | |
| 17 | 27 #include "../../Framework/Common/Integer64Value.h" |
| 28 #include "../../Framework/MySQL/MySQLDatabase.h" | |
| 29 #include "../../Framework/MySQL/MySQLResult.h" | |
| 30 #include "../../Framework/MySQL/MySQLStatement.h" | |
| 0 | 31 #include "../../Framework/Plugins/IndexUnitTests.h" |
| 32 | |
| 33 #include <Core/Logging.h> | |
| 34 | |
| 35 #include <gtest/gtest.h> | |
| 36 | |
| 37 | |
| 38 TEST(MySQLIndex, Lock) | |
| 39 { | |
| 40 OrthancDatabases::MySQLParameters noLock = globalParameters_; | |
| 41 noLock.SetLock(false); | |
| 42 | |
| 43 OrthancDatabases::MySQLParameters lock = globalParameters_; | |
| 44 lock.SetLock(true); | |
| 45 | |
| 46 OrthancDatabases::MySQLIndex db1(noLock); | |
| 47 db1.SetClearAll(true); | |
| 48 db1.Open(); | |
| 49 | |
| 50 { | |
| 51 OrthancDatabases::MySQLIndex db2(lock); | |
| 52 db2.Open(); | |
| 53 | |
| 54 OrthancDatabases::MySQLIndex db3(lock); | |
| 55 ASSERT_THROW(db3.Open(), Orthanc::OrthancException); | |
| 56 } | |
| 57 | |
| 58 OrthancDatabases::MySQLIndex db4(lock); | |
| 59 db4.Open(); | |
| 60 } | |
| 61 | |
| 62 | |
| 17 | 63 static int64_t CountFiles(OrthancDatabases::MySQLDatabase& db) |
| 64 { | |
| 65 OrthancDatabases::Query query("SELECT COUNT(*) FROM StorageArea", true); | |
| 66 OrthancDatabases::MySQLStatement s(db, query); | |
| 67 OrthancDatabases::MySQLTransaction t(db); | |
| 68 OrthancDatabases::Dictionary d; | |
| 69 std::auto_ptr<OrthancDatabases::IResult> result(s.Execute(t, d)); | |
| 70 return dynamic_cast<const OrthancDatabases::Integer64Value&>(result->GetField(0)).GetValue(); | |
| 71 } | |
| 72 | |
| 73 | |
| 74 TEST(MySQL, StorageArea) | |
| 75 { | |
| 76 OrthancDatabases::MySQLStorageArea storageArea(globalParameters_); | |
| 77 storageArea.SetClearAll(true); | |
| 78 | |
| 79 { | |
| 80 OrthancDatabases::DatabaseManager::Transaction transaction(storageArea.GetManager()); | |
| 81 OrthancDatabases::MySQLDatabase& db = | |
| 82 dynamic_cast<OrthancDatabases::MySQLDatabase&>(transaction.GetDatabase()); | |
| 83 | |
| 84 ASSERT_EQ(0, CountFiles(db)); | |
| 85 | |
| 86 for (int i = 0; i < 10; i++) | |
| 87 { | |
| 88 std::string uuid = boost::lexical_cast<std::string>(i); | |
| 89 std::string value = "Value " + boost::lexical_cast<std::string>(i * 2); | |
| 90 storageArea.Create(transaction, uuid, value.c_str(), value.size(), OrthancPluginContentType_Unknown); | |
| 91 } | |
| 92 | |
| 93 std::string tmp; | |
| 94 ASSERT_THROW(storageArea.ReadToString(tmp, transaction, "nope", OrthancPluginContentType_Unknown), | |
| 95 Orthanc::OrthancException); | |
| 96 | |
| 97 ASSERT_EQ(10, CountFiles(db)); | |
| 98 storageArea.Remove(transaction, "5", OrthancPluginContentType_Unknown); | |
| 99 | |
| 100 ASSERT_EQ(9, CountFiles(db)); | |
| 101 | |
| 102 for (int i = 0; i < 10; i++) | |
| 103 { | |
| 104 std::string uuid = boost::lexical_cast<std::string>(i); | |
| 105 std::string expected = "Value " + boost::lexical_cast<std::string>(i * 2); | |
| 106 std::string content; | |
| 107 | |
| 108 if (i == 5) | |
| 109 { | |
| 110 ASSERT_THROW(storageArea.ReadToString(content, transaction, uuid, OrthancPluginContentType_Unknown), | |
| 111 Orthanc::OrthancException); | |
| 112 } | |
| 113 else | |
| 114 { | |
| 115 storageArea.ReadToString(content, transaction, uuid, OrthancPluginContentType_Unknown); | |
| 116 ASSERT_EQ(expected, content); | |
| 117 } | |
| 118 } | |
| 119 | |
| 120 for (int i = 0; i < 10; i++) | |
| 121 { | |
| 122 storageArea.Remove(transaction, boost::lexical_cast<std::string>(i), | |
| 123 OrthancPluginContentType_Unknown); | |
| 124 } | |
| 125 | |
| 126 ASSERT_EQ(0, CountFiles(db)); | |
| 127 | |
| 128 transaction.Commit(); | |
| 129 } | |
| 130 } | |
| 131 | |
| 132 | |
| 0 | 133 int main(int argc, char **argv) |
| 134 { | |
| 135 if (argc < 5) | |
| 136 { | |
|
16
9e419261f1c9
mysql storage area working
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
0
diff
changeset
|
137 std::cerr << "Usage (UNIX): " << argv[0] << " <socket> <username> <password> <database>" |
|
9e419261f1c9
mysql storage area working
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
0
diff
changeset
|
138 << std::endl |
|
9e419261f1c9
mysql storage area working
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
0
diff
changeset
|
139 << "Usage (Windows): " << argv[0] << " <host> <port> <username> <password> <database>" |
| 0 | 140 << std::endl << std::endl |
|
16
9e419261f1c9
mysql storage area working
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
0
diff
changeset
|
141 << "Example (UNIX): " << argv[0] << " /var/run/mysqld/mysqld.sock root root orthanctest" |
|
9e419261f1c9
mysql storage area working
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
0
diff
changeset
|
142 << std::endl |
|
9e419261f1c9
mysql storage area working
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
0
diff
changeset
|
143 << "Example (Windows): " << argv[0] << " localhost 3306 root root orthanctest" |
| 0 | 144 << std::endl << std::endl; |
| 145 return -1; | |
| 146 } | |
| 147 | |
|
16
9e419261f1c9
mysql storage area working
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
0
diff
changeset
|
148 if (argc == 5) |
|
9e419261f1c9
mysql storage area working
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
0
diff
changeset
|
149 { |
|
9e419261f1c9
mysql storage area working
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
0
diff
changeset
|
150 // UNIX |
|
9e419261f1c9
mysql storage area working
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
0
diff
changeset
|
151 globalParameters_.SetUnixSocket(argv[1]); |
|
9e419261f1c9
mysql storage area working
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
0
diff
changeset
|
152 globalParameters_.SetUsername(argv[2]); |
|
9e419261f1c9
mysql storage area working
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
0
diff
changeset
|
153 globalParameters_.SetPassword(argv[3]); |
|
9e419261f1c9
mysql storage area working
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
0
diff
changeset
|
154 globalParameters_.SetDatabase(argv[4]); |
|
9e419261f1c9
mysql storage area working
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
0
diff
changeset
|
155 } |
|
9e419261f1c9
mysql storage area working
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
0
diff
changeset
|
156 else |
|
9e419261f1c9
mysql storage area working
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
0
diff
changeset
|
157 { |
|
9e419261f1c9
mysql storage area working
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
0
diff
changeset
|
158 // Windows |
|
9e419261f1c9
mysql storage area working
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
0
diff
changeset
|
159 globalParameters_.SetHost(argv[1]); |
|
9e419261f1c9
mysql storage area working
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
0
diff
changeset
|
160 globalParameters_.SetPort(boost::lexical_cast<unsigned int>(argv[2])); |
|
9e419261f1c9
mysql storage area working
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
0
diff
changeset
|
161 globalParameters_.SetUsername(argv[3]); |
|
9e419261f1c9
mysql storage area working
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
0
diff
changeset
|
162 globalParameters_.SetPassword(argv[4]); |
|
9e419261f1c9
mysql storage area working
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
0
diff
changeset
|
163 globalParameters_.SetDatabase(argv[5]); |
|
9e419261f1c9
mysql storage area working
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
0
diff
changeset
|
164 } |
| 0 | 165 |
| 166 ::testing::InitGoogleTest(&argc, argv); | |
| 167 Orthanc::Logging::Initialize(); | |
| 168 Orthanc::Logging::EnableInfoLevel(true); | |
| 169 Orthanc::Logging::EnableTraceLevel(true); | |
| 170 | |
| 171 int result = RUN_ALL_TESTS(); | |
| 172 | |
| 173 Orthanc::Logging::Finalize(); | |
| 174 | |
| 175 OrthancDatabases::MySQLDatabase::GlobalFinalization(); | |
| 176 | |
| 177 return result; | |
| 178 } |