add tests for basic write values

DracoLi · DracoLi · commit 4461ccf08afc · 2025-10-03T12:32:49.000-04:00
diff --git a/x/blockdb/readblock_test.go b/x/blockdb/readblock_test.go
@@ -263,7 +263,7 @@ func TestHasBlock(t *testing.T) {
 		{
 			name:     "db_closed",
 			dbClosed: true,
-			wantErr:  ErrDatabaseClosed,
+			wantErr:  database.ErrClosed,
 		},
 	}
 
@@ -276,14 +276,14 @@ func TestHasBlock(t *testing.T) {
 				if i == gapHeight {
 					continue
 				}
-				require.NoError(t, store.WriteBlock(i, randomBlock(t)))
+				require.NoError(t, store.Put(i, randomBlock(t)))
 			}
 
 			if tc.dbClosed {
 				require.NoError(t, store.Close())
 			}
 
-			has, err := store.HasBlock(tc.height)
+			has, err := store.Has(tc.height)
 			if tc.wantErr != nil {
 				require.ErrorIs(t, err, tc.wantErr)
 			} else {
diff --git a/x/blockdb/writeblock_test.go b/x/blockdb/writeblock_test.go
@@ -20,6 +20,41 @@ import (
 )
 
 func TestWriteBlock_Basic(t *testing.T) {
+	tests := []struct {
+		name  string
+		block []byte
+		want  []byte
+	}{
+		{
+			name:  "normal write",
+			block: []byte("hello"),
+			want:  []byte("hello"),
+		},
+		{
+			name:  "empty block",
+			block: []byte{},
+			want:  []byte{},
+		},
+		{
+			name:  "nil block",
+			block: nil,
+			want:  []byte{},
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			db, cleanup := newTestDatabase(t, DefaultConfig())
+			defer cleanup()
+			require.NoError(t, db.Put(0, tt.block))
+
+			got, err := db.Get(0)
+			require.NoError(t, err)
+			require.Equal(t, tt.want, got)
+		})
+	}
+}
+
+func TestWriteBlock_MaxHeight(t *testing.T) {
 	customConfig := DefaultConfig().WithMinimumHeight(10)
 
 	tests := []struct {
