chore(cmd): migrate page command to cobra style

Signed-off-by: Lavish Pal <[email protected]>

Author: lavishpal

cmd/bbolt/command_page_test.go

@@ -0,0 +1,82 @@
+package main_test
+
+import (
+	"bytes"
+	"errors"
+	"fmt"
+	"io"
+	"testing"
+
+	"github.com/stretchr/testify/require"
+
+	"go.etcd.io/bbolt"
+	main "go.etcd.io/bbolt/cmd/bbolt"
+	"go.etcd.io/bbolt/internal/btesting"
+)
+
+// Ensure the "page" command neither panics nor changes the db file.
+func TestPageCommand_Run(t *testing.T) {
+	t.Log("Creating sample DB")
+	// Create a test database
+	db := btesting.MustCreateDB(t)
+	err := db.Update(func(tx *bbolt.Tx) error {
+		// Create buckets and insert sample data
+		for _, name := range []string{"foo", "bar"} {
+			b, err := tx.CreateBucket([]byte(name))
+			if err != nil {
+				return err
+			}
+			for i := 0; i < 3; i++ {
+				key := fmt.Sprintf("%s-%d", name, i)
+				val := fmt.Sprintf("val-%s-%d", name, i)
+				if err := b.Put([]byte(key), []byte(val)); err != nil {
+					return err
+				}
+			}
+		}
+		return nil
+	})
+	require.NoError(t, err)
+	db.Close()
+	defer requireDBNoChange(t, dbData(t, db.Path()), db.Path())
+
+	// Running the page command
+	t.Log("Running page cmd with a valid page ID")
+	rootCmd := main.NewRootCommand()
+	outputBuf := bytes.NewBufferString("")
+	rootCmd.SetOut(outputBuf)
+
+	rootCmd.SetArgs([]string{"page", db.Path(), "0"})
+	err = rootCmd.Execute()
+	require.NoError(t, err)
+
+	// Check that the command does not panic and generates output
+	t.Log("Checking output for page command")
+	_, err = io.ReadAll(outputBuf)
+	require.NoError(t, err)
+
+	// Verify output includes expected headers
+	require.Contains(t, outputBuf.String(), "Page ID    TYPE      ITEMS   OVRFLW")
+
+	// Test Case 2: Handle missing page ID argument
+	t.Log("Running page cmd with missing page ID argument")
+	rootCmd.SetArgs([]string{"page", db.Path()})
+	err = rootCmd.Execute()
+	require.ErrorContains(t, err, "page ID(s) required")
+
+	// Test Case 3: Handle invalid page ID argument (non-numeric)
+	t.Log("Running page cmd with invalid page ID argument")
+	rootCmd.SetArgs([]string{"page", db.Path(), "invalid"})
+	err = rootCmd.Execute()
+	require.ErrorContains(t, err, "invalid page ID")
+}
+
+// Ensure the "page" command handles missing database path gracefully.
+func TestPageCommand_NoArgs(t *testing.T) {
+	t.Log("Running page cmd with missing database path")
+	rootCmd := main.NewRootCommand()
+	expErr := errors.New("accepts 2 arg(s), received 1")
+	rootCmd.SetArgs([]string{"page"})
+	err := rootCmd.Execute()
+	require.ErrorContains(t, err, expErr.Error())
+}

cmd/bbolt/main.go

@@ -132,8 +132,6 @@ func (m *Main) Run(args ...string) error {
 		return newInfoCommand(m).Run(args[1:]...)
 	case "keys":
 		return newKeysCommand(m).Run(args[1:]...)
-	case "page":
-		return newPageCommand(m).Run(args[1:]...)
 	case "pages":
 		return newPagesCommand(m).Run(args[1:]...)
 	case "stats":

cmd/bbolt/main_test.go

@@ -93,34 +93,6 @@ func TestDumpCommand_Run(t *testing.T) {
 		t.Fatalf("unexpected stdout:\n%s\n", m.Stdout.String())
 	}
 }
-
-func TestPageCommand_Run(t *testing.T) {
-	db := btesting.MustCreateDBWithOption(t, &bolt.Options{PageSize: 4096})
-	db.Close()
-
-	defer requireDBNoChange(t, dbData(t, db.Path()), db.Path())
-
-	exp := "Page ID:    0\n" +
-		"Page Type:  meta\n" +
-		"Total Size: 4096 bytes\n" +
-		"Overflow pages: 0\n" +
-		"Version:    2\n" +
-		"Page Size:  4096 bytes\n" +
-		"Flags:      00000000\n" +
-		"Root:       <pgid=3>\n" +
-		"Freelist:   <pgid=2>\n" +
-		"HWM:        <pgid=4>\n" +
-		"Txn ID:     0\n" +
-		"Checksum:   07516e114689fdee\n\n"
-
-	m := NewMain()
-	err := m.Run("page", db.Path(), "0")
-	require.NoError(t, err)
-	if m.Stdout.String() != exp {
-		t.Fatalf("unexpected stdout:\n%s\n%s", m.Stdout.String(), exp)
-	}
-}
-
 func TestPageItemCommand_Run(t *testing.T) {
 	testCases := []struct {
 		name          string