add test helpers

Signed-off-by: Mustafa Elbehery <melbeher@redhat.com>

Author: Elbehery

bucket_test.go

@@ -464,6 +464,81 @@ func TestBucket_Delete_NonExisting(t *testing.T) {
 	}
 }
 
+func TestBucket_MoveBucket2(t *testing.T) {
+	testCases := []struct {
+		name          string
+		srcBucketPath []string
+		dstBucketPath []string
+		isInlined     bool
+		expErr        error
+	}{
+		{
+			name:          "happy path",
+			srcBucketPath: []string{"x", "y", "z"},
+			dstBucketPath: []string{"x", "y"},
+			isInlined:     false,
+			expErr:        nil,
+		},
+	}
+
+	for _, tc := range testCases {
+		t.Run(tc.name, func(*testing.T) {
+			db := btesting.MustCreateDB(t)
+			if err := db.Update(func(tx *bolt.Tx) error {
+
+				bk := createBucketIfNotExist(t, tx, tc.srcBucketPath...)
+				insertRandKeysValuesBucket(t, bk, rand.Int())
+
+				return nil
+			}); err != nil {
+				t.Fatal(err)
+			}
+			db.MustCheck()
+		})
+	}
+}
+func createBucketIfNotExist(t testing.TB, tx *bolt.Tx, paths ...string) *bolt.Bucket {
+	t.Helper()
+
+	bk, err := tx.CreateBucketIfNotExists([]byte(paths[0]))
+	if err != nil {
+		t.Fatalf("error creating bucket %v: %v", paths[0], err)
+	}
+
+	for _, key := range paths[1:] {
+		bk, err = bk.CreateBucketIfNotExists([]byte(key))
+		t.Fatalf("error creating bucket %v: %v", key, err)
+	}
+
+	return bk
+}
+
+func insertRandKeysValuesBucket(t testing.TB, bk *bolt.Bucket, n int) {
+
+	var min, max = 1, 1024
+	var letters = []rune("abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ")
+
+	for i := 0; i < n; i++ {
+		// generate rand key/value length
+		keyLength := rand.Intn(max-min) + min
+		valLength := rand.Intn(max-min) + min
+
+		keyData := make([]rune, keyLength)
+		valData := make([]rune, valLength)
+
+		for i := range keyData {
+			keyData[i] = letters[rand.Intn(len(letters))]
+			valData[i] = letters[rand.Intn(len(letters))]
+		}
+
+		pErr := bk.Put([]byte(string(keyData)), []byte(string(valData)))
+		if pErr != nil {
+			t.Fatalf("error inserting key %v and value %v in bucket %v: %v", string(keyData), string(valData), bk.String(), pErr)
+		}
+	}
+
+}
+
 func generateTestKeysValues(t testing.TB, srcBucketPath []string, db *btesting.DB) {
 	t.Helper()
 
@@ -540,34 +615,6 @@ func generateTestKeysValues(t testing.TB, srcBucketPath []string, db *btesting.D
 	db.MustCheck()
 }
 
-func TestBucket_MoveBucket2(t *testing.T) {
-	testCases := []struct {
-		name          string
-		srcBucketPath []string
-		dstBucketPath []string
-		isInlined     bool
-		expErr        error
-	}{
-		{
-			name:          "happy path",
-			srcBucketPath: []string{"x", "y", "z"},
-			dstBucketPath: []string{"x", "y"},
-			isInlined:     false,
-			expErr:        nil,
-		},
-	}
-
-	for _, tc := range testCases {
-		t.Run(tc.name, func(*testing.T) {
-
-			db := btesting.MustCreateDB(t)
-
-			generateTestKeysValues(t, tc.srcBucketPath, db)
-
-		})
-	}
-}
-
 func TestBucket_MoveBucket(t *testing.T) {
 	testCases := []struct {
 		name                    string