update_test.go

 1package update
 2
 3import (
 4	"context"
 5	"testing"
 6
 7	"github.com/stretchr/testify/require"
 8)
 9
10func TestCheckForUpdate_DevelopmentVersion(t *testing.T) {
11	info, err := Check(t.Context(), "unknown", testClient{"v0.11.0"})
12	require.NoError(t, err)
13	require.NotNil(t, info)
14	require.False(t, info.Available())
15}
16
17func TestCheckForUpdate_Old(t *testing.T) {
18	info, err := Check(t.Context(), "v0.10.0", testClient{"v0.11.0"})
19	require.NoError(t, err)
20	require.NotNil(t, info)
21	require.True(t, info.Available())
22}
23
24func TestCheckForUpdate_Beta(t *testing.T) {
25	t.Run("current is stable", func(t *testing.T) {
26		info, err := Check(t.Context(), "v0.10.0", testClient{"v0.11.0-beta.1"})
27		require.NoError(t, err)
28		require.NotNil(t, info)
29		require.False(t, info.Available())
30	})
31	t.Run("current is also beta", func(t *testing.T) {
32		info, err := Check(t.Context(), "v0.11.0-beta.1", testClient{"v0.11.0-beta.2"})
33		require.NoError(t, err)
34		require.NotNil(t, info)
35		require.True(t, info.Available())
36	})
37	t.Run("current is beta, latest isn't", func(t *testing.T) {
38		info, err := Check(t.Context(), "v0.11.0-beta.1", testClient{"v0.11.0"})
39		require.NoError(t, err)
40		require.NotNil(t, info)
41		require.True(t, info.Available())
42	})
43}
44
45type testClient struct{ tag string }
46
47// Latest implements Client.
48func (t testClient) Latest(ctx context.Context) (*Release, error) {
49	return &Release{
50		TagName: t.tag,
51		HTMLURL: "https://example.org",
52	}, nil
53}