Mercurial > hg > orthanc

comparison UnitTests/SQLite.cpp @ 0:3959d33612cc

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
initial commit
author Sebastien Jodogne <s.jodogne@gmail.com>
date Thu, 19 Jul 2012 14:32:22 +0200
parents
children a15e90e5d6fc
comparison
equal deleted inserted replaced
-1:000000000000 0:3959d33612cc
1 #include "gtest/gtest.h"
2
3 #include "../Core/Toolbox.h"
4 #include "../Core/SQLite/Connection.h"
5 #include "../Core/SQLite/Statement.h"
6
7 #include <sqlite3.h>
8
9 using namespace Palantir;
10
11
12 TEST(SQLite, Configuration)
13 {
14 ASSERT_EQ(1, sqlite3_threadsafe());
15 }
16
17
18 TEST(SQLite, Connection)
19 {
20 Toolbox::RemoveFile("coucou");
21 SQLite::Connection c;
22 c.Open("coucou");
23 c.Execute("CREATE TABLE c(k INTEGER PRIMARY KEY AUTOINCREMENT, v INTEGER)");
24 c.Execute("INSERT INTO c VALUES(NULL, 42);");
25 }
26
27
28 TEST(SQLite, StatementReferenceBasic)
29 {
30 sqlite3* db;
31 sqlite3_open(":memory:", &db);
32
33 {
34 SQLite::StatementReference r(db, "SELECT * FROM sqlite_master");
35 ASSERT_EQ(0u, r.GetReferenceCount());
36
37 {
38 SQLite::StatementReference r1(r);
39 ASSERT_EQ(1u, r.GetReferenceCount());
40 ASSERT_EQ(0u, r1.GetReferenceCount());
41
42 {
43 SQLite::StatementReference r2(r);
44 ASSERT_EQ(2u, r.GetReferenceCount());
45 ASSERT_EQ(0u, r1.GetReferenceCount());
46 ASSERT_EQ(0u, r2.GetReferenceCount());
47
48 SQLite::StatementReference r3(r2);
49 ASSERT_EQ(3u, r.GetReferenceCount());
50 ASSERT_EQ(0u, r1.GetReferenceCount());
51 ASSERT_EQ(0u, r2.GetReferenceCount());
52 ASSERT_EQ(0u, r3.GetReferenceCount());
53 }
54
55 ASSERT_EQ(1u, r.GetReferenceCount());
56 ASSERT_EQ(0u, r1.GetReferenceCount());
57
58 {
59 SQLite::StatementReference r2(r);
60 ASSERT_EQ(2u, r.GetReferenceCount());
61 ASSERT_EQ(0u, r1.GetReferenceCount());
62 ASSERT_EQ(0u, r2.GetReferenceCount());
63 }
64
65 ASSERT_EQ(1u, r.GetReferenceCount());
66 ASSERT_EQ(0u, r1.GetReferenceCount());
67 }
68
69 ASSERT_EQ(0u, r.GetReferenceCount());
70 }
71
72 sqlite3_close(db);
73 }
74
75 TEST(SQLite, StatementBasic)
76 {
77 SQLite::Connection c;
78 c.OpenInMemory();
79
80 SQLite::Statement s(c, "SELECT * from sqlite_master");
81 s.Run();
82
83 for (unsigned int i = 0; i < 5; i++)
84 {
85 SQLite::Statement cs(c, SQLITE_FROM_HERE, "SELECT * from sqlite_master");
86 cs.Step();
87 }
88 }
89
90
91 namespace
92 {
93 static bool destroyed;
94
95 class MyFunc : public SQLite::IScalarFunction
96 {
97 public:
98 MyFunc()
99 {
100 destroyed = false;
101 }
102
103 virtual ~MyFunc()
104 {
105 destroyed = true;
106 }
107
108 virtual const char* GetName() const
109 {
110 return "MYFUNC";
111 }
112
113 virtual unsigned int GetCardinality() const
114 {
115 return 2;
116 }
117
118 virtual void Compute(SQLite::FunctionContext& context)
119 {
120 context.SetIntResult(1000 + context.GetIntValue(0) * context.GetIntValue(1));
121 }
122 };
123
124 class MyDelete : public SQLite::IScalarFunction
125 {
126 public:
127 std::set<int> deleted_;
128
129 virtual const char* GetName() const
130 {
131 return "MYDELETE";
132 }
133
134 virtual unsigned int GetCardinality() const
135 {
136 return 1;
137 }
138
139 virtual void Compute(SQLite::FunctionContext& context)
140 {
141 deleted_.insert(context.GetIntValue(0));
142 context.SetNullResult();
143 }
144 };
145 }
146
147 TEST(SQLite, ScalarFunction)
148 {
149 {
150 SQLite::Connection c;
151 c.OpenInMemory();
152 c.Register(new MyFunc());
153 c.Execute("CREATE TABLE t(id INTEGER PRIMARY KEY, v1 INTEGER, v2 INTEGER);");
154 c.Execute("INSERT INTO t VALUES(NULL, 2, 3);");
155 c.Execute("INSERT INTO t VALUES(NULL, 4, 4);");
156 c.Execute("INSERT INTO t VALUES(NULL, 6, 5);");
157 SQLite::Statement t(c, "SELECT MYFUNC(v1, v2), v1, v2 FROM t");
158 int i = 0;
159 while (t.Step())
160 {
161 ASSERT_EQ(t.ColumnInt(0), 1000 + t.ColumnInt(1) * t.ColumnInt(2));
162 i++;
163 }
164 ASSERT_EQ(3, i);
165 ASSERT_FALSE(destroyed);
166 }
167 ASSERT_TRUE(destroyed);
168 }
169
170 TEST(SQLite, CascadedDeleteCallback)
171 {
172 SQLite::Connection c;
173 c.OpenInMemory();
174 MyDelete *func = new MyDelete();
175 c.Register(func);
176 c.Execute("CREATE TABLE parent(id INTEGER PRIMARY KEY, dummy INTEGER);");
177 c.Execute("CREATE TABLE child("
178 " id INTEGER PRIMARY KEY, "
179 " parent INTEGER REFERENCES parent(id) ON DELETE CASCADE, "
180 " value INTEGER);");
181 c.Execute("CREATE TRIGGER childRemoved "
182 "AFTER DELETE ON child "
183 "FOR EACH ROW BEGIN "
184 " SELECT MYDELETE(old.value); "
185 "END;");
186
187 c.Execute("INSERT INTO parent VALUES(42, 100);");
188 c.Execute("INSERT INTO parent VALUES(43, 101);");
189
190 c.Execute("INSERT INTO child VALUES(NULL, 42, 4200);");
191 c.Execute("INSERT INTO child VALUES(NULL, 42, 4201);");
192
193 c.Execute("INSERT INTO child VALUES(NULL, 43, 4300);");
194 c.Execute("INSERT INTO child VALUES(NULL, 43, 4301);");
195
196 // The following command deletes "parent(43, 101)", then in turns
197 // "child(NULL, 43, 4300/4301)", then calls the MyDelete on 4300 and
198 // 4301
199 c.Execute("DELETE FROM parent WHERE dummy=101");
200
201 ASSERT_EQ(2u, func->deleted_.size());
202 ASSERT_TRUE(func->deleted_.find(4300) != func->deleted_.end());
203 ASSERT_TRUE(func->deleted_.find(4301) != func->deleted_.end());
204 }