Allow using arm64v8 binaries in darwin for postgres 14.2 and up

Author: bdoherty-tovala

version_strategy.go

@@ -1,6 +1,7 @@
 package embeddedpostgres
 
 import (
+	"fmt"
 	"os"
 	"os/exec"
 	"strings"
@@ -36,9 +37,16 @@ func defaultVersionStrategy(config Config, goos, arch string, linuxMachineName f
 			}
 		}
 
-		// at this point, postgres is not available for macos on arm
+		// postgres below version 14.2 is not available for macos on arm
 		if goos == "darwin" && arch == "arm64" {
-			arch = "amd64"
+			var majorVer, minorVer int
+			fmt.Sscanf(string(config.version), "%d.%d", &majorVer, &minorVer)
+
+			if majorVer < 14 || (majorVer == 14 && minorVer < 2) {
+				arch = "amd64"
+			} else {
+				arch += "v8"
+			}
 		}
 
 		return goos, arch, config.version

version_strategy_test.go

@@ -54,28 +54,42 @@ func Test_DefaultVersionStrategy_AllGolangDistributions(t *testing.T) {
 		"windows/arm":     {"windows", "arm"},
 	}
 
+	versionDifferences := map[PostgresVersion]map[string][]string{
+		PostgresVersion("14.0.0"): {},
+		PostgresVersion("14.1.0"): {},
+		PostgresVersion("14.2.0"): {"darwin/arm64": {"darwin", "arm64v8"}},
+		V14:                       {"darwin/arm64": {"darwin", "arm64v8"}},
+	}
 	defaultConfig := DefaultConfig()
 
-	for dist, expected := range allGolangDistributions {
-		dist := dist
-		expected := expected
-
-		t.Run(fmt.Sprintf("DefaultVersionStrategy_%s", dist), func(t *testing.T) {
-			osArch := strings.Split(dist, "/")
-
-			operatingSystem, architecture, postgresVersion := defaultVersionStrategy(
-				defaultConfig,
-				osArch[0],
-				osArch[1],
-				linuxMachineName,
-				func() bool {
-					return false
-				})()
-
-			assert.Equal(t, expected[0], operatingSystem)
-			assert.Equal(t, expected[1], architecture)
-			assert.Equal(t, V14, postgresVersion)
-		})
+	for version, differences := range versionDifferences {
+		defaultConfig.version = version
+
+		for dist, expected := range allGolangDistributions {
+			dist := dist
+			expected := expected
+
+			if override, ok := differences[dist]; ok {
+				expected = override
+			}
+
+			t.Run(fmt.Sprintf("DefaultVersionStrategy_%s", dist), func(t *testing.T) {
+				osArch := strings.Split(dist, "/")
+
+				operatingSystem, architecture, postgresVersion := defaultVersionStrategy(
+					defaultConfig,
+					osArch[0],
+					osArch[1],
+					linuxMachineName,
+					func() bool {
+						return false
+					})()
+
+				assert.Equal(t, expected[0], operatingSystem)
+				assert.Equal(t, expected[1], architecture)
+				assert.Equal(t, version, postgresVersion)
+			})
+		}
 	}
 }