add tests for isUpgrade/isDowngrade

cockroachdb · Oct 7, 2024 · fee94b7 · fee94b7
1 parent 0ca97a4
commit fee94b7
Showing 1 changed file with 58 additions and 0 deletions.
diff --git a/internal/provider/cluster_resource_test.go b/internal/provider/cluster_resource_test.go
@@ -1449,3 +1449,61 @@ func TestClusterSchemaInSync(t *testing.T) {
 	dAttrs := dSchema.Schema.Attributes
 	CheckSchemaAttributesMatch(t, rAttrs, dAttrs)
 }
+
+func TestIsUpgrade(t *testing.T) {
+	_, err := isUpgrade("v22.2", "foo")
+	require.Error(t, err)
+
+	upgrade, err := isUpgrade("v24.1", "v24.2")
+	require.NoError(t, err)
+	require.True(t, upgrade)
+
+	upgrade, err = isUpgrade("v24.1", "v24.3")
+	require.NoError(t, err)
+	require.True(t, upgrade)
+
+	upgrade, err = isUpgrade("v23.2", "v24.1")
+	require.NoError(t, err)
+	require.True(t, upgrade)
+
+	upgrade, err = isUpgrade("v24.2", "v24.2")
+	require.NoError(t, err)
+	require.False(t, upgrade)
+
+	upgrade, err = isUpgrade("v24.2", "v24.1")
+	require.NoError(t, err)
+	require.False(t, upgrade)
+
+	upgrade, err = isUpgrade("v24.2", "v23.2")
+	require.NoError(t, err)
+	require.False(t, upgrade)
+}
+
+func TestIsDowngrade(t *testing.T) {
+	_, err := isDowngrade("v22.2", "foo")
+	require.Error(t, err)
+
+	upgrade, err := isDowngrade("v24.2", "v24.1")
+	require.NoError(t, err)
+	require.True(t, upgrade)
+
+	upgrade, err = isDowngrade("v24.2", "v23.2")
+	require.NoError(t, err)
+	require.True(t, upgrade)
+
+	upgrade, err = isDowngrade("v24.2", "v24.2")
+	require.NoError(t, err)
+	require.False(t, upgrade)
+
+	upgrade, err = isDowngrade("v24.1", "v24.2")
+	require.NoError(t, err)
+	require.False(t, upgrade)
+
+	upgrade, err = isDowngrade("v24.1", "v24.3")
+	require.NoError(t, err)
+	require.False(t, upgrade)
+
+	upgrade, err = isDowngrade("v23.2", "v24.1")
+	require.NoError(t, err)
+	require.False(t, upgrade)
+}