Implement {query,execute}_{one,value}() shortcut functions

Useful in situations where the query is know to return at most one element
(*_one) or exactly one element (*_value).

1 files changed, 204 insertions, 0 deletions

diff --git a/common/query/one/driver.cxx b/common/query/one/driver.cxx
new file mode 100644
index 0000000..e6b2b67
--- /dev/null
+++ b/common/query/one/driver.cxx
@@ -0,0 +1,204 @@
+// file      : common/query/one/driver.cxx
+// copyright : Copyright (c) 2009-2014 Code Synthesis Tools CC
+// license   : GNU GPL v2; see accompanying LICENSE file
+
+// Test query one support.
+//
+// We assume that other tests in common/query/ exercise a variety of
+// different kinds of queries. Here we are concerned with what is
+// specific to query_one() and query_value().
+//
+
+#include <memory>   // std::auto_ptr
+#include <cassert>
+#include <iostream>
+
+#include <odb/database.hxx>
+#include <odb/transaction.hxx>
+
+#include <common/common.hxx>
+
+#include "test.hxx"
+#include "test-odb.hxx"
+
+using namespace std;
+using namespace odb::core;
+
+int
+main (int argc, char* argv[])
+{
+  try
+  {
+    auto_ptr<database> db (create_database (argc, argv));
+    odb::database_id db_id (db->id ());
+
+    transaction t (db->begin ());
+
+    // query_one()
+    //
+    {
+      auto_ptr<object> o (db->query_one<object> ());
+      assert (o.get () == 0);
+    }
+
+    {
+      object o (4);
+      assert (!db->query_one<object> (o) && o.id_ == 4 && o.str_.empty ());
+    }
+
+    /*
+    {
+      object o (db->query_value<object> ());
+      assert (false);
+    }
+    */
+
+    object o (1);
+    o.str_ = "value 1";
+    db->persist (o);
+
+    {
+      auto_ptr<object> o (db->query_one<object> ());
+      assert (o.get () != 0 && o->str_ == "value 1");
+    }
+
+    {
+      object o;
+      assert (db->query_one<object> (o) && o.str_ == "value 1");
+    }
+
+    {
+      object o (db->query_value<object> ());
+      assert (o.str_ == "value 1");
+    }
+
+    // query_one(const char*)
+    //
+    const char* q1_c (db_id == odb::id_oracle ? "\"id\" = 1" : "id = 1");
+    const char* q0_c (db_id == odb::id_oracle ? "\"id\" = 2" : "id = 2");
+
+    {
+      auto_ptr<object> o (db->query_one<object> (q1_c));
+      assert (o.get () != 0 && o->str_ == "value 1");
+    }
+
+    {
+      auto_ptr<object> o (db->query_one<object> (q0_c));
+      assert (o.get () == 0);
+    }
+
+    {
+      object o;
+      assert (db->query_one<object> (q1_c, o) && o.str_ == "value 1");
+    }
+
+    {
+      object o (4);
+      assert (!db->query_one<object> (q0_c, o) &&
+              o.id_ == 4 && o.str_.empty ());
+    }
+
+    {
+      object o (db->query_value<object> (q1_c));
+      assert (o.str_ == "value 1");
+    }
+
+    // query_one(std::string)
+    //
+    string q1_s (q1_c);
+    string q0_s (q0_c);
+
+    {
+      auto_ptr<object> o (db->query_one<object> (q1_s));
+      assert (o.get () != 0 && o->str_ == "value 1");
+    }
+
+    {
+      auto_ptr<object> o (db->query_one<object> (q0_s));
+      assert (o.get () == 0);
+    }
+
+    {
+      object o;
+      assert (db->query_one<object> (q1_s, o) && o.str_ == "value 1");
+    }
+
+    {
+      object o (4);
+      assert (!db->query_one<object> (q0_s, o) &&
+              o.id_ == 4 && o.str_.empty ());
+    }
+
+    {
+      object o (db->query_value<object> (q1_s));
+      assert (o.str_ == "value 1");
+    }
+
+    // query_one(odb::query)
+    //
+    typedef odb::query<object> query;
+
+    query q1 (query::id == 1);
+    query q0 (query::id == 2);
+
+    {
+      auto_ptr<object> o (db->query_one<object> (q1));
+      assert (o.get () != 0 && o->str_ == "value 1");
+    }
+
+    {
+      auto_ptr<object> o (db->query_one<object> (q0));
+      assert (o.get () == 0);
+    }
+
+    {
+      object o;
+      assert (db->query_one<object> (q1, o) && o.str_ == "value 1");
+    }
+
+    {
+      object o (4);
+      assert (!db->query_one<object> (q0, o) && o.id_ == 4 && o.str_.empty ());
+    }
+
+    {
+      object o (db->query_value<object> (q1));
+      assert (o.str_ == "value 1");
+    }
+
+    // Assertion on more than one element.
+    //
+    {
+      object o (2);
+      o.str_ = "value 2";
+      db->persist (o);
+    }
+
+    /*
+    {
+      auto_ptr<object> o (db->query_one<object> ());
+      assert (false);
+    }
+    */
+
+    /*
+    {
+      object o;
+      db->query_one<object> (o);
+      assert (false);
+    }
+    */
+
+    /*
+    {
+      object o (db->query_value<object> ());
+      assert (false);
+    }
+    */
+  }
+  catch (const odb::exception& e)
+  {
+    cerr << e.what () << endl;
+    return 1;
+  }
+}