particle-os/debian-forge-composer
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

tests/image: allow to specify host address

Browse source 
Soon, images will be run non-locally (AWS, Azure). This commit adds support
to test ssh for different machines than those on localhost.
This commit is contained in:
Ondřej Budai 2020-03-27 13:00:53 +01:00 • committed by Tom Gundersen
parent 7baacda6ef
commit cd51cfdcdf
1 changed files with 7 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

14
cmd/osbuild-image-tests/main_test.go
View file

@ -125,7 +125,7 @@ func (*timeoutError) Error() string { return "" }
// that 10 seconds or if systemd-is-running returns starting. // that 10 seconds or if systemd-is-running returns starting.
// It returns nil if systemd-is-running returns running or degraded. // It returns nil if systemd-is-running returns running or degraded.
// It can also return other errors in other error cases. // It can also return other errors in other error cases.
func trySSHOnce(ns *netNS) error { func trySSHOnce(address string, ns *netNS) error {
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second) ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
defer cancel() defer cancel()
@ -135,7 +135,7 @@ func trySSHOnce(ns *netNS) error {
"-i", privateKeyPath, "-i", privateKeyPath,
"-o", "StrictHostKeyChecking=no", "-o", "StrictHostKeyChecking=no",
"-o", "UserKnownHostsFile=/dev/null", "-o", "UserKnownHostsFile=/dev/null",
"redhat@localhost", "redhat@" + address,
"systemctl --wait is-system-running", "systemctl --wait is-system-running",
} }
@ -180,10 +180,10 @@ func trySSHOnce(ns *netNS) error {
// testSSH tests the running image using ssh. // testSSH tests the running image using ssh.
// It tries 20 attempts before giving up. If a major error occurs, it might // It tries 20 attempts before giving up. If a major error occurs, it might
// return earlier. // return earlier.
func testSSH(t *testing.T, ns *netNS) { func testSSH(t *testing.T, address string, ns *netNS) {
const attempts = 20 const attempts = 20
for i := 0; i < attempts; i++ { for i := 0; i < attempts; i++ {
err := trySSHOnce(ns) err := trySSHOnce(address, ns)
if err == nil { if err == nil {
// pass the test // pass the test
return return
@ -212,18 +212,18 @@ func testBoot(t *testing.T, imagePath string, bootType string, outputID string)
fallthrough fallthrough
case "qemu-extract": case "qemu-extract":
return withBootedQemuImage(imagePath, ns, func() error { return withBootedQemuImage(imagePath, ns, func() error {
testSSH(t, &ns) testSSH(t, "localhost", &ns)
return nil return nil
}) })
case "nspawn": case "nspawn":
return withBootedNspawnImage(imagePath, outputID, ns, func() error { return withBootedNspawnImage(imagePath, outputID, ns, func() error {
testSSH(t, &ns) testSSH(t, "localhost", &ns)
return nil return nil
}) })
case "nspawn-extract": case "nspawn-extract":
return withExtractedTarArchive(imagePath, func(dir string) error { return withExtractedTarArchive(imagePath, func(dir string) error {
return withBootedNspawnDirectory(dir, outputID, ns, func() error { return withBootedNspawnDirectory(dir, outputID, ns, func() error {
testSSH(t, &ns) testSSH(t, "localhost", &ns)
return nil return nil
}) })
}) })