fix GPT partition table tests (#179)

Signed-off-by: Avi Deitcher <[email protected]>
diskfs · Apr 9, 2023 · fd11a09 · fd11a09
1 parent 5b6d86c
commit fd11a09
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 6 deletions.
diff --git a/partition/gpt/table_internal_test.go b/partition/gpt/table_internal_test.go
@@ -40,10 +40,7 @@ func GetValidTable() *Table {
 			physicalSectorSize: 512,
 		},
 	}
-	// add 127 Unused partitions to the table
-	for i := 1; i < 128; i++ {
-		parts = append(parts, &Partition{Type: Unused})
-	}
+	// there are 127 Unused partitions, but those are ignored
 	table.Partitions = parts
 	return &table
 }

diff --git a/partition/gpt/table_test.go b/partition/gpt/table_test.go
@@ -358,7 +358,7 @@ func TestTableWrite(t *testing.T) {
 }
 func TestGetPartitionSize(t *testing.T) {
 	table := GetValidTable()
-	request := 1
+	request := 0
 	size := table.Partitions[request].GetSize()
 	expected := int64(table.Partitions[request].Size)
 	if size != expected {
@@ -370,7 +370,7 @@ func TestGetPartitionStart(t *testing.T) {
 	maxPart := len(table.Partitions)
 	request := maxPart - 1
 	start := table.Partitions[request].GetStart()
-	expected := int64(table.Partitions[request].Start)
+	expected := int64(table.Partitions[request].Start * uint64(table.LogicalSectorSize))
 	if start != expected {
 		t.Errorf("received start %d instead of %d", start, expected)
 	}