diff options
Diffstat (limited to 'storage/tokudb/PerconaFT/src/tests/test-xa-prepare.cc')
| -rw-r--r-- | storage/tokudb/PerconaFT/src/tests/test-xa-prepare.cc | 157 |
1 files changed, 157 insertions, 0 deletions
diff --git a/storage/tokudb/PerconaFT/src/tests/test-xa-prepare.cc b/storage/tokudb/PerconaFT/src/tests/test-xa-prepare.cc new file mode 100644 index 00000000..ba04be56 --- /dev/null +++ b/storage/tokudb/PerconaFT/src/tests/test-xa-prepare.cc @@ -0,0 +1,157 @@ +/* -*- 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." + +#include "test.h" +#include <sys/wait.h> + +static void clean_env (const char *envdir) { + const int len = strlen(envdir)+100; + char cmd[len]; + snprintf(cmd, len, "rm -rf %s", envdir); + int r = system(cmd); + CKERR(r); + CKERR(toku_os_mkdir(envdir, S_IRWXU+S_IRWXG+S_IRWXO)); +} + +static void setup_env (DB_ENV **envp, const char *envdir) { + { int chk_r = db_env_create(envp, 0); CKERR(chk_r); } + (*envp)->set_errfile(*envp, stderr); + { int chk_r = (*envp)->set_redzone(*envp, 0); CKERR(chk_r); } + { int chk_r = (*envp)->open(*envp, envdir, DB_INIT_LOCK|DB_INIT_LOG|DB_INIT_MPOOL|DB_INIT_TXN|DB_CREATE|DB_PRIVATE|DB_RECOVER, S_IRWXU+S_IRWXG+S_IRWXO); CKERR(chk_r); } +} + +const unsigned int myformatid = 0x74736554; + +static void setup_env_and_prepare (DB_ENV **envp, const char *envdir, bool commit) { + DB *db; + DB_TXN *txn; + clean_env(envdir); + setup_env(envp, envdir); + CKERR(db_create(&db, *envp, 0)); + CKERR(db->open(db, NULL, "foo.db", 0, DB_BTREE, DB_CREATE | DB_AUTO_COMMIT, S_IRWXU+S_IRWXG+S_IRWXO)); + CKERR((*envp)->txn_begin(*envp, 0, &txn, 0)); + DBT key; + dbt_init(&key, "foo", 4); + CKERR(db->put(db, txn, &key, &key, 0)); + { int chk_r = db->close(db, 0); CKERR(chk_r); } + TOKU_XA_XID x = {.formatID = myformatid, + .gtrid_length = 8, + .bqual_length = 9}; + for (int i=0; i<8+9; i++) x.data[i] = 42+i; + CKERR(txn->xa_prepare(txn, &x, 0)); + if (commit) + CKERR(txn->commit(txn, 0)); +} + +static void test1 (void) { + pid_t pid; + bool do_fork = true; + if (!do_fork || 0==(pid=fork())) { + DB_ENV *env; + setup_env_and_prepare(&env, TOKU_TEST_FILENAME, false); + { + TOKU_XA_XID l[1]; + long count=-1; + CKERR(env->txn_xa_recover(env, l, 1, &count, DB_FIRST)); + printf("%s:%d count=%ld\n", __FILE__, __LINE__, count); + assert(count==1); + assert(myformatid==l[0].formatID); + assert( 8==l[0].gtrid_length); + assert( 9==l[0].bqual_length); + for (int i=0; i<8+9; i++) { + assert(l[0].data[i]==42+i); + } + } + exit(0); + } + int status; + if (do_fork) { + pid_t pid2 = wait(&status); + assert(pid2==pid); + } + + DB_ENV *env2; + char envdir2[TOKU_PATH_MAX+1]; + setup_env_and_prepare(&env2, toku_path_join(envdir2, 2, TOKU_TEST_FILENAME, "envdir2"), true); + + // Now we can look at env2 in the debugger to see if we managed to make it the same + + DB_ENV *env; + setup_env(&env, TOKU_TEST_FILENAME); + + { + TOKU_XA_XID l[1]; + long count=-1; + { + int r = env->txn_xa_recover(env, l, 1, &count, DB_FIRST); + printf("r=%d count=%ld\n", r, count); + } + assert(count==1); + assert(l[0].data[0]==42); + assert(myformatid==l[0].formatID); + assert( 8 ==l[0].gtrid_length); + assert( 9 ==l[0].bqual_length); + for (int i=0; i<8+9; i++) { + assert(l[0].data[i]==42+i); + } + { + DB_TXN *txn; + int r = env->get_txn_from_xid(env, &l[0], &txn); + assert(r==0); + { int chk_r = txn->commit(txn, 0); CKERR(chk_r); } + } + } + { int chk_r = env2->close(env2, 0); CKERR(chk_r); } + { int chk_r = env ->close(env, 0); CKERR(chk_r); } +} + +int test_main (int argc, char *const argv[]) { + default_parse_args(argc, argv); + // first test: open an environment, a db, a txn, and do a prepare. Then do txn_prepare (without even closing the environment). + test1(); + + + // second test: poen environment, a db, a txn, prepare, close the environment. Then reopen and do txn_prepare. + + // third test: make sure there is an fsync on txn_prepare, but not on the following commit. + + + // Then close the environment Find out what BDB does when ask for the txn prepares. + // Other tests: read prepared txns, 1 at a time. Then close it and read them again. + return 0; +} |