safing-portbase/updater/resource_test.go

package updater

import (
	"testing"
)

func TestVersionSelection(t *testing.T) {
	res := registry.newResource("test/a")

	err := res.AddVersion("1.2.3", true, true, false)
	if err != nil {
		t.Fatal(err)
	}
	err = res.AddVersion("1.2.4b", true, false, true)
	if err != nil {
		t.Fatal(err)
	}
	err = res.AddVersion("1.2.2", true, false, false)
	if err != nil {
		t.Fatal(err)
	}
	err = res.AddVersion("1.2.5", false, true, false)
	if err != nil {
		t.Fatal(err)
	}
	err = res.AddVersion("0", true, false, false)
	if err != nil {
		t.Fatal(err)
	}

	registry.Online = true
	registry.Beta = true
	registry.DevMode = true
	res.selectVersion()
	if res.SelectedVersion.VersionNumber != "0" {
		t.Errorf("selected version should be 0, not %s", res.SelectedVersion.VersionNumber)
	}

	registry.DevMode = false
	res.selectVersion()
	if res.SelectedVersion.VersionNumber != "1.2.4b" {
		t.Errorf("selected version should be 1.2.4b, not %s", res.SelectedVersion.VersionNumber)
	}

	registry.Beta = false
	res.selectVersion()
	if res.SelectedVersion.VersionNumber != "1.2.5" {
		t.Errorf("selected version should be 1.2.5, not %s", res.SelectedVersion.VersionNumber)
	}

	registry.Online = false
	res.selectVersion()
	if res.SelectedVersion.VersionNumber != "1.2.3" {
		t.Errorf("selected version should be 1.2.3, not %s", res.SelectedVersion.VersionNumber)
	}

	f123 := res.GetFile()
	f123.markActiveWithLocking()

	err = res.Blacklist("1.2.3")
	if err != nil {
		t.Fatal(err)
	}
	if res.SelectedVersion.VersionNumber != "1.2.2" {
		t.Errorf("selected version should be 1.2.2, not %s", res.SelectedVersion.VersionNumber)
	}

	if !f123.UpgradeAvailable() {
		t.Error("upgrade should be available (flag)")
	}
	select {
	case <-f123.WaitForAvailableUpgrade():
	default:
		t.Error("upgrade should be available (chan)")
	}

	t.Logf("resource: %+v", res)
	for _, rv := range res.Versions {
		t.Logf("version %s: %+v", rv.VersionNumber, rv)
	}
}