mirror of
				https://github.com/go-gitea/gitea.git
				synced 2025-11-03 21:16:26 +01:00 
			
		
		
		
	enable mirror, usestdlibbars and perfsprint part of: https://github.com/go-gitea/gitea/issues/34083 --------- Co-authored-by: wxiaoguang <wxiaoguang@gmail.com>
		
			
				
	
	
		
			338 lines
		
	
	
		
			9.8 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			338 lines
		
	
	
		
			9.8 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
// Copyright 2021 The Gitea Authors. All rights reserved.
 | 
						|
// SPDX-License-Identifier: MIT
 | 
						|
 | 
						|
package integration
 | 
						|
 | 
						|
import (
 | 
						|
	"archive/tar"
 | 
						|
	"bytes"
 | 
						|
	"compress/gzip"
 | 
						|
	"crypto/sha256"
 | 
						|
	"crypto/sha512"
 | 
						|
	"encoding/base64"
 | 
						|
	"fmt"
 | 
						|
	"mime/multipart"
 | 
						|
	"net/http"
 | 
						|
	"testing"
 | 
						|
 | 
						|
	"code.gitea.io/gitea/models/db"
 | 
						|
	"code.gitea.io/gitea/models/packages"
 | 
						|
	"code.gitea.io/gitea/models/unittest"
 | 
						|
	user_model "code.gitea.io/gitea/models/user"
 | 
						|
	"code.gitea.io/gitea/modules/packages/rubygems"
 | 
						|
	"code.gitea.io/gitea/tests"
 | 
						|
 | 
						|
	"github.com/stretchr/testify/assert"
 | 
						|
)
 | 
						|
 | 
						|
type tarFile struct {
 | 
						|
	Name string
 | 
						|
	Data []byte
 | 
						|
}
 | 
						|
 | 
						|
func makeArchiveFileTar(files []*tarFile) []byte {
 | 
						|
	buf := new(bytes.Buffer)
 | 
						|
	tarWriter := tar.NewWriter(buf)
 | 
						|
	for _, file := range files {
 | 
						|
		_ = tarWriter.WriteHeader(&tar.Header{
 | 
						|
			Typeflag: tar.TypeReg,
 | 
						|
			Name:     file.Name,
 | 
						|
			Mode:     0o644,
 | 
						|
			Size:     int64(len(file.Data)),
 | 
						|
		})
 | 
						|
		_, _ = tarWriter.Write(file.Data)
 | 
						|
	}
 | 
						|
	_ = tarWriter.Close()
 | 
						|
	return buf.Bytes()
 | 
						|
}
 | 
						|
 | 
						|
func makeArchiveFileGz(data []byte) []byte {
 | 
						|
	buf := new(bytes.Buffer)
 | 
						|
	gzWriter, _ := gzip.NewWriterLevel(buf, gzip.NoCompression)
 | 
						|
	_, _ = gzWriter.Write(data)
 | 
						|
	_ = gzWriter.Close()
 | 
						|
	return buf.Bytes()
 | 
						|
}
 | 
						|
 | 
						|
