38 lines
674 B
Go
38 lines
674 B
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()
|
|
|
|
}
|