test: Add tests for pypi and pypi_resolver

2026-08-03 07:24:09 +02:00 · 2025-05-17 22:14:38 +05:30
parent fc9fddb324
commit 722ee05376
4 changed files with 285 additions and 8 deletions
@@ -0,0 +1,147 @@
+package packagemanager
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestPipParsePackageInfo(t *testing.T) {
+	cases := []struct {
+		name    string
+		input   string
+		pkgName string
+		version string
+		wantErr bool
+	}{
+		{
+			name:    "simple package name",
+			input:   "fastapi",
+			pkgName: "fastapi",
+			version: "",
+			wantErr: false,
+		},
+		{
+			name:    "package with exact version",
+			input:   "fastapi==0.115.7",
+			pkgName: "fastapi",
+			version: "==0.115.7",
+			wantErr: false,
+		},
+		{
+			name:    "package with version range",
+			input:   "requests>=2.0,<3.0",
+			pkgName: "requests",
+			version: ">=2.0,<3.0",
+			wantErr: false,
+		},
+		{
+			name:    "package with exclusion",
+			input:   "pydantic!=1.8,!=1.8.1",
+			pkgName: "pydantic",
+			version: "!=1.8,!=1.8.1",
+			wantErr: false,
+		},
+		{
+			name:    "package with compatible release",
+			input:   "django~=3.1.0",
+			pkgName: "django",
+			version: "~=3.1.0",
+			wantErr: false,
+		},
+		{
+			name:    "package with greater than",
+			input:   "numpy>1.20.0",
+			pkgName: "numpy",
+			version: ">1.20.0",
+			wantErr: false,
+		},
+		{
+			name:    "package with less than",
+			input:   "pandas<2.0.0",
+			pkgName: "pandas",
+			version: "<2.0.0",
+			wantErr: false,
+		},
+		{
+			name:    "empty input",
+			input:   "",
+			pkgName: "",
+			version: "",
+			wantErr: true,
+		},
+		{
+			name:    "only version specifier",
+			input:   "==1.0.0",
+			pkgName: "",
+			version: "",
+			wantErr: true,
+		},
+		{
+			name:    "package with whitespace",
+			input:   "  requests  ==  2.0.0  ",
+			pkgName: "requests",
+			version: "==  2.0.0",
+			wantErr: false,
+		},
+	}
+
+	for _, tc := range cases {
+		t.Run(tc.name, func(t *testing.T) {
+			pkgName, version, err := pipParsePackageInfo(tc.input)
+			if tc.wantErr {
+				assert.Error(t, err)
+			} else {
+				assert.NoError(t, err)
+				assert.Equal(t, tc.pkgName, pkgName)
+				assert.Equal(t, tc.version, version)
+			}
+		})
+	}
+}
+
+func TestPipConvertCompatibleRelease(t *testing.T) {
+	cases := []struct {
+		name     string
+		input    string
+		expected string
+	}{
+		{
+			name:     "standard version",
+			input:    "~=3.1.0",
+			expected: ">=3.1.0,<3.2.0",
+		},
+		{
+			name:     "single digit minor",
+			input:    "~=2.1.5",
+			expected: ">=2.1.5,<2.2.0",
+		},
+		{
+			name:     "double digit minor",
+			input:    "~=1.10.0",
+			expected: ">=1.10.0,<1.11.0",
+		},
+		{
+			name:     "invalid format",
+			input:    "~=1",
+			expected: "",
+		},
+		{
+			name:     "missing prefix",
+			input:    "3.1.0",
+			expected: "3.1.0",
+		},
+		{
+			name:     "extra segments",
+			input:    "~=2.1.5.2",
+			expected: ">=2.1.5.2,<2.1.6",
+		},
+	}
+
+	for _, tc := range cases {
+		t.Run(tc.name, func(t *testing.T) {
+			result := pipConvertCompatibleRelease(tc.input)
+			assert.Equal(t, tc.expected, result)
+		})
+	}
+}