forgejo/routers/api/packages/pypi/pypi_test.go

// Copyright 2022 The Gitea Authors. All rights reserved.
// Use of this source code is governed by a MIT-style
// license that can be found in the LICENSE file.

package pypi

import (
	"testing"

	"github.com/stretchr/testify/assert"
)

func TestIsValidNameAndVersion(t *testing.T) {
	// The test cases below were created from the following Python PEPs:
	// https://peps.python.org/pep-0426/#name
	// https://peps.python.org/pep-0440/#appendix-b-parsing-version-strings-with-regular-expressions

	// Valid Cases
	assert.True(t, isValidNameAndVersion("A", "1.0.1"))
	assert.True(t, isValidNameAndVersion("Test.Name.1234", "1.0.1"))
	assert.True(t, isValidNameAndVersion("test_name", "1.0.1"))
	assert.True(t, isValidNameAndVersion("test-name", "1.0.1"))
	assert.True(t, isValidNameAndVersion("test-name", "v1.0.1"))
	assert.True(t, isValidNameAndVersion("test-name", "2012.4"))
	assert.True(t, isValidNameAndVersion("test-name", "1.0.1-alpha"))
	assert.True(t, isValidNameAndVersion("test-name", "1.0.1a1"))
	assert.True(t, isValidNameAndVersion("test-name", "1.0b2.r345.dev456"))
	assert.True(t, isValidNameAndVersion("test-name", "1!1.0.1"))
	assert.True(t, isValidNameAndVersion("test-name", "1.0.1+local.1"))

	// Invalid Cases
	assert.False(t, isValidNameAndVersion(".test-name", "1.0.1"))
	assert.False(t, isValidNameAndVersion("test!name", "1.0.1"))
	assert.False(t, isValidNameAndVersion("-test-name", "1.0.1"))
	assert.False(t, isValidNameAndVersion("test-name-", "1.0.1"))
	assert.False(t, isValidNameAndVersion("test-name", "a1.0.1"))
	assert.False(t, isValidNameAndVersion("test-name", "1.0.1aa"))
	assert.False(t, isValidNameAndVersion("test-name", "1.0.0-alpha.beta"))
}