Add test for queries involving object relationships

1 files changed, 140 insertions, 0 deletions

diff --git a/common/relationship-query/test.hxx b/common/relationship-query/test.hxx
new file mode 100644
index 0000000..c832405
--- /dev/null
+++ b/common/relationship-query/test.hxx
@@ -0,0 +1,140 @@
+// file      : common/relationship-query/test.hxx
+// author    : Boris Kolpackov <boris@codesynthesis.com>
+// copyright : Copyright (c) 2009-2011 Code Synthesis Tools CC
+// license   : GNU GPL v2; see accompanying LICENSE file
+
+#ifndef TEST_HXX
+#define TEST_HXX
+
+#include <common/config.hxx> // HAVE_TR1_MEMORY
+
+#ifdef HAVE_TR1_MEMORY
+
+#include <string>
+
+#include <odb/core.hxx>
+#include <odb/tr1/memory.hxx>
+
+struct country;
+
+#pragma db value
+struct residence_info
+{
+  residence_info (bool p, std::tr1::shared_ptr<country> l)
+      : permanent (p), location (l)
+  {
+  }
+
+  residence_info ()
+  {
+  }
+
+  bool permanent;
+
+  #pragma db not_null
+  std::tr1::shared_ptr<country> location;
+};
+
+#pragma db object pointer(std::tr1::shared_ptr)
+struct person
+{
+  person (unsigned long i,
+          const std::string& fn,
+          const std::string& ln,
+          unsigned short a,
+          std::tr1::shared_ptr<country> r,
+          bool p,
+          std::tr1::shared_ptr<country> n)
+      : id (i),
+        first_name (fn),
+        last_name (ln),
+        age (a),
+        residence (p, r),
+        nationality (n)
+  {
+  }
+
+  person ()
+  {
+  }
+
+  #pragma db id
+  unsigned long id;
+
+  #pragma db column ("first")
+  std::string first_name;
+
+  #pragma db column ("last")
+  std::string last_name;
+
+  unsigned short age;
+
+  residence_info residence;
+
+  #pragma db not_null
+  std::tr1::shared_ptr<country> nationality;
+
+  std::tr1::shared_ptr<person> husband; // Self-join.
+};
+
+struct employer;
+
+#pragma db object pointer(std::tr1::shared_ptr)
+struct employee: person
+{
+  employee (unsigned long i,
+          const std::string& fn,
+          const std::string& ln,
+          unsigned short a,
+          std::tr1::shared_ptr<country> r,
+          bool p,
+          std::tr1::shared_ptr<country> n,
+          std::tr1::shared_ptr<employer> e)
+      : person (i, fn, ln, a, r, p, n),
+        employed_by (e)
+  {
+  }
+
+  employee ()
+  {
+  }
+
+  std::tr1::shared_ptr<employer> employed_by;
+};
+
+#pragma db object pointer(std::tr1::shared_ptr)
+struct employer
+{
+  employer (const std::string& n)
+      : name (n)
+  {
+  }
+
+  employer ()
+  {
+  }
+
+  #pragma db id
+  std::string name;
+};
+
+#pragma db object pointer(std::tr1::shared_ptr)
+struct country
+{
+  country (const std::string& c, std::string const& n)
+      : code (c), name (n)
+  {
+  }
+
+  country ()
+  {
+  }
+
+  #pragma db id
+  std::string code; // ISO 2-letter country code.
+
+  std::string name;
+};
+
+#endif // HAVE_TR1_MEMORY
+#endif // TEST_HXX