Add an AsyncResult type to iterate over the Select resultsasync_sql

1 files changed, 23 insertions, 0 deletions

diff --git a/tests/database.cpp b/tests/database.cpp
index 070a460..aab10f4 100644
--- a/tests/database.cpp
+++ b/tests/database.cpp
@@ -8,6 +8,7 @@
 
 #include <database/database.hpp>
 #include <database/save.hpp>
+#include <database/select_query.hpp>
 
 #include <config/config.hpp>
 
@@ -165,6 +166,28 @@ TEST_CASE("Database")
       CHECK(after_connection_commands.size() == 2);
     }
 
+  SECTION("async_select")
+    {
+      constexpr auto db_size = 8;
+      CHECK(Database::count(Database::muc_log_lines) == 0);
+      auto uuid = Database::store_muc_message("owner", "#chan", "irc.example.com", std::chrono::system_clock::now(), "hello!", "louiz'");
+      CHECK(Database::count(Database::muc_log_lines) == 1);
+      CHECK(uuid.size() == 36);
+
+      for (int i = 1; i < db_size; ++i)
+        Database::store_muc_message("owner", "#chan", "irc.example.com", std::chrono::system_clock::now(), "hello "s + std::to_string(i) + "!"s, "louiz'");
+      CHECK(Database::count(Database::muc_log_lines) == db_size);
+
+      auto query = select(Database::muc_log_lines);
+      query.where() << Database::Owner{} << "=" << "owner"s;
+      auto async_result = query.execute_async(*Database::db);
+
+      int i = 0;
+      for (auto rows: async_result)
+        ++i;
+      CHECK(i == db_size);
+    }
+
   Database::close();
 }
 #endif