1 files changed, 91 insertions, 0 deletions
diff --git a/storage/tokudb/PerconaFT/src/tests/update.cc b/storage/tokudb/PerconaFT/src/tests/update.cc
new file mode 100644
index 00000000000..e569660525c
--- /dev/null
+++ b/storage/tokudb/PerconaFT/src/tests/update.cc
@@ -0,0 +1,91 @@
+/* -*- mode: C++; c-basic-offset: 4; indent-tabs-mode: nil -*- */
+// vim: ft=cpp:expandtab:ts=8:sw=4:softtabstop=4:
+#ident "$Id$"
+/*======
+This file is part of PerconaFT.
+
+
+Copyright (c) 2006, 2015, Percona and/or its affiliates. All rights reserved.
+
+    PerconaFT is free software: you can redistribute it and/or modify
+    it under the terms of the GNU General Public License, version 2,
+    as published by the Free Software Foundation.
+
+    PerconaFT is distributed in the hope that it will be useful,
+    but WITHOUT ANY WARRANTY; without even the implied warranty of
+    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+    GNU General Public License for more details.
+
+    You should have received a copy of the GNU General Public License
+    along with PerconaFT.  If not, see <http://www.gnu.org/licenses/>.
+
+----------------------------------------
+
+    PerconaFT is free software: you can redistribute it and/or modify
+    it under the terms of the GNU Affero General Public License, version 3,
+    as published by the Free Software Foundation.
+
+    PerconaFT is distributed in the hope that it will be useful,
+    but WITHOUT ANY WARRANTY; without even the implied warranty of
+    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+    GNU Affero General Public License for more details.
+
+    You should have received a copy of the GNU Affero General Public License
+    along with PerconaFT.  If not, see <http://www.gnu.org/licenses/>.
+======= */
+
+#ident "Copyright (c) 2006, 2015, Percona and/or its affiliates. All rights reserved."
+
+/* test the update functionality. */
+
+#include "test.h"
+
+DB_ENV *env;
+
+// the commands are: byte 1 is "nop" "add" or "del".  Byte 2 is the amount to add.
+enum cmd { CNOP, CADD, CDEL };
+
+static int increment_update (DB *db __attribute__((__unused__)),
+                             const DBT *key __attribute__((__unused__)),
+                             const DBT *old_val, const DBT *extra,
+                             void (*set_val)(const DBT *new_val,
+                                             void *set_extra),
+                             void *set_extra) {
+    assert (extra->size==2);
+    assert (old_val->size==4);
+    unsigned char *CAST_FROM_VOIDP(extra_data, extra->data);
+    switch ((enum cmd)(extra_data[0])) {
+    case CNOP:
+        return 0;
+    case CADD: {
+        unsigned int data = *(unsigned int*)old_val->data;
+        data += extra_data[1];
+        DBT new_val = {.data=&data, .size=4, .ulen=0, .flags=0};
+        set_val(&new_val, set_extra);
+        return 0;
+    }
+    case CDEL:
+        set_val(NULL, set_extra);
+        return 0;
+    }
+    assert(0); return 0; // enumeration failed.
+}
+
+static void setup (void) {
+    toku_os_recursive_delete(TOKU_TEST_FILENAME);
+    { int r=toku_os_mkdir(TOKU_TEST_FILENAME, S_IRWXU+S_IRWXG+S_IRWXO);   CKERR(r); }
+    { int r=db_env_create(&env, 0);                           CKERR(r); }
+    env->set_errfile(env, stderr);
+    env->set_update(env, increment_update);
+}
+
+static void cleanup (void) {
+    { int r = env->close(env, 0);                             CKERR(r); }
+}
+
+int test_main (int argc __attribute__((__unused__)), char *const argv[] __attribute__((__unused__))) {
+
+    setup();
+    cleanup();
+    return 0;
+}