blob: 93f2cefa54f89bdd7ca5c698a8c2104d280beffe [file] [log] [blame] [edit]
// Copyright 2020 The Fuchsia Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
package main
import (
"bytes"
"testing"
)
func assertArrayEq(t *testing.T, result []byte, expected []byte) {
if len(result) != len(expected) {
t.Fatalf("test output is incorrect length!\nExpected: %d\nGot: %d\n", len(expected), len(result))
}
ok := true
for i := range result {
if result[i] != expected[i] {
t.Errorf("expected and result differ at index %d. Expected %d, got %d\n", i, expected[i], result[i])
ok = false
}
}
if !ok {
t.FailNow()
}
}
func TestWriteAbrA(t *testing.T) {
var buffer bytes.Buffer
WriteAbr(BOOT_A, &buffer)
expected := []byte{
// magic
0, 'A', 'B', '0',
// version_major, version_minor, reserved x2
2, 1, 0, 0,
// slot 1 data
// priority, tries_remaining, successful_boot, reserved
abrMaxPriority, abrMaxTriesRemaining, 0, 0,
// slot 2 data
// priority, tries_remaining, successful_boot, reserved
1, abrMaxTriesRemaining, 0, 0,
// one_shot_recovery_boot, reserved x11
0, 0, 0, 0,
0, 0, 0, 0,
0, 0, 0, 0,
// crc32 of the previous 28 bytes in big endian.
52, 118, 193, 121,
}
var result [32]byte
n, err := buffer.Read(result[:])
if err != nil {
t.Fatal(err)
}
assertArrayEq(t, result[0:n], expected)
}
func TestWriteAbrB(t *testing.T) {
var buffer bytes.Buffer
WriteAbr(BOOT_B, &buffer)
expected := []byte{
// magic
0, 'A', 'B', '0',
// version_major, version_minor, reserved x2
2, 1, 0, 0,
// slot 1 data
// priority, tries_remaining, successful_boot, reserved
1, abrMaxTriesRemaining, 0, 0,
// slot 2 data
// priority, tries_remaining, successful_boot, reserved
abrMaxPriority, abrMaxTriesRemaining, 0, 0,
// one_shot_recovery_boot, reserved x11
0, 0, 0, 0,
0, 0, 0, 0,
0, 0, 0, 0,
// crc32 of the previous 28 bytes in big endian.
13, 12, 124, 181,
}
var result [32]byte
n, err := buffer.Read(result[:])
if err != nil {
t.Fatal(err)
}
assertArrayEq(t, result[0:n], expected)
}
func TestWriteAbrRecovery(t *testing.T) {
var buffer bytes.Buffer
WriteAbr(BOOT_RECOVERY, &buffer)
expected := []byte{
// magic
0, 'A', 'B', '0',
// version_major, version_minor, reserved x2
2, 1, 0, 0,
// slot 1 data
// priority, tries_remaining, successful_boot, reserved
0, abrMaxTriesRemaining, 0, 0,
// slot 2 data
// priority, tries_remaining, successful_boot, reserved
0, abrMaxTriesRemaining, 0, 0,
// one_shot_recovery_boot, reserved x11
0, 0, 0, 0,
0, 0, 0, 0,
0, 0, 0, 0,
// crc32 of the previous 28 bytes in big endian.
230, 129, 32, 201,
}
var result [32]byte
n, err := buffer.Read(result[:])
if err != nil {
t.Fatal(err)
}
assertArrayEq(t, result[0:n], expected)
}