orion/emlid/updater/reboot_now_test.go

package updater

import (
	"context"
	"testing"
	"time"

	"forge.cadoles.com/Pyxis/orion/emlid"
)

func TestClientRebootNow(t *testing.T) {

	if !*runUpdaterIntegrationTests {
		t.Skip("To run this test, use: go test -updater-integration")
	}

	if !*runRebootTest {
		t.Skip("To run this test, use: go test -updater-reboot-test")
	}

	client := NewClient(
		emlid.WithStandardLogger(),
		emlid.WithEndpoint(*reachHost, 80),
	)
	if err := client.Connect(); err != nil {
		t.Fatal(err)
	}

	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
	defer cancel()
	if err := client.RebootNow(ctx, true); err != nil {
		t.Error(err)
	}

	defer client.Close()

}