mirror of
https://github.com/scottlamb/moonfire-nvr.git
synced 2025-01-30 16:06:00 -05:00
106 lines
3.8 KiB
C++
106 lines
3.8 KiB
C++
// This file is part of Moonfire NVR, a security camera network video recorder.
|
|
// Copyright (C) 2016 Scott Lamb <slamb@slamb.org>
|
|
//
|
|
// This program is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU General Public License as published by
|
|
// the Free Software Foundation, either version 3 of the License, or
|
|
// (at your option) any later version.
|
|
//
|
|
// In addition, as a special exception, the copyright holders give
|
|
// permission to link the code of portions of this program with the
|
|
// OpenSSL library under certain conditions as described in each
|
|
// individual source file, and distribute linked combinations including
|
|
// the two.
|
|
//
|
|
// You must obey the GNU General Public License in all respects for all
|
|
// of the code used other than OpenSSL. If you modify file(s) with this
|
|
// exception, you may extend this exception to your version of the
|
|
// file(s), but you are not obligated to do so. If you do not wish to do
|
|
// so, delete this exception statement from your version. If you delete
|
|
// this exception statement from all source files in the program, then
|
|
// also delete it here.
|
|
//
|
|
// This program is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU General Public License for more details.
|
|
//
|
|
// You should have received a copy of the GNU General Public License
|
|
// along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
//
|
|
// sqlite-test.cc: tests of the sqlite.h interface.
|
|
|
|
#include <string>
|
|
|
|
#include <gflags/gflags.h>
|
|
#include <glog/logging.h>
|
|
#include <gmock/gmock.h>
|
|
#include <gtest/gtest.h>
|
|
|
|
#include "sqlite.h"
|
|
#include "string.h"
|
|
#include "testutil.h"
|
|
|
|
DECLARE_bool(alsologtostderr);
|
|
|
|
namespace moonfire_nvr {
|
|
namespace {
|
|
|
|
class SqliteTest : public testing::Test {
|
|
protected:
|
|
SqliteTest() {
|
|
std::string error_message;
|
|
CHECK(db_.Open(":memory:", SQLITE_OPEN_READWRITE | SQLITE_OPEN_CREATE,
|
|
&error_message))
|
|
<< error_message;
|
|
|
|
std::string create_sql = ReadFileOrDie("../src/schema.sql");
|
|
CHECK(RunStatements(&db_, create_sql, &error_message)) << error_message;
|
|
}
|
|
|
|
Database db_;
|
|
};
|
|
|
|
TEST_F(SqliteTest, JustCreate) {}
|
|
|
|
TEST_F(SqliteTest, BindAndColumn) {
|
|
std::string error_message;
|
|
auto insert_stmt = db_.Prepare(
|
|
"insert into camera (uuid, short_name, retain_bytes) "
|
|
"values (?, ?, ?)",
|
|
nullptr, &error_message);
|
|
ASSERT_TRUE(insert_stmt != nullptr) << error_message;
|
|
const char kBlob[] = {0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07,
|
|
0x08, 0x09, 0x0a, 0x0b, 0x0c, 0x0d, 0x0e, 0x0f};
|
|
re2::StringPiece blob_piece = re2::StringPiece(kBlob, sizeof(kBlob));
|
|
const char kText[] = "foo";
|
|
const int64_t kInt64 = INT64_C(0xdeadbeeffeedface);
|
|
ASSERT_TRUE(insert_stmt->BindBlob(1, blob_piece, &error_message))
|
|
<< error_message;
|
|
ASSERT_TRUE(insert_stmt->BindText(2, kText, &error_message)) << error_message;
|
|
ASSERT_TRUE(insert_stmt->BindInt64(3, kInt64, &error_message))
|
|
<< error_message;
|
|
ASSERT_EQ(SQLITE_DONE, insert_stmt->Step());
|
|
|
|
auto select_stmt =
|
|
db_.Prepare("select uuid, short_name, retain_bytes from camera", nullptr,
|
|
&error_message);
|
|
ASSERT_TRUE(select_stmt != nullptr) << error_message;
|
|
ASSERT_EQ(SQLITE_ROW, select_stmt->Step());
|
|
EXPECT_EQ(ToHex(blob_piece), ToHex(select_stmt->ColumnBlob(0)));
|
|
EXPECT_EQ(kText, select_stmt->ColumnText(1).as_string());
|
|
EXPECT_EQ(kInt64, select_stmt->ColumnInt64(2));
|
|
ASSERT_EQ(SQLITE_DONE, select_stmt->Step());
|
|
}
|
|
|
|
} // namespace
|
|
} // namespace moonfire_nvr
|
|
|
|
int main(int argc, char **argv) {
|
|
FLAGS_alsologtostderr = true;
|
|
google::ParseCommandLineFlags(&argc, &argv, true);
|
|
testing::InitGoogleTest(&argc, argv);
|
|
google::InitGoogleLogging(argv[0]);
|
|
return RUN_ALL_TESTS();
|
|
}
|