1 files changed, 84 insertions, 0 deletions
diff --git a/test/format/rebalance.c b/test/format/rebalance.c
new file mode 100644
index 00000000000..477368c36db
--- /dev/null
+++ b/test/format/rebalance.c
@@ -0,0 +1,84 @@
+/*-
+ * Public Domain 2014-2015 MongoDB, Inc.
+ * Public Domain 2008-2014 WiredTiger, Inc.
+ *
+ * This is free and unencumbered software released into the public domain.
+ *
+ * Anyone is free to copy, modify, publish, use, compile, sell, or
+ * distribute this software, either in source code form or as a compiled
+ * binary, for any purpose, commercial or non-commercial, and by any
+ * means.
+ *
+ * In jurisdictions that recognize copyright laws, the author or authors
+ * of this software dedicate any and all copyright interest in the
+ * software to the public domain. We make this dedication for the benefit
+ * of the public at large and to the detriment of our heirs and
+ * successors. We intend this dedication to be an overt act of
+ * relinquishment in perpetuity of all present and future rights to this
+ * software under copyright law.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+ * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+ * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
+ * IN NO EVENT SHALL THE AUTHORS BE LIABLE FOR ANY CLAIM, DAMAGES OR
+ * OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
+ * ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
+ * OTHER DEALINGS IN THE SOFTWARE.
+ */
+
+#include "format.h"
+
+void
+wts_rebalance(void)
+{
+	WT_CONNECTION *conn;
+	WT_SESSION *session;
+	int ret;
+	char cmd[1024];
+
+	if (g.c_rebalance == 0)
+		return;
+
+	track("rebalance", 0ULL, NULL);
+
+	/* Dump the current object. */
+	(void)snprintf(cmd, sizeof(cmd),
+	    "../../wt -h %s dump -f %s/rebalance.orig %s",
+	    g.home, g.home, g.uri);
+	if ((ret = system(cmd)) != 0)
+		die(ret, "command failed: %s", cmd);
+
+	/* Rebalance, then verify the object. */
+	wts_reopen();
+	conn = g.wts_conn;
+	if ((ret = conn->open_session(conn, NULL, NULL, &session)) != 0)
+		die(ret, "connection.open_session");
+	if (g.logging != 0)
+		(void)g.wt_api->msg_printf(g.wt_api, session,
+		    "=============== rebalance start ===============");
+
+	if ((ret = session->rebalance(session, g.uri, NULL)) != 0)
+		die(ret, "session.rebalance: %s: %s", g.uri);
+
+	if (g.logging != 0)
+		(void)g.wt_api->msg_printf(g.wt_api, session,
+		    "=============== rebalance stop ===============");
+	if ((ret = session->close(session, NULL)) != 0)
+		die(ret, "session.close");
+
+	wts_verify("post-rebalance verify");
+	wts_close();
+
+	(void)snprintf(cmd, sizeof(cmd),
+	    "../../wt -h %s dump -f %s/rebalance.new %s",
+	    g.home, g.home, g.uri);
+	if ((ret = system(cmd)) != 0)
+		die(ret, "command failed: %s", cmd);
+
+	/* Compare the old/new versions of the object. */
+	(void)snprintf(cmd, sizeof(cmd),
+	    "cmp %s/rebalance.orig %s/rebalance.new > /dev/null",
+	    g.home, g.home);
+	if ((ret = system(cmd)) != 0)
+		die(ret, "command failed: %s", cmd);
+}