1*6a6c8299Sjmmv // Copyright 2011 Google Inc.
2*6a6c8299Sjmmv // All rights reserved.
3*6a6c8299Sjmmv //
4*6a6c8299Sjmmv // Redistribution and use in source and binary forms, with or without
5*6a6c8299Sjmmv // modification, are permitted provided that the following conditions are
6*6a6c8299Sjmmv // met:
7*6a6c8299Sjmmv //
8*6a6c8299Sjmmv // * Redistributions of source code must retain the above copyright
9*6a6c8299Sjmmv //   notice, this list of conditions and the following disclaimer.
10*6a6c8299Sjmmv // * Redistributions in binary form must reproduce the above copyright
11*6a6c8299Sjmmv //   notice, this list of conditions and the following disclaimer in the
12*6a6c8299Sjmmv //   documentation and/or other materials provided with the distribution.
13*6a6c8299Sjmmv // * Neither the name of Google Inc. nor the names of its contributors
14*6a6c8299Sjmmv //   may be used to endorse or promote products derived from this software
15*6a6c8299Sjmmv //   without specific prior written permission.
16*6a6c8299Sjmmv //
17*6a6c8299Sjmmv // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
18*6a6c8299Sjmmv // "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
19*6a6c8299Sjmmv // LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
20*6a6c8299Sjmmv // A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
21*6a6c8299Sjmmv // OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
22*6a6c8299Sjmmv // SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
23*6a6c8299Sjmmv // LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
24*6a6c8299Sjmmv // DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
25*6a6c8299Sjmmv // THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
26*6a6c8299Sjmmv // (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
27*6a6c8299Sjmmv // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
28*6a6c8299Sjmmv 
29*6a6c8299Sjmmv #include "cli/cmd_db_exec.hpp"
30*6a6c8299Sjmmv 
31*6a6c8299Sjmmv #include <cstring>
32*6a6c8299Sjmmv 
33*6a6c8299Sjmmv #include <atf-c++.hpp>
34*6a6c8299Sjmmv 
35*6a6c8299Sjmmv #include "utils/format/macros.hpp"
36*6a6c8299Sjmmv #include "utils/sqlite/database.hpp"
37*6a6c8299Sjmmv #include "utils/sqlite/statement.ipp"
38*6a6c8299Sjmmv 
39*6a6c8299Sjmmv namespace sqlite = utils::sqlite;
40*6a6c8299Sjmmv 
41*6a6c8299Sjmmv 
42*6a6c8299Sjmmv namespace {
43*6a6c8299Sjmmv 
44*6a6c8299Sjmmv 
45*6a6c8299Sjmmv /// Performs a test for the cli::format_cell() function.
46*6a6c8299Sjmmv ///
47*6a6c8299Sjmmv /// \tparam Cell The type of the value to insert into the test column.
48*6a6c8299Sjmmv /// \param column_type The SQL type of the test column.
49*6a6c8299Sjmmv /// \param value The value to insert into the test column.
50*6a6c8299Sjmmv /// \param exp_value The expected return value of cli::format_cell().
51*6a6c8299Sjmmv template< class Cell >
52*6a6c8299Sjmmv static void
do_format_cell_test(const std::string column_type,const Cell & value,const std::string & exp_value)53*6a6c8299Sjmmv do_format_cell_test(const std::string column_type,
54*6a6c8299Sjmmv                     const Cell& value, const std::string& exp_value)
55*6a6c8299Sjmmv {
56*6a6c8299Sjmmv     sqlite::database db = sqlite::database::in_memory();
57*6a6c8299Sjmmv 
58*6a6c8299Sjmmv     sqlite::statement create = db.create_statement(
59*6a6c8299Sjmmv         F("CREATE TABLE test (column %s)") % column_type);
60*6a6c8299Sjmmv     create.step_without_results();
61*6a6c8299Sjmmv 
62*6a6c8299Sjmmv     sqlite::statement insert = db.create_statement(
63*6a6c8299Sjmmv         "INSERT INTO test (column) VALUES (:column)");
64*6a6c8299Sjmmv     insert.bind(":column", value);
65*6a6c8299Sjmmv     insert.step_without_results();
66*6a6c8299Sjmmv 
67*6a6c8299Sjmmv     sqlite::statement query = db.create_statement("SELECT * FROM test");
68*6a6c8299Sjmmv     ATF_REQUIRE(query.step());
69*6a6c8299Sjmmv     ATF_REQUIRE_EQ(exp_value, cli::format_cell(query, 0));
70*6a6c8299Sjmmv     ATF_REQUIRE(!query.step());
71*6a6c8299Sjmmv }
72*6a6c8299Sjmmv 
73*6a6c8299Sjmmv 
74*6a6c8299Sjmmv }  // anonymous namespace
75*6a6c8299Sjmmv 
76*6a6c8299Sjmmv 
77*6a6c8299Sjmmv ATF_TEST_CASE_WITHOUT_HEAD(format_cell__blob);
ATF_TEST_CASE_BODY(format_cell__blob)78*6a6c8299Sjmmv ATF_TEST_CASE_BODY(format_cell__blob)
79*6a6c8299Sjmmv {
80*6a6c8299Sjmmv     const char* contents = "Some random contents";
81*6a6c8299Sjmmv     do_format_cell_test(
82*6a6c8299Sjmmv         "BLOB", sqlite::blob(contents, std::strlen(contents)),
83*6a6c8299Sjmmv         F("BLOB of %s bytes") % strlen(contents));
84*6a6c8299Sjmmv }
85*6a6c8299Sjmmv 
86*6a6c8299Sjmmv 
87*6a6c8299Sjmmv ATF_TEST_CASE_WITHOUT_HEAD(format_cell__float);
ATF_TEST_CASE_BODY(format_cell__float)88*6a6c8299Sjmmv ATF_TEST_CASE_BODY(format_cell__float)
89*6a6c8299Sjmmv {
90*6a6c8299Sjmmv     do_format_cell_test("FLOAT", 3.5, "3.5");
91*6a6c8299Sjmmv }
92*6a6c8299Sjmmv 
93*6a6c8299Sjmmv 
94*6a6c8299Sjmmv ATF_TEST_CASE_WITHOUT_HEAD(format_cell__integer);
ATF_TEST_CASE_BODY(format_cell__integer)95*6a6c8299Sjmmv ATF_TEST_CASE_BODY(format_cell__integer)
96*6a6c8299Sjmmv {
97*6a6c8299Sjmmv     do_format_cell_test("INTEGER", 123456, "123456");
98*6a6c8299Sjmmv }
99*6a6c8299Sjmmv 
100*6a6c8299Sjmmv 
101*6a6c8299Sjmmv ATF_TEST_CASE_WITHOUT_HEAD(format_cell__null);
ATF_TEST_CASE_BODY(format_cell__null)102*6a6c8299Sjmmv ATF_TEST_CASE_BODY(format_cell__null)
103*6a6c8299Sjmmv {
104*6a6c8299Sjmmv     do_format_cell_test("TEXT", sqlite::null(), "NULL");
105*6a6c8299Sjmmv }
106*6a6c8299Sjmmv 
107*6a6c8299Sjmmv 
108*6a6c8299Sjmmv ATF_TEST_CASE_WITHOUT_HEAD(format_cell__text);
ATF_TEST_CASE_BODY(format_cell__text)109*6a6c8299Sjmmv ATF_TEST_CASE_BODY(format_cell__text)
110*6a6c8299Sjmmv {
111*6a6c8299Sjmmv     do_format_cell_test("TEXT", "Hello, world", "Hello, world");
112*6a6c8299Sjmmv }
113*6a6c8299Sjmmv 
114*6a6c8299Sjmmv 
115*6a6c8299Sjmmv ATF_TEST_CASE_WITHOUT_HEAD(format_headers);
ATF_TEST_CASE_BODY(format_headers)116*6a6c8299Sjmmv ATF_TEST_CASE_BODY(format_headers)
117*6a6c8299Sjmmv {
118*6a6c8299Sjmmv     sqlite::database db = sqlite::database::in_memory();
119*6a6c8299Sjmmv 
120*6a6c8299Sjmmv     sqlite::statement create = db.create_statement(
121*6a6c8299Sjmmv         "CREATE TABLE test (c1 TEXT, c2 TEXT, c3 TEXT)");
122*6a6c8299Sjmmv     create.step_without_results();
123*6a6c8299Sjmmv 
124*6a6c8299Sjmmv     sqlite::statement query = db.create_statement(
125*6a6c8299Sjmmv         "SELECT c1, c2, c3 AS c3bis FROM test");
126*6a6c8299Sjmmv     ATF_REQUIRE_EQ("c1,c2,c3bis", cli::format_headers(query));
127*6a6c8299Sjmmv }
128*6a6c8299Sjmmv 
129*6a6c8299Sjmmv 
130*6a6c8299Sjmmv ATF_TEST_CASE_WITHOUT_HEAD(format_row);
ATF_TEST_CASE_BODY(format_row)131*6a6c8299Sjmmv ATF_TEST_CASE_BODY(format_row)
132*6a6c8299Sjmmv {
133*6a6c8299Sjmmv     sqlite::database db = sqlite::database::in_memory();
134*6a6c8299Sjmmv 
135*6a6c8299Sjmmv     sqlite::statement create = db.create_statement(
136*6a6c8299Sjmmv         "CREATE TABLE test (c1 TEXT, c2 BLOB)");
137*6a6c8299Sjmmv     create.step_without_results();
138*6a6c8299Sjmmv 
139*6a6c8299Sjmmv     const char* memory = "BLOB contents";
140*6a6c8299Sjmmv     sqlite::statement insert = db.create_statement(
141*6a6c8299Sjmmv         "INSERT INTO test VALUES (:v1, :v2)");
142*6a6c8299Sjmmv     insert.bind(":v1", "A string");
143*6a6c8299Sjmmv     insert.bind(":v2", sqlite::blob(memory, std::strlen(memory)));
144*6a6c8299Sjmmv     insert.step_without_results();
145*6a6c8299Sjmmv 
146*6a6c8299Sjmmv     sqlite::statement query = db.create_statement("SELECT * FROM test");
147*6a6c8299Sjmmv     query.step();
148*6a6c8299Sjmmv     ATF_REQUIRE_EQ(
149*6a6c8299Sjmmv         (F("A string,BLOB of %s bytes") % std::strlen(memory)).str(),
150*6a6c8299Sjmmv         cli::format_row(query));
151*6a6c8299Sjmmv }
152*6a6c8299Sjmmv 
153*6a6c8299Sjmmv 
ATF_INIT_TEST_CASES(tcs)154*6a6c8299Sjmmv ATF_INIT_TEST_CASES(tcs)
155*6a6c8299Sjmmv {
156*6a6c8299Sjmmv     ATF_ADD_TEST_CASE(tcs, format_cell__blob);
157*6a6c8299Sjmmv     ATF_ADD_TEST_CASE(tcs, format_cell__float);
158*6a6c8299Sjmmv     ATF_ADD_TEST_CASE(tcs, format_cell__integer);
159*6a6c8299Sjmmv     ATF_ADD_TEST_CASE(tcs, format_cell__null);
160*6a6c8299Sjmmv     ATF_ADD_TEST_CASE(tcs, format_cell__text);
161*6a6c8299Sjmmv 
162*6a6c8299Sjmmv     ATF_ADD_TEST_CASE(tcs, format_headers);
163*6a6c8299Sjmmv 
164*6a6c8299Sjmmv     ATF_ADD_TEST_CASE(tcs, format_row);
165*6a6c8299Sjmmv }
166