Mercurial > hg > orthanc-databases
annotate MySQL/UnitTests/UnitTestsMain.cpp @ 22:1e9bad493475
prevent running unit tests on a non-existing db
author | Sebastien Jodogne <s.jodogne@gmail.com> |
---|---|
date | Wed, 11 Jul 2018 14:39:59 +0200 |
parents | 54ea251aed70 |
children | b2ff1cd2907a |
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 | |
22
1e9bad493475
prevent running unit tests on a non-existing db
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
17
diff
changeset
|
148 std::vector<std::string> args; |
1e9bad493475
prevent running unit tests on a non-existing db
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
17
diff
changeset
|
149 for (int i = 1; i < argc; i++) |
16
9e419261f1c9
mysql storage area working
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
0
diff
changeset
|
150 { |
22
1e9bad493475
prevent running unit tests on a non-existing db
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
17
diff
changeset
|
151 // Ignore arguments beginning with "-" to allow passing arguments |
1e9bad493475
prevent running unit tests on a non-existing db
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
17
diff
changeset
|
152 // to Google Test such as "--gtest_filter=" |
1e9bad493475
prevent running unit tests on a non-existing db
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
17
diff
changeset
|
153 if (argv[i] != NULL && |
1e9bad493475
prevent running unit tests on a non-existing db
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
17
diff
changeset
|
154 argv[i][0] != '-') |
1e9bad493475
prevent running unit tests on a non-existing db
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
17
diff
changeset
|
155 { |
1e9bad493475
prevent running unit tests on a non-existing db
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
17
diff
changeset
|
156 args.push_back(std::string(argv[i])); |
1e9bad493475
prevent running unit tests on a non-existing db
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
17
diff
changeset
|
157 } |
16
9e419261f1c9
mysql storage area working
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
0
diff
changeset
|
158 } |
22
1e9bad493475
prevent running unit tests on a non-existing db
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
17
diff
changeset
|
159 |
0 | 160 ::testing::InitGoogleTest(&argc, argv); |
161 Orthanc::Logging::Initialize(); | |
162 Orthanc::Logging::EnableInfoLevel(true); | |
163 Orthanc::Logging::EnableTraceLevel(true); | |
22
1e9bad493475
prevent running unit tests on a non-existing db
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
17
diff
changeset
|
164 |
1e9bad493475
prevent running unit tests on a non-existing db
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
17
diff
changeset
|
165 if (args.size() == 4) |
1e9bad493475
prevent running unit tests on a non-existing db
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
17
diff
changeset
|
166 { |
1e9bad493475
prevent running unit tests on a non-existing db
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
17
diff
changeset
|
167 // UNIX flavor |
1e9bad493475
prevent running unit tests on a non-existing db
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
17
diff
changeset
|
168 globalParameters_.SetUnixSocket(args[0]); |
1e9bad493475
prevent running unit tests on a non-existing db
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
17
diff
changeset
|
169 globalParameters_.SetUsername(args[1]); |
1e9bad493475
prevent running unit tests on a non-existing db
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
17
diff
changeset
|
170 globalParameters_.SetPassword(args[2]); |
1e9bad493475
prevent running unit tests on a non-existing db
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
17
diff
changeset
|
171 globalParameters_.SetDatabase(args[3]); |
1e9bad493475
prevent running unit tests on a non-existing db
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
17
diff
changeset
|
172 } |
1e9bad493475
prevent running unit tests on a non-existing db
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
17
diff
changeset
|
173 else if (args.size() == 5) |
1e9bad493475
prevent running unit tests on a non-existing db
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
17
diff
changeset
|
174 { |
1e9bad493475
prevent running unit tests on a non-existing db
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
17
diff
changeset
|
175 // Windows flavor |
1e9bad493475
prevent running unit tests on a non-existing db
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
17
diff
changeset
|
176 globalParameters_.SetHost(args[0]); |
1e9bad493475
prevent running unit tests on a non-existing db
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
17
diff
changeset
|
177 globalParameters_.SetPort(boost::lexical_cast<unsigned int>(args[1])); |
1e9bad493475
prevent running unit tests on a non-existing db
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
17
diff
changeset
|
178 globalParameters_.SetUsername(args[2]); |
1e9bad493475
prevent running unit tests on a non-existing db
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
17
diff
changeset
|
179 globalParameters_.SetPassword(args[3]); |
1e9bad493475
prevent running unit tests on a non-existing db
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
17
diff
changeset
|
180 globalParameters_.SetDatabase(args[4]); |
1e9bad493475
prevent running unit tests on a non-existing db
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
17
diff
changeset
|
181 } |
1e9bad493475
prevent running unit tests on a non-existing db
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
17
diff
changeset
|
182 else |
1e9bad493475
prevent running unit tests on a non-existing db
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
17
diff
changeset
|
183 { |
1e9bad493475
prevent running unit tests on a non-existing db
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
17
diff
changeset
|
184 LOG(ERROR) << "Bad number of arguments"; |
1e9bad493475
prevent running unit tests on a non-existing db
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
17
diff
changeset
|
185 return -1; |
1e9bad493475
prevent running unit tests on a non-existing db
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
17
diff
changeset
|
186 } |
1e9bad493475
prevent running unit tests on a non-existing db
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
17
diff
changeset
|
187 |
1e9bad493475
prevent running unit tests on a non-existing db
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
17
diff
changeset
|
188 Json::Value config; |
1e9bad493475
prevent running unit tests on a non-existing db
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
17
diff
changeset
|
189 globalParameters_.Format(config); |
1e9bad493475
prevent running unit tests on a non-existing db
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
17
diff
changeset
|
190 std::cout << "Parameters of the MySQL connection: " << std::endl |
1e9bad493475
prevent running unit tests on a non-existing db
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
17
diff
changeset
|
191 << config.toStyledString() << std::endl; |
0 | 192 |
193 int result = RUN_ALL_TESTS(); | |
194 | |
195 Orthanc::Logging::Finalize(); | |
196 | |
197 OrthancDatabases::MySQLDatabase::GlobalFinalization(); | |
198 | |
199 return result; | |
200 } |