1 files changed, 256 insertions, 0 deletions
diff --git a/subprojects/boost-sqlite/test/vtable.cpp b/subprojects/boost-sqlite/test/vtable.cpp
new file mode 100644
index 0000000..4f38732
--- /dev/null
+++ b/subprojects/boost-sqlite/test/vtable.cpp
@@ -0,0 +1,256 @@
+//
+// Copyright (c) 2022 Klemens Morgenstern (klemens.morgenstern@gmx.net)
+//
+// Distributed under the Boost Software License, Version 1.0. (See accompanying
+// file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
+//
+
+#include <boost/test/unit_test.hpp>
+#include "test.hpp"
+
+#include <iostream>
+#include <boost/sqlite.hpp>
+#include <boost/unordered_map.hpp>
+#include <boost/intrusive/list.hpp>
+
+#include <string>
+#include <vector>
+
+using namespace boost;
+
+
+BOOST_AUTO_TEST_SUITE(vtable_);
+
+struct del_info : sqlite3_index_info
+{
+  del_info() : sqlite3_index_info()
+  {
+
+  }
+
+  ~del_info()
+  {
+    if (needToFreeIdxStr)
+      sqlite3_free(idxStr);
+  }
+};
+
+struct trivial_struct
+{
+  trivial_struct() = default;
+  int i{1}, j{2}, k{3};
+};
+
+
+
+struct simple_cursor final : sqlite::vtab::cursor<core::string_view>
+{
+
+  simple_cursor(std::vector<std::string>::const_iterator itr,
+                std::vector<std::string>::const_iterator end) : itr(itr), end(end) {}
+  std::vector<std::string>::const_iterator  itr, end;
+
+  sqlite::result<void> next() {itr++; return {};}
+  sqlite::result<sqlite3_int64> row_id() {return *reinterpret_cast<sqlite3_int64*>(&itr);}
+
+  sqlite::result<core::string_view> column(int i, bool /* no_change */)
+  {
+    if (i > 0)
+      throw_exception(std::out_of_range("column out of range"));
+
+    return *itr;
+  }
+
+  bool eof() noexcept {return itr == end;}
+};
+
+struct simple_table final : sqlite::vtab::table<simple_cursor>
+{
+  const char * declaration()
+  {
+    return R"(create table x(name text);)";
+  }
+
+  simple_table(std::vector<std::string> & names) : names(names) {}
+  std::vector<std::string> & names;
+
+  sqlite::result<cursor_type> open()
+  {
+    return cursor_type{names.begin(), names.end()};
+  }
+
+};
+
+struct simple_test_impl final : sqlite::vtab::eponymous_module<simple_table>
+{
+  std::vector<std::string> names = {"ruben", "vinnie", "richard", "klemens"};
+
+
+
+
+  sqlite::result<table_type> connect(sqlite::connection,
+                                     int /*argc*/, const char * const * /*argv*/)
+  {
+    return table_type{names};
+  }
+
+};
+
+
+BOOST_AUTO_TEST_CASE(simple_reader)
+{
+  sqlite::connection conn(":memory:");
+  auto & m = create_module(conn, "test_table", simple_test_impl{});
+
+  auto itr = m.names.begin();
+  for (auto q : conn.query("select * from test_table ;"))
+  {
+    BOOST_CHECK(q.size() == 1);
+    BOOST_CHECK(q.at(0).get_text() == *itr++);
+  }
+
+  m.names.emplace_back("marcelo");
+  itr = m.names.begin();
+  for (auto q : conn.query("select * from test_table ;"))
+  {
+    BOOST_CHECK(q.size() == 1);
+    BOOST_CHECK(q.at(0).get_text() == *itr++);
+  }
+
+  BOOST_CHECK_THROW(conn.query("insert into test_table values('chris')"), boost::system::system_error);
+}
+
+struct modifyable_table;
+
+struct modifyable_cursor final : sqlite::vtab::cursor<variant2::variant<std::int64_t, core::string_view>>
+{
+  using iterator = boost::unordered_map<std::int64_t, std::string>::const_iterator;
+
+  modifyable_cursor(iterator itr, iterator end) : itr(itr), end(end) {}
+
+  iterator itr, end;
+
+  sqlite::result<void> next()            noexcept {itr++; return {};}
+  sqlite::result<sqlite3_int64> row_id() noexcept {return itr->first;}
+
+  sqlite::result<column_type> column(int i, bool /* no_change */) noexcept
+  {
+    switch (i)
+    {
+      case 0: return itr->first;
+      case 1: return itr->second;
+      default:
+        return sqlite::error(SQLITE_RANGE, sqlite::error_info("column out of range"));
+
+    }
+  }
+
+  bool eof() noexcept {return itr == end;}
+};
+
+struct modifyable_table final :
+    sqlite::vtab::table<modifyable_cursor>,
+    intrusive::list_base_hook<intrusive::link_mode<intrusive::auto_unlink> >,
+    sqlite::vtab::modifiable
+{
+  const char * declaration()
+  {
+    return R"(create table x(id integer primary key autoincrement, name text);)";
+  }
+
+  std::string name;
+  boost::unordered_map<std::int64_t, std::string> names;
+
+  int last_index = 0;
+
+  modifyable_table() = default;
+  modifyable_table(modifyable_table && lhs)
+    : name(std::move(lhs.name)), names(std::move(lhs.names)), last_index(lhs.last_index )
+  {
+    this->swap_nodes(lhs);
+  }
+
+
+  sqlite::result<modifyable_cursor> open()
+  {
+    return modifyable_cursor{names.begin(), names.end()};
+  }
+
+  sqlite::result<void> delete_(sqlite::value key)
+  {
+    BOOST_CHECK(names.erase(key.get_int()) == 1);
+    return {};
+  }
+  sqlite::result<sqlite_int64> insert(sqlite::value /*key*/, span<sqlite::value> values,
+                      int /*on_conflict*/)
+  {
+    int id = values[0].is_null() ? last_index++ : values[0].get_int();
+    auto itr = names.emplace(id, values[1].get_text()).first;
+    return itr->first;
+  }
+  sqlite::result<sqlite_int64> update(sqlite::value old_key, sqlite::value new_key,
+                      span<sqlite::value> values,
+                      int /*on_conflict*/)
+  {
+    if (new_key.get_int() != old_key.get_int())
+      names.erase(old_key.get_int());
+    names.insert_or_assign(new_key.get_int(), values[1].get_text());
+    return 0u;
+  }
+};
+
+struct modifyable_test_impl final : sqlite::vtab::eponymous_module<modifyable_table>
+{
+  modifyable_test_impl() = default;
+
+  intrusive::list<table_type, intrusive::constant_time_size<false>> list;
+
+  sqlite::result<table_type> connect(sqlite::connection,
+                                     int /*argc*/, const char * const  argv[]) noexcept
+  {
+    table_type tt{};
+    tt.name.assign(argv[2]);
+    list.push_back(tt);
+    return sqlite::result<table_type>(std::move(tt));
+  }
+};
+
+
+BOOST_AUTO_TEST_CASE(modifyable_reader)
+{
+  sqlite::connection conn(":memory:");
+
+  auto & m = create_module(conn, "name_table", modifyable_test_impl{});
+  conn.execute("create virtual table test_table USING name_table; ");
+
+  BOOST_CHECK(m.list.size() == 1);
+  BOOST_CHECK(m.list.front().name == "test_table");
+  BOOST_CHECK(m.list.front().names.empty());
+
+  conn.prepare("insert into test_table(name) values ($1), ($2), ($3), ($4);")
+      .execute(std::make_tuple("vinnie", "richard", "ruben", "peter"));
+
+  auto & t = m.list.front();
+  BOOST_CHECK(t.names.size() == 4);
+  BOOST_CHECK(t.names[0] == "vinnie");
+  BOOST_CHECK(t.names[1] == "richard");
+  BOOST_CHECK(t.names[2] == "ruben");
+  BOOST_CHECK(t.names[3] == "peter");
+
+  conn.prepare("delete from test_table where name = $1;").execute(std::make_tuple("richard"));
+
+  BOOST_CHECK(t.names.size() == 3);
+  BOOST_CHECK(t.names[0] == "vinnie");
+  BOOST_CHECK(t.names[2] == "ruben");
+  BOOST_CHECK(t.names[3] == "peter");
+
+  conn.prepare("update test_table set name = $1 where id = 0;").execute(std::make_tuple("richard"));
+
+  BOOST_CHECK(t.names.size() == 3);
+  BOOST_CHECK(t.names[0] == "richard");
+  BOOST_CHECK(t.names[2] == "ruben");
+  BOOST_CHECK(t.names[3] == "peter");
+}
+
+
+BOOST_AUTO_TEST_SUITE_END()