test_db.cpp

Go to the documentation of this file.

/// \file test_db.cpp
/// \brief Модульные тесты для BlockerDB (обёртка SQLite).
 
#include "db.h"
#include <gtest/gtest.h>
 
#include <cstdlib>
#include <cstring>
#include <filesystem>
 
/// Тестовое окружение: создаёт свежую временную БД для каждого теста.
class DBTest : public ::testing::Test {
protected:
  void SetUp() override {
    path_ = "/tmp/ebpf_blocker_test.db";
    std::filesystem::remove(path_);
    db_ = std::make_unique<BlockerDB>(path_);
    db_->Migrate();
  }
 
  void TearDown() override {
    db_.reset();
    std::filesystem::remove(path_);
  }
 
  std::string path_; ///< Путь к временной тестовой БД.
  std::unique_ptr<BlockerDB> db_; ///< Тестируемый экземпляр БД.
};
 
TEST_F(DBTest, EmptyDatabase) {
  auto whitelist = db_->GetWhitelist();
  auto blacklist = db_->GetBlacklist();
  EXPECT_TRUE(whitelist.empty());
  EXPECT_TRUE(blacklist.empty());
}
 
TEST_F(DBTest, InsertAndReadBlacklist) {
  {
    sqlite3 *raw = nullptr;
    sqlite3_open(path_.c_str(), &raw);
    sqlite3_exec(raw,
                 "INSERT INTO blocked_ips (ip, comment) VALUES ('10.0.0.0/8', "
                 "'test subnet');"
                 "INSERT INTO blocked_ips (ip, comment) VALUES ('1.2.3.4', "
                 "'single IP');",
                 nullptr, nullptr, nullptr);
    sqlite3_close(raw);
  }
 
  auto entries = db_->GetBlacklist();
  EXPECT_EQ(entries.size(), 2);
  EXPECT_EQ(entries[0].ip, "10.0.0.0/8");
  EXPECT_EQ(entries[1].ip, "1.2.3.4");
}
 
TEST_F(DBTest, InsertAndReadWhitelist) {
  {
    sqlite3 *raw = nullptr;
    sqlite3_open(path_.c_str(), &raw);
    sqlite3_exec(raw,
                 "INSERT INTO whitelist_ips (ip, comment) VALUES "
                 "('192.168.0.0/16', 'LAN');"
                 "INSERT INTO whitelist_ips (ip, comment) VALUES ('127.0.0.1', "
                 "'localhost');",
                 nullptr, nullptr, nullptr);
    sqlite3_close(raw);
  }
 
  auto entries = db_->GetWhitelist();
  EXPECT_EQ(entries.size(), 2);
  EXPECT_EQ(entries[0].ip, "192.168.0.0/16");
  EXPECT_EQ(entries[1].ip, "127.0.0.1");
}
 
TEST_F(DBTest, WhitelistAndBlacklistAreSeparate) {
  {
    sqlite3 *raw = nullptr;
    sqlite3_open(path_.c_str(), &raw);
    sqlite3_exec(raw,
                 "INSERT INTO whitelist_ips (ip) VALUES ('10.0.0.0/8');"
                 "INSERT INTO blocked_ips (ip) VALUES ('20.0.0.0/8');",
                 nullptr, nullptr, nullptr);
    sqlite3_close(raw);
  }
 
  auto wl = db_->GetWhitelist();
  auto bl = db_->GetBlacklist();
  EXPECT_EQ(wl.size(), 1);
  EXPECT_EQ(bl.size(), 1);
  EXPECT_EQ(wl[0].ip, "10.0.0.0/8");
  EXPECT_EQ(bl[0].ip, "20.0.0.0/8");
}
 
TEST_F(DBTest, MigrationCreatesTables) {
  sqlite3 *raw = nullptr;
  sqlite3_open(path_.c_str(), &raw);
 
  auto check_table = [&](const char *name) {
    char **results = nullptr;
    int rows = 0, cols = 0;
    std::string sql =
        "SELECT name FROM sqlite_master WHERE type='table' AND name='" +
        std::string(name) + "'";
    sqlite3_get_table(raw, sql.c_str(), &results, &rows, &cols, nullptr);
    bool found = (rows > 0);
    sqlite3_free_table(results);
    return found;
  };
 
  EXPECT_TRUE(check_table("whitelist_ips"));
  EXPECT_TRUE(check_table("blocked_ips"));
 
  sqlite3_close(raw);
}