1 files changed, 45 insertions, 0 deletions
diff --git a/test/collation.cpp b/test/collation.cpp
new file mode 100644
index 0000000..8893e42
--- /dev/null
+++ b/test/collation.cpp
@@ -0,0 +1,45 @@
+//
+// 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/sqlite/collation.hpp>
+#include <string>
+#include <vector>
+
+#include "test.hpp"
+
+using namespace boost;
+
+struct collate_length
+{
+  int operator()(core::string_view l, core::string_view r) noexcept
+  {
+    return std::stoull(r) - l.size();
+  }
+};
+
+BOOST_AUTO_TEST_CASE(collation)
+{
+  sqlite::connection conn(":memory:");
+  conn.execute(
+#include "test-db.sql"
+  );
+
+  sqlite::create_collation(conn, "length", collate_length{});
+
+  std::vector<std::string> names;
+
+  // language=sqlite
+  for (auto r : conn.query("select first_name from author where first_name = 5 collate length order by last_name asc;"))
+    names.emplace_back(r.at(0).get_text());
+
+  std::vector<std::string> cmp = {"peter", "ruben"};
+  BOOST_CHECK(names == cmp);
+
+  sqlite::delete_collation(conn, "length");
+
+  BOOST_CHECK_THROW(conn.query("select first_name from author where first_name = 5 collate length order by last_name asc;"), system::system_error);
+}
+\ No newline at end of file