Unit tests

Author: ofalvai

toolprovider/mise/install_test.go

@@ -5,6 +5,7 @@ import (
 	"fmt"
 	"testing"
 
+	"github.com/bitrise-io/bitrise/v2/toolprovider/mise/nixpkgs"
 	"github.com/bitrise-io/bitrise/v2/toolprovider/provider"
 	"github.com/stretchr/testify/require"
 )
@@ -86,7 +87,6 @@ func TestMiseVersionString(t *testing.T) {
 			want:    "nixpkgs:[email protected]",
 			wantErr: false,
 		},
-		// TODO: more test cases?
 	}
 
 	for _, tt := range tests {
@@ -181,3 +181,136 @@ func TestIsAlreadyInstalled(t *testing.T) {
 		})
 	}
 }
+
+func TestCanBeInstalledWithNix(t *testing.T) {
+	tests := []struct {
+		name      string
+		toolID    provider.ToolID
+		version   string
+		resolutionStrategy provider.ResolutionStrategy
+		setupFake func(m *fakeExecEnv)
+		want      bool
+	}{
+		{
+			name: "concrete Ruby version that exists in index",
+			toolID: provider.ToolID("ruby"),
+			version: "3.3.9",
+			resolutionStrategy: provider.ResolutionStrategyStrict,
+			setupFake: func(m *fakeExecEnv) {
+				m.setResponse(fmt.Sprintf("plugin install %s %s", nixpkgs.PluginName, nixpkgs.PluginGitURL), "")
+				m.setResponse("ls --installed --json --quiet ruby", "[]")
+				m.setResponse("ls-remote --quiet nixpkgs:[email protected]", "3.3.9")
+			},
+			want:    true,
+		},
+		{
+			name: "fuzzy Ruby version that matches an existing version in index",
+			toolID: provider.ToolID("ruby"),
+			version: "3.3",
+			resolutionStrategy: provider.ResolutionStrategyLatestReleased,
+			setupFake: func(m *fakeExecEnv) {
+				m.setResponse(fmt.Sprintf("plugin install %s %s", nixpkgs.PluginName, nixpkgs.PluginGitURL), "")
+				m.setResponse("ls --installed --json --quiet ruby", "[]")
+				m.setResponse("ls-remote --quiet nixpkgs:[email protected]", "3.3.8\n3.3.9")
+			},
+			want:    true,
+		},
+		{
+			name: "concrete Ruby version that doesn't exist in index",
+			toolID: provider.ToolID("ruby"),
+			version: "0.0.1",
+			resolutionStrategy: provider.ResolutionStrategyStrict,
+			setupFake: func(m *fakeExecEnv) {
+				m.setResponse(fmt.Sprintf("plugin install %s %s", nixpkgs.PluginName, nixpkgs.PluginGitURL), "")
+				m.setResponse("ls --installed --json --quiet ruby", "[]")
+				m.setResponse("ls-remote --quiet nixpkgs:[email protected]", "")
+			},
+			want:    false,
+		},
+		{
+			name: "nixpkgs plugin install error",
+			toolID: provider.ToolID("ruby"),
+			version: "3.3.9",
+			resolutionStrategy: provider.ResolutionStrategyStrict,
+			setupFake: func(m *fakeExecEnv) {
+				m.setError(fmt.Sprintf("plugin install %s %s", nixpkgs.PluginName, nixpkgs.PluginGitURL), fmt.Errorf("fake error"))
+			},
+			want:    false,
+		},
+		{
+			name: "nixpkgs index check error",
+			toolID: provider.ToolID("ruby"),
+			version: "3.3.9",
+			resolutionStrategy: provider.ResolutionStrategyStrict,
+			setupFake: func(m *fakeExecEnv) {
+				m.setResponse(fmt.Sprintf("plugin install %s %s", nixpkgs.PluginName, nixpkgs.PluginGitURL), "")
+				m.setResponse("ls --installed --json --quiet ruby", "[]")
+				m.setError("ls-remote --quiet nixpkgs:[email protected]", fmt.Errorf("fake error"))
+			},
+			want:    false,
+		},
+	}
+
+	t.Setenv("BITRISE_TOOLSETUP_FAST_INSTALL", "1")
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			execEnv := newFakeExecEnv()
+			tt.setupFake(execEnv)
+
+			request := provider.ToolRequest{
+				ToolName:        tt.toolID,
+				UnparsedVersion: tt.version,
+				ResolutionStrategy: tt.resolutionStrategy,
+			}
+
+			got := canBeInstalledWithNix(request, execEnv)
+			require.Equal(t, tt.want, got)
+
+		})
+	}
+}
+
+func TestInstallRequest(t *testing.T) {
+	tests := []struct {
+		name   string
+		tool   provider.ToolRequest
+		useNix bool
+		want   provider.ToolRequest
+	}{
+		{
+			name: "without nixpkgs",
+			tool: provider.ToolRequest{
+				ToolName:        "node",
+				UnparsedVersion: "18.20.0",
+				ResolutionStrategy: provider.ResolutionStrategyStrict,
+			},
+			useNix: false,
+			want: provider.ToolRequest{
+				ToolName:        "node",
+				UnparsedVersion: "18.20.0",
+				ResolutionStrategy: provider.ResolutionStrategyStrict,
+			},
+		},
+		{
+			name: "with nixpkgs",
+			tool: provider.ToolRequest{
+				ToolName:        "node",
+				UnparsedVersion: "18",
+				ResolutionStrategy: provider.ResolutionStrategyLatestInstalled,
+			},
+			useNix: true,
+			want: provider.ToolRequest{
+				ToolName:        "nixpkgs:node",
+				UnparsedVersion: "18",
+				ResolutionStrategy: provider.ResolutionStrategyLatestInstalled,
+			},
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			got := installRequest(tt.tool, tt.useNix)
+			require.Equal(t, tt.want, got)
+		})
+	}
+}