aboutsummaryrefslogtreecommitdiff
path: root/evolution/add-foreign-key/driver.cxx
diff options
context:
space:
mode:
Diffstat (limited to 'evolution/add-foreign-key/driver.cxx')
-rw-r--r--evolution/add-foreign-key/driver.cxx140
1 files changed, 140 insertions, 0 deletions
diff --git a/evolution/add-foreign-key/driver.cxx b/evolution/add-foreign-key/driver.cxx
new file mode 100644
index 0000000..0d62889
--- /dev/null
+++ b/evolution/add-foreign-key/driver.cxx
@@ -0,0 +1,140 @@
+// file : evolution/add-foreign-key/driver.cxx
+// copyright : Copyright (c) 2009-2013 Code Synthesis Tools CC
+// license : GNU GPL v2; see accompanying LICENSE file
+
+// Test adding a foreign key.
+//
+
+#include <memory> // std::auto_ptr
+#include <cassert>
+#include <iostream>
+
+#include <odb/database.hxx>
+#include <odb/transaction.hxx>
+
+#include <common/common.hxx>
+
+#include "test1.hxx"
+#include "test2.hxx"
+#include "test1-odb.hxx"
+#include "test2-odb.hxx"
+
+using namespace std;
+using namespace odb::core;
+
+int
+main (int argc, char* argv[])
+{
+ try
+ {
+ auto_ptr<database> db (create_database (argc, argv));
+
+ // 1 - base version
+ // 2 - migration
+ // 3 - current version
+ //
+ unsigned short pass (*argv[argc - 1] - '0');
+
+ switch (pass)
+ {
+ case 1:
+ {
+ using namespace v2;
+
+ object o (1);
+
+ {
+ transaction t (db->begin ());
+ db->persist (o);
+ t.commit ();
+ }
+ break;
+ }
+ case 2:
+ {
+ using namespace v3;
+
+ // Both pointers are now NULL.
+ //
+ {
+ transaction t (db->begin ());
+ auto_ptr<object> p (db->load<object> (1));
+
+ assert (p->o1 == 0 && p->o2 == 0);
+
+ // Migration. The foreign key constraint is not yet there.
+ //
+ p->o1 = new object1 (1);
+ p->o2 = new object2 (1);
+ db->update (*p);
+
+ t.commit ();
+ }
+
+ // Migration. Add the missing objects.
+ //
+ object1 o1 (1);
+ object2 o2 (1);
+
+ {
+ transaction t (db->begin ());
+ db->persist (o1);
+ db->persist (o2);
+ t.commit ();
+ }
+
+ break;
+ }
+ case 3:
+ {
+ using namespace v3;
+
+ {
+ transaction t (db->begin ());
+ auto_ptr<object> p (db->load<object> (1));
+ assert (p->o1->id_ == 1);
+ assert (p->o2->id_ == 1);
+ t.commit ();
+ }
+
+ // Now the foreign key constraint is there.
+ //
+ try
+ {
+ object o (2);
+ o.o1 = new object1 (2);
+ o.o2 = new object2 (2);
+
+ transaction t (db->begin ());
+ db->persist (o);
+ assert (false);
+ }
+ catch (const odb::exception& ) {}
+
+ // As well as the NOT NULL.
+ //
+ try
+ {
+ object o (3);
+ o.o2 = 0;
+
+ transaction t (db->begin ());
+ db->persist (o);
+ assert (false);
+ }
+ catch (const odb::exception& ) {}
+ break;
+ }
+ default:
+ {
+ cerr << "unknown pass number '" << argv[argc - 1] << "'" << endl;
+ return 1;
+ }
+ }
+ }
+ catch (const odb::exception& e)
+ {
+ cerr << e.what () << endl;
+ return 1;
+ }
+}