Issue 121: JaQu simple update and merge methods

h2/src/test/org/h2/test/TestAll.java

@@ -62,6 +62,7 @@ import org.h2.test.db.TestTwoPhaseCommit;
 import org.h2.test.db.TestView;
 import org.h2.test.jaqu.AliasMapTest;
 import org.h2.test.jaqu.SamplesTest;
+import org.h2.test.jaqu.UpdateTest;
 import org.h2.test.jdbc.TestBatchUpdates;
 import org.h2.test.jdbc.TestCallableStatement;
 import org.h2.test.jdbc.TestCancel;
@@ -530,6 +531,7 @@ kill -9 `jps -l | grep "org.h2.test." | cut -d " " -f 1`
         // jaqu
         new AliasMapTest().runTest(this);
         new SamplesTest().runTest(this);
+        new UpdateTest().runTest(this);
 
         // jdbc
         new TestBatchUpdates().runTest(this);

h2/src/test/org/h2/test/jaqu/UpdateTest.java

+/*
+ * Copyright 2004-2009 H2 Group. Multiple-Licensed under the H2 License,
+ * Version 1.0, and under the Eclipse Public License, Version 1.0
+ * (http://h2database.com/html/license.html).
+ * Initial Developer: H2 Group
+ */
+package org.h2.test.jaqu;
+
+import org.h2.jaqu.Db;
+import org.h2.test.TestBase;
+
+import static java.sql.Date.valueOf;
+
+/**
+ * Tests the Db.update() function.
+ *
+ * @author dmoebius at scoop slash gmbh dot de
+ */
+public class UpdateTest extends TestBase {
+
+    Db db;
+
+    /**
+     * This method is called when executing this application from the command
+     * line.
+     *
+     * @param args the command line parameters
+     */
+    public static void main(String[] args) throws Exception {
+        new UpdateTest().test();
+    }
+
+    public void test() throws Exception {
+        db = Db.open("jdbc:h2:mem:", "sa", "sa");
+        db.insertAll(Product.getList());
+        db.insertAll(Customer.getList());
+        db.insertAll(Order.getList());
+
+        testSimpleUpdate();
+        testSimpleUpdateWithCombinedPrimaryKey();
+        testSimpleMerge();
+        testSimpleMergeWithCombinedPrimaryKey();
+
+        db.close();
+    }
+
+    private void testSimpleUpdate() {
+        Product p = new Product();
+        Product pChang = db.from(p).where(p.productName).is("Chang").selectFirst();
+        // update unitPrice from 19.0 to 19.5
+        pChang.unitPrice = 19.5;
+        // update unitsInStock from 17 to 16
+        pChang.unitsInStock = 16;
+        db.update(pChang);
+
+        Product p2 = new Product();
+        Product pChang2 = db.from(p2).where(p2.productName).is("Chang").selectFirst();
+        assertEquals(19.5, pChang2.unitPrice);
+        assertEquals(new Integer(16), pChang2.unitsInStock);
+
+        // undo update
+        pChang.unitPrice = 19.0;
+        pChang.unitsInStock = 17;
+        db.update(pChang);
+    }
+
+    private void testSimpleUpdateWithCombinedPrimaryKey() {
+        Order o = new Order();
+        Order ourOrder = db.from(o).where(o.orderDate).is(valueOf("2007-01-02")).selectFirst();
+        ourOrder.orderDate = valueOf("2007-01-03");
+        db.update(ourOrder);
+
+        Order ourUpdatedOrder = db.from(o).where(o.orderDate).is(valueOf("2007-01-03")).selectFirst();
+        assertTrue("updated order not found", ourUpdatedOrder != null);
+
+        // undo update
+        ourOrder.orderDate = valueOf("2007-01-02");
+        db.update(ourOrder);
+    }
+
+    private void testSimpleMerge() {
+        Product p = new Product();
+        Product pChang = db.from(p).where(p.productName).is("Chang").selectFirst();
+        // update unitPrice from 19.0 to 19.5
+        pChang.unitPrice = 19.5;
+        // update unitsInStock from 17 to 16
+        pChang.unitsInStock = 16;
+        db.merge(pChang);
+
+        Product p2 = new Product();
+        Product pChang2 = db.from(p2).where(p2.productName).is("Chang").selectFirst();
+        assertEquals(19.5, pChang2.unitPrice);
+        assertEquals(new Integer(16), pChang2.unitsInStock);
+
+        // undo update
+        pChang.unitPrice = 19.0;
+        pChang.unitsInStock = 17;
+        db.merge(pChang);
+    }
+
+    private void testSimpleMergeWithCombinedPrimaryKey() {
+        Order o = new Order();
+        Order ourOrder = db.from(o).where(o.orderDate).is(valueOf("2007-01-02")).selectFirst();
+        ourOrder.orderDate = valueOf("2007-01-03");
+        db.merge(ourOrder);
+
+        Order ourUpdatedOrder = db.from(o).where(o.orderDate).is(valueOf("2007-01-03")).selectFirst();
+        assertTrue("updated order not found", ourUpdatedOrder != null);
+
+        // undo update
+        ourOrder.orderDate = valueOf("2007-01-02");
+        db.merge(ourOrder);
+    }
+
+}