Fix lib upgrade command not installing new libraries dependencies

Author: silvanocerza

Diff for: commands/lib/upgrade.go

@@ -17,11 +17,8 @@ package lib
 
 import (
 	"context"
-	"errors"
 
 	"github.com/arduino/arduino-cli/arduino"
-	"github.com/arduino/arduino-cli/arduino/libraries"
-	"github.com/arduino/arduino-cli/arduino/libraries/librariesmanager"
 	"github.com/arduino/arduino-cli/commands"
 	rpc "github.com/arduino/arduino-cli/rpc/cc/arduino/cli/commands/v1"
 )
@@ -33,7 +30,7 @@ func LibraryUpgradeAll(req *rpc.LibraryUpgradeAllRequest, downloadCB rpc.Downloa
 		return &arduino.InvalidInstanceError{}
 	}
 
-	if err := upgrade(lm, listLibraries(lm, true, false), downloadCB, taskCB); err != nil {
+	if err := upgrade(req.Instance, listLibraries(lm, true, false), downloadCB, taskCB); err != nil {
 		return err
 	}
 
@@ -47,6 +44,9 @@ func LibraryUpgradeAll(req *rpc.LibraryUpgradeAllRequest, downloadCB rpc.Downloa
 // LibraryUpgrade upgrades a library
 func LibraryUpgrade(ctx context.Context, req *rpc.LibraryUpgradeRequest, downloadCB rpc.DownloadProgressCB, taskCB rpc.TaskProgressCB) error {
 	lm := commands.GetLibraryManager(req)
+	if lm == nil {
+		return &arduino.InvalidInstanceError{}
+	}
 
 	// Get the library to upgrade
 	name := req.GetName()
@@ -61,23 +61,21 @@ func LibraryUpgrade(ctx context.Context, req *rpc.LibraryUpgradeRequest, downloa
 	}
 
 	// Install update
-	return upgrade(lm, []*installedLib{lib}, downloadCB, taskCB)
+	return upgrade(req.Instance, []*installedLib{lib}, downloadCB, taskCB)
 }
 
-func upgrade(lm *librariesmanager.LibrariesManager, libs []*installedLib, downloadCB rpc.DownloadProgressCB, taskCB rpc.TaskProgressCB) error {
-	// Go through the list and download them
+func upgrade(instance *rpc.Instance, libs []*installedLib, downloadCB rpc.DownloadProgressCB, taskCB rpc.TaskProgressCB) error {
 	for _, lib := range libs {
-		if err := downloadLibrary(lm, lib.Available, downloadCB, taskCB); err != nil {
-			return err
+		libInstallReq := &rpc.LibraryInstallRequest{
+			Instance:    instance,
+			Name:        lib.Library.Name,
+			Version:     "",
+			NoDeps:      false,
+			NoOverwrite: false,
 		}
-	}
-
-	// Go through the list and install them
-	for _, lib := range libs {
-		if err := installLibrary(lm, lib.Available, libraries.User, taskCB); err != nil {
-			if !errors.Is(err, librariesmanager.ErrAlreadyInstalled) {
-				return err
-			}
+		err := LibraryInstall(context.Background(), libInstallReq, downloadCB, taskCB)
+		if err != nil {
+			return err
 		}
 	}
 

Diff for: internal/integrationtest/lib/lib_test.go

@@ -199,3 +199,38 @@ func TestLibDepsOutput(t *testing.T) {
 	require.Equal(t, "WiFiNINA", jsonDeps.Dependencies[6].Name)
 	require.Equal(t, jsonDeps.Dependencies[6].VersionInstalled, jsonDeps.Dependencies[6].VersionRequired)
 }
+
+func TestUpgradeLibraryWithDependencies(t *testing.T) {
+	env, cli := integrationtest.CreateArduinoCLIWithEnvironment(t)
+	defer env.CleanUp()
+
+	// Updates index for cores and libraries
+	_, _, err := cli.Run("core", "update-index")
+	require.NoError(t, err)
+	_, _, err = cli.Run("lib", "update-index")
+	require.NoError(t, err)
+
+	// Install library
+	_, _, err = cli.Run("lib", "install", "[email protected]")
+	require.NoError(t, err)
+	stdOut, _, err := cli.Run("lib", "deps", "[email protected]", "--format", "json")
+	require.NoError(t, err)
+	expectedOutput := `{"dependencies":[
+	{"name":"Arduino_ConnectionHandler","version_required":"0.3.3","version_installed":"0.3.3"},
+	{"name":"Arduino_DebugUtils","version_required":"1.3.0","version_installed":"1.3.0"},
+	{"name":"MKRGSM","version_required":"1.5.0","version_installed":"1.5.0"},
+	{"name":"MKRNB","version_required":"1.5.1","version_installed":"1.5.1"},
+	{"name":"WiFi101","version_required":"0.16.1","version_installed":"0.16.1"},
+	{"name":"WiFiNINA","version_required":"1.8.13","version_installed":"1.8.13"}]}`
+	require.JSONEq(t, expectedOutput, string(stdOut))
+
+	// Test lib upgrade also install new dependencies of already installed library
+	_, _, err = cli.Run("lib", "upgrade", "Arduino_ConnectionHandler")
+	require.NoError(t, err)
+	stdOut, _, err = cli.Run("lib", "deps", "Arduino_ConnectionHandler", "--format", "json")
+	require.NoError(t, err)
+
+	jsonOut := requirejson.Parse(t, stdOut)
+	dependency := jsonOut.Query(`.dependencies[] | select(.name=="MKRWAN")`)
+	require.Equal(t, dependency.Query(".version_required"), dependency.Query(".version_installed"))
+}