|
| 1 | +//go:build contract |
| 2 | + |
| 3 | +package contract_test |
| 4 | + |
| 5 | +import ( |
| 6 | + "bytes" |
| 7 | + "encoding/json" |
| 8 | + "os" |
| 9 | + "path/filepath" |
| 10 | + "testing" |
| 11 | + |
| 12 | + "github.com/stretchr/testify/assert" |
| 13 | + "github.com/stretchr/testify/require" |
| 14 | + |
| 15 | + "github.com/openbootdotdev/openboot/internal/config" |
| 16 | + "github.com/openbootdotdev/openboot/internal/snapshot" |
| 17 | +) |
| 18 | + |
| 19 | +type canonicalPackageEntry struct { |
| 20 | + Name string `json:"name"` |
| 21 | + Desc string `json:"desc"` |
| 22 | +} |
| 23 | + |
| 24 | +type canonicalRemoteConfig struct { |
| 25 | + Username string `json:"username"` |
| 26 | + Slug string `json:"slug"` |
| 27 | + Name string `json:"name"` |
| 28 | + Preset string `json:"preset"` |
| 29 | + Packages []canonicalPackageEntry `json:"packages"` |
| 30 | + Casks []canonicalPackageEntry `json:"casks"` |
| 31 | + Taps []string `json:"taps"` |
| 32 | + Npm []canonicalPackageEntry `json:"npm"` |
| 33 | + DotfilesRepo string `json:"dotfiles_repo"` |
| 34 | + PostInstall []string `json:"post_install"` |
| 35 | + Shell *config.RemoteShellConfig `json:"shell"` |
| 36 | + MacOSPrefs []config.RemoteMacOSPref `json:"macos_prefs"` |
| 37 | +} |
| 38 | + |
| 39 | +func TestRemoteConfigFixtureIsConsumedLosslessly(t *testing.T) { |
| 40 | + data := readContractFixture(t, "config-v1.json") |
| 41 | + |
| 42 | + var wire canonicalRemoteConfig |
| 43 | + decoder := json.NewDecoder(bytes.NewReader(data)) |
| 44 | + decoder.DisallowUnknownFields() |
| 45 | + require.NoError(t, decoder.Decode(&wire), "contract fixture must use the canonical remote-config shape") |
| 46 | + require.NotEmpty(t, wire.Packages, "fixture must exercise formulae") |
| 47 | + require.NotEmpty(t, wire.Casks, "fixture must exercise casks") |
| 48 | + require.NotEmpty(t, wire.Taps, "fixture must exercise taps") |
| 49 | + require.NotEmpty(t, wire.Npm, "fixture must exercise npm packages") |
| 50 | + |
| 51 | + got, err := config.UnmarshalRemoteConfigFlexible(data) |
| 52 | + require.NoError(t, err) |
| 53 | + require.NoError(t, got.Validate()) |
| 54 | + |
| 55 | + want := &config.RemoteConfig{ |
| 56 | + Username: wire.Username, |
| 57 | + Slug: wire.Slug, |
| 58 | + Name: wire.Name, |
| 59 | + Preset: wire.Preset, |
| 60 | + Packages: packageEntries(wire.Packages), |
| 61 | + Casks: packageEntries(wire.Casks), |
| 62 | + Taps: wire.Taps, |
| 63 | + Npm: packageEntries(wire.Npm), |
| 64 | + DotfilesRepo: wire.DotfilesRepo, |
| 65 | + PostInstall: wire.PostInstall, |
| 66 | + Shell: wire.Shell, |
| 67 | + MacOSPrefs: wire.MacOSPrefs, |
| 68 | + } |
| 69 | + assert.Equal(t, want, got, "CLI decoding must not repair, move, or drop fields from the canonical fixture") |
| 70 | +} |
| 71 | + |
| 72 | +func TestSnapshotFixtureIsConsumedLosslessly(t *testing.T) { |
| 73 | + data := readContractFixture(t, "snapshot-v1.json") |
| 74 | + |
| 75 | + var wire struct { |
| 76 | + Packages struct { |
| 77 | + Formulae []string `json:"formulae"` |
| 78 | + Casks []string `json:"casks"` |
| 79 | + Taps []string `json:"taps"` |
| 80 | + Npm []string `json:"npm"` |
| 81 | + } `json:"packages"` |
| 82 | + } |
| 83 | + require.NoError(t, json.Unmarshal(data, &wire)) |
| 84 | + require.NotEmpty(t, wire.Packages.Formulae, "fixture must exercise formulae") |
| 85 | + require.NotEmpty(t, wire.Packages.Casks, "fixture must exercise casks") |
| 86 | + require.NotEmpty(t, wire.Packages.Taps, "fixture must exercise taps") |
| 87 | + require.NotEmpty(t, wire.Packages.Npm, "fixture must exercise npm packages") |
| 88 | + |
| 89 | + got, err := snapshot.ParseBytes(data) |
| 90 | + require.NoError(t, err) |
| 91 | + assert.Equal(t, wire.Packages.Formulae, got.Packages.Formulae) |
| 92 | + assert.Equal(t, wire.Packages.Casks, got.Packages.Casks) |
| 93 | + assert.Equal(t, wire.Packages.Taps, got.Packages.Taps) |
| 94 | + assert.Equal(t, wire.Packages.Npm, got.Packages.Npm) |
| 95 | +} |
| 96 | + |
| 97 | +func readContractFixture(t *testing.T, name string) []byte { |
| 98 | + t.Helper() |
| 99 | + |
| 100 | + contractDir := os.Getenv("OPENBOOT_CONTRACT_DIR") |
| 101 | + require.NotEmpty(t, contractDir, "OPENBOOT_CONTRACT_DIR must point to an openboot-contract checkout") |
| 102 | + |
| 103 | + data, err := os.ReadFile(filepath.Join(contractDir, "fixtures", name)) |
| 104 | + require.NoError(t, err) |
| 105 | + return data |
| 106 | +} |
| 107 | + |
| 108 | +func packageEntries(entries []canonicalPackageEntry) config.PackageEntryList { |
| 109 | + result := make(config.PackageEntryList, len(entries)) |
| 110 | + for i, entry := range entries { |
| 111 | + result[i] = config.PackageEntry{Name: entry.Name, Desc: entry.Desc} |
| 112 | + } |
| 113 | + return result |
| 114 | +} |
0 commit comments