From 29ccf4aa7a683cd3b5470a24259ed8774aede64f Mon Sep 17 00:00:00 2001
From: Quincey Koziol <koziol@hdfgroup.org>
Date: Thu, 17 Feb 2005 16:43:36 -0500
Subject: [svn-r10038] Purpose:     New test

Description:
    Add tests for splitting leaves in level-2 B-tree

Platforms tested:
    FreeBSD 4.11 (sleipnir)
    Too minor to require h5committest
---
 test/btree2.c | 164 +++++++++++++++++++++++++++++++++++++++++++++++++++-------
 1 file changed, 146 insertions(+), 18 deletions(-)

diff --git a/test/btree2.c b/test/btree2.c
index 4526d96..14bf4a1 100644
--- a/test/btree2.c
+++ b/test/btree2.c
@@ -127,7 +127,7 @@ test_insert_basic(hid_t fapl)
     /*
      * Test inserting record into v2 B-tree 
      */
-    TESTING("B-tree insert");
+    TESTING("B-tree insert - several records");
     record=42;
     if (H5B2_insert(f, H5P_DATASET_XFER_DEFAULT, H5B2_TEST, bt2_addr, &record)<0) {
 	H5_FAILED();
@@ -231,7 +231,7 @@ test_insert_split_root(hid_t fapl)
     /*
      * Test inserting many records into v2 B-tree 
      */
-    TESTING("B-tree many - split root");
+    TESTING("B-tree insert - split root");
 
     for(u=0; u<INSERT_SPLIT_ROOT_NREC; u++) {
         record=u+2;
@@ -332,7 +332,7 @@ test_insert_level1_2leaf_redistrib(hid_t fapl)
     /*
      * Test inserting many records into v2 B-tree 
      */
-    TESTING("B-tree many - redistribute 2 leaves in level 1 B-tree (l->r)");
+    TESTING("B-tree insert - redistribute 2 leaves in level 1 B-tree (l->r)");
 
     /* Insert enough records to force root to split into 2 leaves */
     for(u=0; u<INSERT_SPLIT_ROOT_NREC; u++) {
@@ -367,7 +367,7 @@ test_insert_level1_2leaf_redistrib(hid_t fapl)
     /*
      * Test inserting many records into v2 B-tree 
      */
-    TESTING("B-tree many - redistribute 2 leaves in level 1 B-tree (r->l)");
+    TESTING("B-tree insert - redistribute 2 leaves in level 1 B-tree (r->l)");
 
     /* Insert enough records to force root to split into 2 leaves */
     for(u=0; u<INSERT_SPLIT_ROOT_NREC; u++) {
@@ -455,7 +455,7 @@ test_insert_level1_2leaf_split(hid_t fapl)
     /*
      * Test inserting many records into v2 B-tree 
      */
-    TESTING("B-tree many - split 2 leaves to 3 in level 1 B-tree (l->r)");
+    TESTING("B-tree insert - split 2 leaves to 3 in level 1 B-tree (l->r)");
 
     /* Insert enough records to force root to split into 2 leaves */
     for(u=0; u<INSERT_SPLIT_ROOT_NREC; u++) {
@@ -490,7 +490,7 @@ test_insert_level1_2leaf_split(hid_t fapl)
     /*
      * Test inserting many records into v2 B-tree 
      */
-    TESTING("B-tree many - split 2 leaves to 3 in level 1 B-tree (r->l)");
+    TESTING("B-tree insert - split 2 leaves to 3 in level 1 B-tree (r->l)");
 
     /* Insert enough records to force root to split into 2 leaves */
     for(u=0; u<INSERT_SPLIT_ROOT_NREC; u++) {
@@ -580,7 +580,7 @@ test_insert_level1_3leaf_redistrib(hid_t fapl)
     /*
      * Test inserting many records into v2 B-tree 
      */
-    TESTING("B-tree many - redistribute 3 leaves in level 1 B-tree");
+    TESTING("B-tree insert - redistribute 3 leaves in level 1 B-tree");
 
     /* Insert enough records to force root to split into 2 leaves */
     for(u=0; u<INSERT_SPLIT_ROOT_NREC; u++) {
@@ -681,7 +681,7 @@ test_insert_level1_3leaf_split(hid_t fapl)
     /*
      * Test inserting many records into v2 B-tree 
      */
-    TESTING("B-tree many - split 3 leaves to 4 in level 1 B-tree");
+    TESTING("B-tree insert - split 3 leaves to 4 in level 1 B-tree");
 
     /* Insert enough records to force root to split into 2 leaves */
     for(u=0; u<INSERT_SPLIT_ROOT_NREC; u++) {
@@ -767,7 +767,7 @@ test_insert_make_level2(hid_t fapl)
     /*
      * Test inserting many records into v2 B-tree 
      */
-    TESTING("B-tree many - make level 2 B-tree");
+    TESTING("B-tree insert - make level 2 B-tree");
 
     /* Insert enough records to force root to split into 2 leaves */
     for(u=0; u<INSERT_SPLIT_ROOT_NREC*5; u++) {
@@ -842,24 +842,25 @@ error:
 
 
 /*-------------------------------------------------------------------------
- * Function:	test_insert_level2_2leaf_redistrib
+ * Function:	test_insert_level2_leaf_redistrib
  *
  * Purpose:	Basic tests for the B-tree v2 code.  This test inserts enough
- *              records to make a level 2 B-tree
+ *              records to make a level 2 B-tree and then adds enough more
+ *              records to force the leaves to redistribute
  *
  * Return:	Success:	0
  *
  *		Failure:	1
  *
  * Programmer:	Quincey Koziol
- *              Friday, February 11, 2005
+ *              Thursday, February 17, 2005
  *
  * Modifications:
  *
  *-------------------------------------------------------------------------
  */
 static int
-test_insert_level2_2leaf_redistrib(hid_t fapl)
+test_insert_level2_leaf_redistrib(hid_t fapl)
 {
     hid_t	file=-1;
     char	filename[1024];
@@ -892,7 +893,7 @@ test_insert_level2_2leaf_redistrib(hid_t fapl)
     /*
      * Test inserting many records into v2 B-tree 
      */
-    TESTING("B-tree many - redistrib right-most leaf in level 2 B-tree");
+    TESTING("B-tree insert - redistrib right-most leaf in level 2 B-tree");
 
     /* Insert enough records to force root to split into 2 internal nodes */
     /* Also redistributes right leaf */
@@ -915,7 +916,7 @@ test_insert_level2_2leaf_redistrib(hid_t fapl)
 
     PASSED();
 
-    TESTING("B-tree many - redistrib left-most leaf in level 2 B-tree");
+    TESTING("B-tree insert - redistrib left-most leaf in level 2 B-tree");
 
     /* Add more records to left-most leaf, to force a 2 node redistribution on left leaf */
     for(u=0; u<INSERT_SPLIT_ROOT_NREC/2; u++) {
@@ -929,7 +930,7 @@ test_insert_level2_2leaf_redistrib(hid_t fapl)
 
     PASSED();
 
-    TESTING("B-tree many - redistrib middle leaf in level 2 B-tree");
+    TESTING("B-tree insert - redistrib middle leaf in level 2 B-tree");
 
     /* Add more records to middle leaf, to force a 3 node redistribution on middle leaf */
     for(u=0; u<INSERT_SPLIT_ROOT_NREC/2; u++) {
@@ -963,7 +964,133 @@ error:
 	H5Fclose(file);
     } H5E_END_TRY;
     return 1;
-} /* test_insert_level2_2leaf_redistrib() */
+} /* test_insert_level2_leaf_redistrib() */
+
+
+/*-------------------------------------------------------------------------
+ * Function:	test_insert_level2_leaf_split
+ *
+ * Purpose:	Basic tests for the B-tree v2 code.  This test inserts enough
+ *              records to make a level 2 B-tree and then adds enough more
+ *              records to force leaves to split.
+ *
+ * Return:	Success:	0
+ *
+ *		Failure:	1
+ *
+ * Programmer:	Quincey Koziol
+ *              Thursday, February 17, 2005
+ *
+ * Modifications:
+ *
+ *-------------------------------------------------------------------------
+ */
+static int
+test_insert_level2_leaf_split(hid_t fapl)
+{
+    hid_t	file=-1;
+    char	filename[1024];
+    H5F_t	*f=NULL;
+    hsize_t     record;                 /* Record to insert into tree */
+    haddr_t     bt2_addr;               /* Address of B-tree created */
+    hsize_t     idx;                    /* Index within B-tree, for iterator */
+    unsigned    u;                      /* Local index variable */
+
+    h5_fixname(FILENAME[0], fapl, filename, sizeof filename);
+
+    /* Create the file to work on */
+    if ((file=H5Fcreate(filename, H5F_ACC_TRUNC, H5P_DEFAULT, fapl))<0) TEST_ERROR;
+	
+    /* Get a pointer to the internal file object */
+    if (NULL==(f=H5I_object(file))) {
+	H5Eprint_stack(H5E_DEFAULT, stdout);
+	goto error;
+    }
+
+    /*
+     * Create v2 B-tree 
+     */
+    if (H5B2_create(f, H5P_DATASET_XFER_DEFAULT, H5B2_TEST, 512, 8, 100, 40, &bt2_addr/*out*/)<0) {
+	H5_FAILED();
+	H5Eprint_stack(H5E_DEFAULT, stdout);
+	goto error;
+    }
+
+    /*
+     * Test inserting many records into v2 B-tree 
+     */
+    TESTING("B-tree insert - split right-most leaf in level 2 B-tree");
+
+    /* Insert enough records to force root to split into 2 internal nodes */
+    /* Also splits right leaf */
+    for(u=0; u<INSERT_SPLIT_ROOT_NREC*4; u++) {
+        record=u+INSERT_SPLIT_ROOT_NREC;
+        if (H5B2_insert(f, H5P_DATASET_XFER_DEFAULT, H5B2_TEST, bt2_addr, &record)<0) {
+            H5_FAILED();
+            H5Eprint_stack(H5E_DEFAULT, stdout);
+            goto error;
+        }
+    }
+    for(; u<INSERT_SPLIT_ROOT_NREC*12; u++) {
+        record=u+(INSERT_SPLIT_ROOT_NREC*2);
+        if (H5B2_insert(f, H5P_DATASET_XFER_DEFAULT, H5B2_TEST, bt2_addr, &record)<0) {
+            H5_FAILED();
+            H5Eprint_stack(H5E_DEFAULT, stdout);
+            goto error;
+        }
+    }
+
+    PASSED();
+
+    TESTING("B-tree insert - split left-most leaf in level 2 B-tree");
+
+    /* Add more records to left-most leaf, to force a 2->3 node split on left leaf */
+    for(u=0; u<INSERT_SPLIT_ROOT_NREC; u++) {
+        record=u;
+        if (H5B2_insert(f, H5P_DATASET_XFER_DEFAULT, H5B2_TEST, bt2_addr, &record)<0) {
+            H5_FAILED();
+            H5Eprint_stack(H5E_DEFAULT, stdout);
+            goto error;
+        }
+    }
+
+    PASSED();
+
+    TESTING("B-tree insert - split middle leaf in level 2 B-tree");
+
+    /* Add more records to middle leaf, to force a 3->4 node split on middle leaf */
+    for(u=0; u<INSERT_SPLIT_ROOT_NREC; u++) {
+        record=u+(INSERT_SPLIT_ROOT_NREC*5);
+        if (H5B2_insert(f, H5P_DATASET_XFER_DEFAULT, H5B2_TEST, bt2_addr, &record)<0) {
+            H5_FAILED();
+            H5Eprint_stack(H5E_DEFAULT, stdout);
+            goto error;
+        }
+    }
+
+    /* Iterate over B-tree to check records have been inserted correctly */
+    idx = 0;
+    if(H5B2_iterate(f, H5P_DATASET_XFER_DEFAULT, H5B2_TEST, bt2_addr, iter_cb, &idx)<0) {
+	H5_FAILED();
+	H5Eprint_stack(H5E_DEFAULT, stdout);
+	goto error;
+    }
+
+    /* Make certain that the index is correct */
+    if(idx != (INSERT_SPLIT_ROOT_NREC*14)) TEST_ERROR;
+
+    PASSED();
+
+    if (H5Fclose(file)<0) TEST_ERROR;
+
+    return 0;
+
+error:
+    H5E_BEGIN_TRY {
+	H5Fclose(file);
+    } H5E_END_TRY;
+    return 1;
+} /* test_insert_level2_leaf_redistrib() */
 
 
 /*-------------------------------------------------------------------------
@@ -1000,7 +1127,8 @@ main(void)
     nerrors += test_insert_level1_3leaf_redistrib(fapl);
     nerrors += test_insert_level1_3leaf_split(fapl);
     nerrors += test_insert_make_level2(fapl);
-    nerrors += test_insert_level2_2leaf_redistrib(fapl);
+    nerrors += test_insert_level2_leaf_redistrib(fapl);
+    nerrors += test_insert_level2_leaf_split(fapl);
 
     if (nerrors) goto error;
     puts("All v2 B-tree tests passed.");
-- 
cgit v0.12