func makeRubyGem(name, version string) []byte {
 | 
						|
	metadataContent := fmt.Sprintf(`--- !ruby/object:Gem::Specification
 | 
						|
name: %s
 | 
						|
version: !ruby/object:Gem::Version
 | 
						|
  version: %s
 | 
						|
platform: ruby
 | 
						|
authors:
 | 
						|
- Gitea
 | 
						|
autorequire:
 | 
						|
bindir: bin
 | 
						|
cert_chain: []
 | 
						|
date: 2021-08-23 00:00:00.000000000 Z
 | 
						|
dependencies:
 | 
						|
- !ruby/object:Gem::Dependency
 | 
						|
  name: runtime-dep
 | 
						|
  requirement: !ruby/object:Gem::Requirement
 | 
						|
    requirements:
 | 
						|
    - - ">="
 | 
						|
      - !ruby/object:Gem::Version
 | 
						|
        version: 1.2.0
 | 
						|
    - - "<"
 | 
						|
      - !ruby/object:Gem::Version
 | 
						|
        version: '2.0'
 | 
						|
  type: :runtime
 | 
						|
  prerelease: false
 | 
						|
  version_requirements: !ruby/object:Gem::Requirement
 | 
						|
    requirements:
 | 
						|
    - - ">="
 | 
						|
      - !ruby/object:Gem::Version
 | 
						|
        version: 1.2.0
 | 
						|
    - - "<"
 | 
						|
      - !ruby/object:Gem::Version
 | 
						|
        version: '2.0'
 | 
						|
- !ruby/object:Gem::Dependency
 | 
						|
  name: dev-dep
 | 
						|
  requirement: !ruby/object:Gem::Requirement
 | 
						|
    requirements:
 | 
						|
    - - "~>"
 | 
						|
      - !ruby/object:Gem::Version
 | 
						|
        version: '5.2'
 | 
						|
  type: :development
 | 
						|
  prerelease: false
 | 
						|
  version_requirements: !ruby/object:Gem::Requirement
 | 
						|
    requirements:
 | 
						|
    - - "~>"
 | 
						|
      - !ruby/object:Gem::Version
 | 
						|
        version: '5.2'
 | 
						|
description: RubyGems package test
 | 
						|
email: rubygems@gitea.io
 | 
						|
executables: []
 | 
						|
extensions: []
 | 
						|
extra_rdoc_files: []
 | 
						|
files:
 | 
						|
- lib/gitea.rb
 | 
						|
homepage: https://gitea.io/
 | 
						|
licenses:
 | 
						|
- MIT
 | 
						|
metadata: {}
 | 
						|
post_install_message:
 | 
						|
rdoc_options: []
 | 
						|
require_paths:
 | 
						|
- lib
 | 
						|
required_ruby_version: !ruby/object:Gem::Requirement
 | 
						|
  requirements:
 | 
						|
  - - ">="
 | 
						|
    - !ruby/object:Gem::Version
 | 
						|
      version: 2.3.0
 | 
						|
required_rubygems_version: !ruby/object:Gem::Requirement
 | 
						|
  requirements:
 | 
						|
  - - ">="
 | 
						|
    - !ruby/object:Gem::Version
 | 
						|
      version: '1.0'
 | 
						|
requirements: []
 | 
						|
rubyforge_project:
 | 
						|
rubygems_version: 2.7.6.2
 | 
						|
signing_key:
 | 
						|
specification_version: 4
 | 
						|
summary: Gitea package
 | 
						|
test_files: []
 | 
						|
`, name, version)
 | 
						|
 | 
						|
	metadataGz := makeArchiveFileGz([]byte(metadataContent))
 | 
						|
	dataTarGz := makeArchiveFileGz(makeArchiveFileTar([]*tarFile{
 | 
						|
		{
 | 
						|
			Name: "lib/gitea.rb",
 | 
						|
			Data: []byte("class Gitea\nend"),
 | 
						|
		},
 | 
						|
	}))
 | 
						|
 | 
						|
	checksumsYaml := fmt.Sprintf(`---
 | 
						|
SHA256:
 | 
						|
  metadata.gz: %x
 | 
						|
  data.tar.gz: %x
 | 
						|
SHA512:
 | 
						|
  metadata.gz: %x
 | 
						|
  data.tar.gz: %x
 | 
						|
`, sha256.Sum256(metadataGz), sha256.Sum256(dataTarGz), sha512.Sum512(metadataGz), sha512.Sum512(dataTarGz))
 | 
						|
 | 
						|
	files := []*tarFile{
 | 
						|
		{
 | 
						|
			Name: "data.tar.gz",
 | 
						|
			Data: dataTarGz,
 | 
						|
		},
 | 
						|
		{
 | 
						|
			Name: "metadata.gz",
 | 
						|
			Data: metadataGz,
 | 
						|
		},
 | 
						|
		{
 | 
						|
			Name: "checksums.yaml.gz",
 | 
						|
			Data: makeArchiveFileGz([]byte(checksumsYaml)),
 | 
						|
		},
 | 
						|
	}
 | 
						|
	return makeArchiveFileTar(files)
 | 
						|
}
 | 
						|
 | 
						|
