forge/modules/sitemap/sitemap_test.go at agent/forge-agent-access

mirror of https://github.com/hanzoai/forge.git synced 2026-10-30 15:00:39 +00:00

Files

T

zeekayandHanzo Dev aa87442fdd Record upstream authorship centrally in NOTICE instead of in every file header

Source files keep their SPDX-License-Identifier and Hanzo copyright line; the
Gitea, Gogs and LFS Test Server copyrights they used to repeat now live once in
NOTICE, alongside LICENSE. The header lint accepts any copyright line above the
SPDX line, so it no longer demands an upstream one.

Co-authored-by: Hanzo Dev <[email protected]>

2026-09-10 17:50:35 -07:00

167 lines

4.3 KiB

Go

Raw Permalink Blame History

 // Copyright 2026 Hanzo AI, Inc. All rights reserved.
 // SPDX-License-Identifier: MIT
 package sitemap
 import (
 	"bytes"
 	"encoding/xml"
 	"strings"
 	"testing"
 	"time"
 	"github.com/stretchr/testify/assert"
 )
 func TestNewSitemap(t *testing.T) {
 	ts := time.Unix(1651322008, 0).UTC()
 	tests := []struct {
 		name    string
 		urls    []URL
 		want    string
 		wantErr string
 	}{
 		{
 			name: "empty",
 			urls: []URL{},
 			want: xml.Header + `<urlset xmlns="http://www.sitemaps.org/schemas/sitemap/0.9">` +
 				"" +
 				"</urlset>\n",
 		},
 		{
 			name: "regular",
 			urls: []URL{
 				{URL: "https://forge.example/test1", LastMod: &ts},
 			},
 			want: xml.Header + `<urlset xmlns="http://www.sitemaps.org/schemas/sitemap/0.9">` +
 				"<url><loc>https://forge.example/test1</loc><lastmod>2022-04-30T12:33:28Z</lastmod></url>" +
 				"</urlset>\n",
 		},
 		{
 			name: "without lastmod",
 			urls: []URL{
 				{URL: "https://forge.example/test1"},
 			},
 			want: xml.Header + `<urlset xmlns="http://www.sitemaps.org/schemas/sitemap/0.9">` +
 				"<url><loc>https://forge.example/test1</loc></url>" +
 				"</urlset>\n",
 		},
 		{
 			name: "multiple",
 			urls: []URL{
 				{URL: "https://forge.example/test1", LastMod: &ts},
 				{URL: "https://forge.example/test2", LastMod: nil},
 			},
 			want: xml.Header + `<urlset xmlns="http://www.sitemaps.org/schemas/sitemap/0.9">` +
 				"<url><loc>https://forge.example/test1</loc><lastmod>2022-04-30T12:33:28Z</lastmod></url>" +
 				"<url><loc>https://forge.example/test2</loc></url>" +
 				"</urlset>\n",
 		},
 		{
 			name:    "too many urls",
 			urls:    make([]URL, 50001),
 			wantErr: "sitemap contains 50001 URLs, but only 50000 are allowed",
 		},
 		{
 			name: "too big file",
 			urls: []URL{
 				{URL: strings.Repeat("b", 50*1024*1024+1)},
 			},
 			wantErr: "sitemap has 52428932 bytes, but only 52428800 are allowed",
 		},
 	}
 	for _, tt := range tests {
 		t.Run(tt.name, func(t *testing.T) {
 			s := NewSitemap()
 			for _, url := range tt.urls {
 				s.Add(url)
 			}
 			buf := &bytes.Buffer{}
 			_, err := s.WriteTo(buf)
 			if tt.wantErr != "" {
 				assert.EqualError(t, err, tt.wantErr)
 			} else {
 				assert.NoError(t, err)
 				assert.Equalf(t, tt.want, buf.String(), "NewSitemap()")
 			}
 		})
 	}
 }
 func TestNewSitemapIndex(t *testing.T) {
 	ts := time.Unix(1651322008, 0).UTC()
 	tests := []struct {
 		name    string
 		urls    []URL
 		want    string
 		wantErr string
 	}{
 		{
 			name: "empty",
 			urls: []URL{},
 			want: xml.Header + `<sitemapindex xmlns="http://www.sitemaps.org/schemas/sitemap/0.9">` +
 				"" +
 				"</sitemapindex>\n",
 		},
 		{
 			name: "regular",
 			urls: []URL{
 				{URL: "https://forge.example/test1", LastMod: &ts},
 			},
 			want: xml.Header + `<sitemapindex xmlns="http://www.sitemaps.org/schemas/sitemap/0.9">` +
 				"<sitemap><loc>https://forge.example/test1</loc><lastmod>2022-04-30T12:33:28Z</lastmod></sitemap>" +
 				"</sitemapindex>\n",
 		},
 		{
 			name: "without lastmod",
 			urls: []URL{
 				{URL: "https://forge.example/test1"},
 			},
 			want: xml.Header + `<sitemapindex xmlns="http://www.sitemaps.org/schemas/sitemap/0.9">` +
 				"<sitemap><loc>https://forge.example/test1</loc></sitemap>" +
 				"</sitemapindex>\n",
 		},
 		{
 			name: "multiple",
 			urls: []URL{
 				{URL: "https://forge.example/test1", LastMod: &ts},
 				{URL: "https://forge.example/test2", LastMod: nil},
 			},
 			want: xml.Header + `<sitemapindex xmlns="http://www.sitemaps.org/schemas/sitemap/0.9">` +
 				"<sitemap><loc>https://forge.example/test1</loc><lastmod>2022-04-30T12:33:28Z</lastmod></sitemap>" +
 				"<sitemap><loc>https://forge.example/test2</loc></sitemap>" +
 				"</sitemapindex>\n",
 		},
 		{
 			name:    "too many sitemaps",
 			urls:    make([]URL, 50001),
 			wantErr: "sitemap contains 50001 sub-sitemaps, but only 50000 are allowed",
 		},
 		{
 			name: "too big file",
 			urls: []URL{
 				{URL: strings.Repeat("b", 50*1024*1024+1)},
 			},
 			wantErr: "sitemap has 52428952 bytes, but only 52428800 are allowed",
 		},
 	}
 	for _, tt := range tests {
 		t.Run(tt.name, func(t *testing.T) {
 			s := NewSitemapIndex()
 			for _, url := range tt.urls {
 				s.Add(url)
 			}
 			buf := &bytes.Buffer{}
 			_, err := s.WriteTo(buf)
 			if tt.wantErr != "" {
 				assert.EqualError(t, err, tt.wantErr)
 			} else {
 				assert.NoError(t, err)
 				assert.Equalf(t, tt.want, buf.String(), "NewSitemapIndex()")
 			}
 		})
 	}
 }

167 lines 4.3 KiB Go Raw Permalink Blame History

167 lines

4.3 KiB

Go

Raw Permalink Blame History