From d9029face32cd23f8f7c93fd64acbb5933082e69 Mon Sep 17 00:00:00 2001
From: Joey Holtzman <89617856+jdholtz@users.noreply.github.com>
Date: Thu, 6 Jul 2023 01:20:51 -0500
Subject: [PATCH] Don't upgrade repo packages when --aur is specified (#2234)

fix(aur_install): Don't upgrade repo packages when --aur is specified
---
 aur_install.go      |  6 ++++
 aur_install_test.go | 88 +++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 94 insertions(+)

diff --git a/aur_install.go b/aur_install.go
index d9a46275..fffd9b34 100644
--- a/aur_install.go
+++ b/aur_install.go
@@ -420,6 +420,12 @@ func (installer *Installer) installSyncPackages(ctx context.Context, cmdArgs *pa
 	arguments.Op = "S"
 	arguments.ClearTargets()
 	arguments.AddTarget(repoTargets...)
+
+	// Don't upgrade all repo packages if only AUR upgrades are specified
+	if installer.targetMode == parser.ModeAUR {
+		arguments.DelArg("u", "upgrades")
+	}
+
 	if len(excluded) > 0 {
 		arguments.CreateOrAppendOption("ignore", excluded...)
 	}
diff --git a/aur_install_test.go b/aur_install_test.go
index 94e57396..ecfff9bc 100644
--- a/aur_install_test.go
+++ b/aur_install_test.go
@@ -1248,3 +1248,91 @@ func TestInstaller_InstallRebuild(t *testing.T) {
 		})
 	}
 }
+
+func TestInstaller_InstallUpgrade(t *testing.T) {
+	t.Parallel()
+
+	makepkgBin := t.TempDir() + "/makepkg"
+	pacmanBin := t.TempDir() + "/pacman"
+	f, err := os.OpenFile(makepkgBin, os.O_RDONLY|os.O_CREATE, 0o755)
+	require.NoError(t, err)
+	require.NoError(t, f.Close())
+
+	f, err = os.OpenFile(pacmanBin, os.O_RDONLY|os.O_CREATE, 0o755)
+	require.NoError(t, err)
+	require.NoError(t, f.Close())
+
+	type testCase struct {
+		desc       string
+		targetMode parser.TargetMode
+	}
+
+	tmpDir := t.TempDir()
+
+	testCases := []testCase{
+		{
+			desc:       "target any",
+			targetMode: parser.ModeAny,
+		},
+		{
+			desc:       "target repo",
+			targetMode: parser.ModeRepo,
+		},
+		{
+			desc:       "target aur",
+			targetMode: parser.ModeAUR,
+		},
+	}
+
+	for _, tc := range testCases {
+		tc := tc
+		t.Run(tc.desc, func(td *testing.T) {
+			mockDB := &mock.DBExecutor{}
+			mockRunner := &exe.MockRunner{}
+			cmdBuilder := &exe.CmdBuilder{
+				MakepkgBin:      makepkgBin,
+				SudoBin:         "su",
+				PacmanBin:       pacmanBin,
+				Runner:          mockRunner,
+				SudoLoopEnabled: false,
+			}
+
+			installer := NewInstaller(mockDB, cmdBuilder, &vcs.Mock{}, tc.targetMode,
+				parser.RebuildModeNo, false, NewTestLogger())
+
+			cmdArgs := parser.MakeArguments()
+			cmdArgs.AddArg("u", "upgrades") // Make sure both args are removed
+
+			targets := []map[string]*dep.InstallInfo{
+				{
+					"linux": {
+						Source:     dep.Sync,
+						Reason:     dep.Dep,
+						Version:    "17.0.0-1",
+						SyncDBName: ptrString("core"),
+					},
+				},
+			}
+
+			errI := installer.Install(context.Background(), cmdArgs, targets, map[string]string{}, []string{}, false)
+			require.NoError(td, errI)
+
+			require.NotEmpty(td, mockRunner.ShowCalls)
+
+			// The first call is the only call being test
+			call := mockRunner.ShowCalls[0]
+			show := call.Args[0].(*exec.Cmd).String()
+			show = strings.ReplaceAll(show, tmpDir, "/testdir") // replace the temp dir with a static path
+			show = strings.ReplaceAll(show, makepkgBin, "makepkg")
+			show = strings.ReplaceAll(show, pacmanBin, "pacman")
+
+			if tc.targetMode == parser.ModeAUR {
+				assert.NotContains(td, show, "--upgrades")
+				assert.NotContains(td, show, "-u")
+			} else {
+				assert.Contains(td, show, "--upgrades")
+				assert.Contains(td, show, "-u")
+			}
+		})
+	}
+}