func TestPackageRubyGems(t *testing.T) {
 | 
						|
	defer tests.PrepareTestEnv(t)()
 | 
						|
 | 
						|
	user := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 2})
 | 
						|
 | 
						|
	testGemName := "gitea"
 | 
						|
	testGemVersion := "1.0.5"
 | 
						|
	testGemContent := makeRubyGem(testGemName, testGemVersion)
 | 
						|
	testGemContentChecksum := fmt.Sprintf("%x", sha256.Sum256(testGemContent))
 | 
						|
 | 
						|
	testAnotherGemName := "gitea-another"
 | 
						|
	testAnotherGemVersion := "0.99"
 | 
						|
 | 
						|
	root := fmt.Sprintf("/api/packages/%s/rubygems", user.Name)
 | 
						|
 | 
						|
	uploadFile := func(t *testing.T, content []byte, expectedStatus int) {
 | 
						|
		req := NewRequestWithBody(t, "POST", root+"/api/v1/gems", bytes.NewReader(content)).
 | 
						|
			AddBasicAuth(user.Name)
 | 
						|
		MakeRequest(t, req, expectedStatus)
 | 
						|
	}
 | 
						|
 | 
						|
	t.Run("Upload", func(t *testing.T) {
 | 
						|
		defer tests.PrintCurrentTest(t)()
 | 
						|
 | 
						|
		uploadFile(t, testGemContent, http.StatusCreated)
 | 
						|
 | 
						|
		pvs, err := packages.GetVersionsByPackageType(db.DefaultContext, user.ID, packages.TypeRubyGems)
 | 
						|
		assert.NoError(t, err)
 | 
						|
		assert.Len(t, pvs, 1)
 | 
						|
 | 
						|
		pd, err := packages.GetPackageDescriptor(db.DefaultContext, pvs[0])
 | 
						|
		assert.NoError(t, err)
 | 
						|
		assert.NotNil(t, pd.SemVer)
 | 
						|
		assert.IsType(t, &rubygems.Metadata{}, pd.Metadata)
 | 
						|
		assert.Equal(t, testGemName, pd.Package.Name)
 | 
						|
		assert.Equal(t, testGemVersion, pd.Version.Version)
 | 
						|
 | 
						|
		pfs, err := packages.GetFilesByVersionID(db.DefaultContext, pvs[0].ID)
 | 
						|
		assert.NoError(t, err)
 | 
						|
		assert.Len(t, pfs, 1)
 | 
						|
		assert.Equal(t, fmt.Sprintf("%s-%s.gem", testGemName, testGemVersion), pfs[0].Name)
 | 
						|
		assert.True(t, pfs[0].IsLead)
 | 
						|
 | 
						|
		pb, err := packages.GetBlobByID(db.DefaultContext, pfs[0].BlobID)
 | 
						|
		assert.NoError(t, err)
 | 
						|
		assert.EqualValues(t, len(testGemContent), pb.Size)
 | 
						|
	})
 | 
						|
 | 
						|
	t.Run("UploadExists", func(t *testing.T) {
 | 
						|
		defer tests.PrintCurrentTest(t)()
 | 
						|
		uploadFile(t, testGemContent, http.StatusConflict)
 | 
						|
	})
 | 
						|
 | 
						|
	t.Run("Download", func(t *testing.T) {
 | 
						|
		defer tests.PrintCurrentTest(t)()
 | 
						|
 | 
						|
		req := NewRequest(t, "GET", fmt.Sprintf("%s/gems/%s-%s.gem", root, testGemName, testGemVersion)).
 | 
						|
			AddBasicAuth(user.Name)
 | 
						|
		resp := MakeRequest(t, req, http.StatusOK)
 | 
						|
 | 
						|
		assert.Equal(t, testGemContent, resp.Body.Bytes())
 | 
						|
 | 
						|
		pvs, err := packages.GetVersionsByPackageType(db.DefaultContext, user.ID, packages.TypeRubyGems)
 | 
						|
		assert.NoError(t, err)
 | 
						|
		assert.Len(t, pvs, 1)
 | 
						|
		assert.Equal(t, int64(1), pvs[0].DownloadCount)
 | 
						|
	})
 | 
						|
 | 
						|
	t.Run("DownloadGemspec", func(t *testing.T) {
 | 
						|
		defer tests.PrintCurrentTest(t)()
 | 
						|
 | 
						|
		req := NewRequest(t, "GET", fmt.Sprintf("%s/quick/Marshal.4.8/%s-%s.gemspec.rz", root, testGemName, testGemVersion)).
 | 
						|
			AddBasicAuth(user.Name)
 | 
						|
		resp := MakeRequest(t, req, http.StatusOK)
 | 
						|
 | 
						|
		b, _ := base64.StdEncoding.DecodeString(`eJxi4Si1EndPzbWyCi5ITc5My0xOLMnMz2M8zMIRLeGpxGWsZ6RnzGbF5hqSyempxJWeWZKayGbN
 | 
						|
EBJqJQjWFZZaVJyZnxfN5qnEZahnoGcKkjTwVBJyB6lUKEhMzk5MTwULGngqcRaVJlWCONEMBp5K
 | 
						|
DGAWSKc7zFhPJamg0qRK99TcYphehZLU4hKInFhGSUlBsZW+PtgZepn5+iDxECRzDUDGcfh6hoA4
 | 
						|
gAAAAP//MS06Gw==`)
 | 
						|
		assert.Equal(t, b, resp.Body.Bytes())
 | 
						|
 | 
						|
		pvs, err := packages.GetVersionsByPackageType(db.DefaultContext, user.ID, packages.TypeRubyGems)
 | 
						|
		assert.NoError(t, err)
 | 
						|
		assert.Len(t, pvs, 1)
 | 
						|
		assert.Equal(t, int64(1), pvs[0].DownloadCount)
 | 
						|
	})
 | 
						|
 | 
						|
	t.Run("EnumeratePackages", func(t *testing.T) {
 | 
						|
		defer tests.PrintCurrentTest(t)()
 | 
						|
 | 
						|
		enumeratePackages := func(t *testing.T, endpoint string, expectedContent []byte) {
 | 
						|
			req := NewRequest(t, "GET", fmt.Sprintf("%s/%s", root, endpoint)).
 | 
						|
				AddBasicAuth(user.Name)
 | 
						|
			resp := MakeRequest(t, req, http.StatusOK)
 | 
						|
 | 
						|
			assert.Equal(t, expectedContent, resp.Body.Bytes())
 | 
						|
		}
 | 
						|
 | 
						|
		b, _ := base64.StdEncoding.DecodeString(`H4sICAAAAAAA/3NwZWNzLjQuOABi4Yhmi+bwVOJKzyxJTWSzYnMNCbUSdE/NtbIKSy0qzszPi2bzVOIy1DPQM2WzZgjxVOIsKk2qBDEBAQAA///xOEYKOwAAAA==`)
 | 
						|
		enumeratePackages(t, "specs.4.8.gz", b)
 | 
						|
		b, _ = base64.StdEncoding.DecodeString(`H4sICAAAAAAA/2xhdGVzdF9zcGVjcy40LjgAYuGIZovm8FTiSs8sSU1ks2JzDQm1EnRPzbWyCkstKs7Mz4tm81TiMtQz0DNls2YI8VTiLCpNqgQxAQEAAP//8ThGCjsAAAA=`)
 | 
						|
		enumeratePackages(t, "latest_specs.4.8.gz", b)
 | 
						|
		b, _ = base64.StdEncoding.DecodeString(`H4sICAAAAAAA/3ByZXJlbGVhc2Vfc3BlY3MuNC44AGLhiGYABAAA//9snXr5BAAAAA==`)
 | 
						|
		enumeratePackages(t, "prerelease_specs.4.8.gz", b)
 | 
						|
	})
 | 
						|
 | 
						|
	t.Run("UploadAnother", func(t *testing.T) {
 | 
						|
		defer tests.PrintCurrentTest(t)()
 | 
						|
		uploadFile(t, makeRubyGem(testAnotherGemName, testAnotherGemVersion), http.StatusCreated)
 | 
						|
	})
 | 
						|
 | 
						|
	t.Run("PackageInfo", func(t *testing.T) {
 | 
						|
		defer tests.PrintCurrentTest(t)()
 | 
						|
 | 
						|
		req := NewRequest(t, "GET", fmt.Sprintf("%s/info/%s", root, testGemName)).AddBasicAuth(user.Name)
 | 
						|
		resp := MakeRequest(t, req, http.StatusOK)
 | 
						|
		expected := fmt.Sprintf(`---
 | 
						|
1.0.5 runtime-dep:>= 1.2.0&< 2.0|checksum:%s,ruby:>= 2.3.0,rubygems:>= 1.0
 | 
						|
`, testGemContentChecksum)
 | 
						|
		assert.Equal(t, expected, resp.Body.String())
 | 
						|
	})
 | 
						|
 | 
						|
	t.Run("Versions", func(t *testing.T) {
 | 
						|
		defer tests.PrintCurrentTest(t)()
 | 
						|
		req := NewRequest(t, "GET", root+"/versions").AddBasicAuth(user.Name)
 | 
						|
		resp := MakeRequest(t, req, http.StatusOK)
 | 
						|
		assert.Equal(t, `---
 | 
						|
gitea 1.0.5 08843c2dd0ea19910e6b056b98e38f1c
 | 
						|
gitea-another 0.99 8b639e4048d282941485368ec42609be
 | 
						|
`, resp.Body.String())
 | 
						|
	})
 | 
						|
 | 
						|
	deleteGemPackage := func(t *testing.T, packageName, packageVersion string) {
 | 
						|
		body := bytes.Buffer{}
 | 
						|
		writer := multipart.NewWriter(&body)
 | 
						|
		_ = writer.WriteField("gem_name", packageName)
 | 
						|
		_ = writer.WriteField("version", packageVersion)
 | 
						|
		_ = writer.Close()
 | 
						|
		req := NewRequestWithBody(t, "DELETE", root+"/api/v1/gems/yank", &body).
 | 
						|
			SetHeader("Content-Type", writer.FormDataContentType()).
 | 
						|
			AddBasicAuth(user.Name)
 | 
						|
		MakeRequest(t, req, http.StatusOK)
 | 
						|
	}
 | 
						|
 | 
						|
	t.Run("DeleteAll", func(t *testing.T) {
 | 
						|
		defer tests.PrintCurrentTest(t)()
 | 
						|
		deleteGemPackage(t, testGemName, testGemVersion)
 | 
						|
		deleteGemPackage(t, testAnotherGemName, testAnotherGemVersion)
 | 
						|
		pvs, err := packages.GetVersionsByPackageType(db.DefaultContext, user.ID, packages.TypeRubyGems)
 | 
						|
		assert.NoError(t, err)
 | 
						|
		assert.Empty(t, pvs)
 | 
						|
	})
 | 
						|
 | 
						|
	t.Run("PackageInfoAfterDelete", func(t *testing.T) {
 | 
						|
		defer tests.PrintCurrentTest(t)()
 | 
						|
		req := NewRequest(t, "GET", fmt.Sprintf("%s/info/%s", root, testGemName)).AddBasicAuth(user.Name)
 | 
						|
		MakeRequest(t, req, http.StatusNotFound)
 | 
						|
	})
 | 
						|
 | 
						|
	t.Run("VersionsAfterDelete", func(t *testing.T) {
 | 
						|
		defer tests.PrintCurrentTest(t)()
 | 
						|
		req := NewRequest(t, "GET", root+"/versions").AddBasicAuth(user.Name)
 | 
						|
		resp := MakeRequest(t, req, http.StatusOK)
 | 
						|
		assert.Equal(t, "---\n", resp.Body.String())
 | 
						|
	})
 | 
						|
}
 |