|
15 | 15 | package main |
16 | 16 |
|
17 | 17 | import ( |
| 18 | + "errors" |
18 | 19 | "os" |
19 | 20 | "path/filepath" |
20 | 21 | "strings" |
21 | 22 | "testing" |
22 | 23 |
|
23 | 24 | "github.com/stretchr/testify/assert" |
24 | 25 | "github.com/stretchr/testify/require" |
| 26 | + "go.uber.org/zap/zaptest" |
25 | 27 |
|
26 | 28 | mergestrategypb "github.com/uber/submitqueue/api/base/mergestrategy/protopb" |
27 | 29 | ) |
@@ -113,6 +115,133 @@ func TestLoadMergeConfig_EmptyFileIsNoop(t *testing.T) { |
113 | 115 | assert.False(t, cfg.usesGit()) |
114 | 116 | } |
115 | 117 |
|
| 118 | +func TestResolveMergerStartupConfig(t *testing.T) { |
| 119 | + tests := []struct { |
| 120 | + name string |
| 121 | + merger string |
| 122 | + checkoutPath string |
| 123 | + configPath string |
| 124 | + configContents string |
| 125 | + hasConfigFile bool |
| 126 | + wantSelection string |
| 127 | + wantDefaultType string |
| 128 | + wantUsesGit bool |
| 129 | + wantErr error |
| 130 | + wantUnclassified bool |
| 131 | + }{ |
| 132 | + { |
| 133 | + name: "unset without config uses automatic noop fallback", |
| 134 | + wantDefaultType: mergerTypeNoop, |
| 135 | + }, |
| 136 | + { |
| 137 | + name: "explicit git without config fails", |
| 138 | + merger: mergerTypeGit, |
| 139 | + wantErr: errExplicitGitConfigurationRequired, |
| 140 | + }, |
| 141 | + { |
| 142 | + name: "explicit git with checkout environment is configured", |
| 143 | + merger: mergerTypeGit, |
| 144 | + checkoutPath: "/var/checkouts/r", |
| 145 | + wantSelection: mergerTypeGit, |
| 146 | + wantDefaultType: mergerTypeGit, |
| 147 | + wantUsesGit: true, |
| 148 | + }, |
| 149 | + { |
| 150 | + name: "explicit git with config file containing a git target is configured", |
| 151 | + merger: mergerTypeGit, |
| 152 | + configContents: gitTargetConfig(), |
| 153 | + hasConfigFile: true, |
| 154 | + wantSelection: mergerTypeGit, |
| 155 | + wantDefaultType: mergerTypeNoop, |
| 156 | + wantUsesGit: true, |
| 157 | + }, |
| 158 | + { |
| 159 | + name: "explicit git with config file containing no git target fails", |
| 160 | + merger: mergerTypeGit, |
| 161 | + configContents: noopTargetConfig(), |
| 162 | + hasConfigFile: true, |
| 163 | + wantErr: errExplicitGitConfigurationRequired, |
| 164 | + }, |
| 165 | + { |
| 166 | + name: "explicit noop bypasses merge configuration", |
| 167 | + merger: mergerTypeNoop, |
| 168 | + configPath: "/missing/merge.yaml", |
| 169 | + wantSelection: mergerTypeNoop, |
| 170 | + }, |
| 171 | + { |
| 172 | + name: "explicit fake preserves test behavior", |
| 173 | + merger: mergerOverrideFake, |
| 174 | + configPath: "/missing/merge.yaml", |
| 175 | + wantSelection: mergerOverrideFake, |
| 176 | + }, |
| 177 | + { |
| 178 | + name: "invalid merger fails", |
| 179 | + merger: "magic", |
| 180 | + wantUnclassified: true, |
| 181 | + }, |
| 182 | + { |
| 183 | + name: "invalid merge config still fails", |
| 184 | + configContents: "defaults:\n merger: {type: magic}\n", |
| 185 | + hasConfigFile: true, |
| 186 | + wantUnclassified: true, |
| 187 | + }, |
| 188 | + } |
| 189 | + |
| 190 | + for _, tt := range tests { |
| 191 | + t.Run(tt.name, func(t *testing.T) { |
| 192 | + setMergerStartupEnv(t, tt.merger, tt.checkoutPath) |
| 193 | + switch { |
| 194 | + case tt.hasConfigFile: |
| 195 | + t.Setenv("MERGE_CONFIG_PATH", writeConfig(t, tt.configContents)) |
| 196 | + case tt.configPath != "": |
| 197 | + t.Setenv("MERGE_CONFIG_PATH", tt.configPath) |
| 198 | + } |
| 199 | + |
| 200 | + startup, err := resolveMergerStartupConfig(zaptest.NewLogger(t)) |
| 201 | + if tt.wantErr != nil { |
| 202 | + require.ErrorIs(t, err, tt.wantErr) |
| 203 | + return |
| 204 | + } |
| 205 | + if tt.wantUnclassified { |
| 206 | + require.Error(t, err) |
| 207 | + assert.False(t, errors.Is(err, errExplicitGitConfigurationRequired)) |
| 208 | + return |
| 209 | + } |
| 210 | + require.NoError(t, err) |
| 211 | + assert.Equal(t, tt.wantSelection, startup.selection) |
| 212 | + assert.Equal(t, tt.wantDefaultType, startup.targets.Defaults.Merger.Type) |
| 213 | + assert.Equal(t, tt.wantUsesGit, startup.targets.usesGit()) |
| 214 | + }) |
| 215 | + } |
| 216 | +} |
| 217 | + |
| 218 | +func setMergerStartupEnv(t *testing.T, merger, checkoutPath string) { |
| 219 | + t.Helper() |
| 220 | + t.Setenv("MERGER", merger) |
| 221 | + t.Setenv("MERGE_CONFIG_PATH", "") |
| 222 | + t.Setenv("MERGE_CHECKOUT_PATH", checkoutPath) |
| 223 | +} |
| 224 | + |
| 225 | +func gitTargetConfig() string { |
| 226 | + return ` |
| 227 | +defaults: |
| 228 | + merger: {type: noop} |
| 229 | +queues: |
| 230 | + - name: demo |
| 231 | + merger: {type: git, checkoutPath: /var/checkouts/r} |
| 232 | +` |
| 233 | +} |
| 234 | + |
| 235 | +func noopTargetConfig() string { |
| 236 | + return ` |
| 237 | +defaults: |
| 238 | + merger: {type: noop} |
| 239 | +queues: |
| 240 | + - name: demo |
| 241 | + merger: {type: noop} |
| 242 | +` |
| 243 | +} |
| 244 | + |
116 | 245 | func TestLoadMergeConfig_SharedCheckoutForSameTargetIsAllowed(t *testing.T) { |
117 | 246 | // Two queues landing on one target must resolve to one merger instance, so |
118 | 247 | // sharing a checkout is the correct configuration rather than a mistake. |
|
0 commit comments