From 8ae631a06a7c7599fd9833d232fc86c2e2e7be03 Mon Sep 17 00:00:00 2001
From: Alex Gaetano Padula <apadula@cursusdb.com>
Date: Fri, 11 Oct 2024 00:12:36 -0400
Subject: [PATCH] TestBTree_Put2 test corrections and clean up

---
 btree_test.go | 26 ++++++++++++++------------
 1 file changed, 14 insertions(+), 12 deletions(-)

diff --git a/btree_test.go b/btree_test.go
index 0de952b..0fa1040 100644
--- a/btree_test.go
+++ b/btree_test.go
@@ -111,7 +111,7 @@ func TestBTree_Put2(t *testing.T) {
 
 	defer btree.Close()
 
-	for i := 0; i < 500; i++ {
+	for i := 0; i < 1000; i++ {
 
 		err := btree.Put([]byte(strconv.Itoa(i)), []byte(strconv.Itoa(i)))
 		if err != nil {
@@ -120,18 +120,20 @@ func TestBTree_Put2(t *testing.T) {
 
 	}
 
-	//btree.PrintTree()
+	for i := 0; i < 500; i++ {
+		key, err := btree.Get([]byte(strconv.Itoa(i)))
+		if err != nil {
+			t.Fatal(err)
+		}
 
-	//for i := 0; i < 500; i++ {
-	//	key, err := btree.Get([]byte(strconv.Itoa(i)))
-	//	if err != nil {
-	//		t.Fatal(err)
-	//	}
-	//
-	//	if key == nil {
-	//		t.Fatal("expected key to be not nil")
-	//	}
-	//}
+		if string(key.V[0]) != strconv.Itoa(i) {
+			t.Fatalf("expected value to be %d, got %s", i, key.V[0])
+		}
+
+		if key == nil {
+			t.Fatal("expected key to be not nil")
+		}
+	}
 }
 
 func TestBTree_Delete(t *testing.T) {