orion/reach/wifi_networks_test.go

package reach

import (
	"fmt"
	"math/rand"
	"testing"
	"time"
)

func TestClientSavedWiFiNetworks(t *testing.T) {

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

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

	_, err := client.WifiNetworks()
	if err != nil {
		t.Error(err)
	}

	defer client.Close()

}

func TestClientCRUDWiFiNetwork(t *testing.T) {

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

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

	ssid := fmt.Sprintf("wifi_test_%d", rand.Uint32())

	done, err := client.AddWifiNetwork(ssid, SecurityOpen, "")
	if err != nil {
		t.Error(err)
	}

	if g, e := done, true; g != e {
		t.Errorf("AddWifiNetwork() -> done: got '%v', expected '%v'", g, e)
	}

	networks, err := client.WifiNetworks()
	if err != nil {
		t.Error(err)
	}

	found := false
	for _, n := range networks {
		if n.SSID == ssid {
			found = true
			break
		}
	}

	if g, e := found, true; g != e {
		t.Errorf("wifi network '%s' should exists", ssid)
	}

	done, err = client.RemoveWifiNetwork(ssid)
	if err != nil {
		t.Error(err)
	}

	if g, e := done, true; g != e {
		t.Errorf("RemoveWifiNetwork() -> done: got '%v', expected '%v'", g, e)
	}

	defer client.Close()

}

func TestClientWifiNetworkConnect(t *testing.T) {

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

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

	ssid := fmt.Sprintf("wifi_test_%d", rand.Uint32())

	done, err := client.AddWifiNetwork(ssid, SecurityOpen, "")
	if err != nil {
		t.Error(err)
	}

	if g, e := done, true; g != e {
		t.Errorf("AddWifiNetwork() -> done: got '%v', expected '%v'", g, e)
	}

	if err := client.ConnectToWifiNetwork(ssid, true); err != nil {
		t.Error(err)
	}

	time.Sleep(5 * time.Second)

	defer client.Close()

}