mirror of
https://github.com/bitechdev/ResolveSpec.git
synced 2026-03-07 14:08:54 +00:00
staticweb package for easier static web server hosting
This commit is contained in:
Hein
393
pkg/server/staticweb/providers/providers_test.go
Normal file
393
pkg/server/staticweb/providers/providers_test.go
Normal file
@@ -0,0 +1,393 @@
|
||||
package providers
|
||||
|
||||
import (
|
||||
"archive/zip"
|
||||
"bytes"
|
||||
"io"
|
||||
"io/fs"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"testing"
|
||||
"time"
|
||||
)
|
||||
|
||||
func TestLocalFSProvider(t *testing.T) {
|
||||
// Create a temporary directory with test files
|
||||
tmpDir := t.TempDir()
|
||||
|
||||
testFile := filepath.Join(tmpDir, "test.txt")
|
||||
if err := os.WriteFile(testFile, []byte("test content"), 0644); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
provider, err := NewLocalFSProvider(tmpDir)
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to create provider: %v", err)
|
||||
}
|
||||
defer provider.Close()
|
||||
|
||||
// Test opening a file
|
||||
file, err := provider.Open("test.txt")
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to open file: %v", err)
|
||||
}
|
||||
defer file.Close()
|
||||
|
||||
// Read the file
|
||||
data, err := io.ReadAll(file)
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to read file: %v", err)
|
||||
}
|
||||
|
||||
if string(data) != "test content" {
|
||||
t.Errorf("Expected 'test content', got %q", string(data))
|
||||
}
|
||||
|
||||
// Test type
|
||||
if provider.Type() != "local" {
|
||||
t.Errorf("Expected type 'local', got %q", provider.Type())
|
||||
}
|
||||
}
|
||||
|
||||
func TestZipFSProvider(t *testing.T) {
|
||||
// Create a temporary zip file
|
||||
tmpDir := t.TempDir()
|
||||
zipPath := filepath.Join(tmpDir, "test.zip")
|
||||
|
||||
// Create zip file with test content
|
||||
zipFile, err := os.Create(zipPath)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
zipWriter := zip.NewWriter(zipFile)
|
||||
fileWriter, err := zipWriter.Create("test.txt")
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
_, err = fileWriter.Write([]byte("zip content"))
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
if err := zipWriter.Close(); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
if err := zipFile.Close(); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
// Test the provider
|
||||
provider, err := NewZipFSProvider(zipPath)
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to create provider: %v", err)
|
||||
}
|
||||
defer provider.Close()
|
||||
|
||||
// Test opening a file
|
||||
file, err := provider.Open("test.txt")
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to open file: %v", err)
|
||||
}
|
||||
defer file.Close()
|
||||
|
||||
// Read the file
|
||||
data, err := io.ReadAll(file)
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to read file: %v", err)
|
||||
}
|
||||
|
||||
if string(data) != "zip content" {
|
||||
t.Errorf("Expected 'zip content', got %q", string(data))
|
||||
}
|
||||
|
||||
// Test type
|
||||
if provider.Type() != "zip" {
|
||||
t.Errorf("Expected type 'zip', got %q", provider.Type())
|
||||
}
|
||||
}
|
||||
|
||||
func TestZipFSProviderReload(t *testing.T) {
|
||||
// Create a temporary zip file
|
||||
tmpDir := t.TempDir()
|
||||
zipPath := filepath.Join(tmpDir, "test.zip")
|
||||
|
||||
// Helper to create zip with content
|
||||
createZip := func(content string) {
|
||||
zipFile, err := os.Create(zipPath)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
defer zipFile.Close()
|
||||
|
||||
zipWriter := zip.NewWriter(zipFile)
|
||||
fileWriter, err := zipWriter.Create("test.txt")
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
_, err = fileWriter.Write([]byte(content))
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
if err := zipWriter.Close(); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
}
|
||||
|
||||
// Create initial zip
|
||||
createZip("original content")
|
||||
|
||||
// Test the provider
|
||||
provider, err := NewZipFSProvider(zipPath)
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to create provider: %v", err)
|
||||
}
|
||||
defer provider.Close()
|
||||
|
||||
// Read initial content
|
||||
file, err := provider.Open("test.txt")
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to open file: %v", err)
|
||||
}
|
||||
data, _ := io.ReadAll(file)
|
||||
file.Close()
|
||||
|
||||
if string(data) != "original content" {
|
||||
t.Errorf("Expected 'original content', got %q", string(data))
|
||||
}
|
||||
|
||||
// Update the zip file
|
||||
createZip("updated content")
|
||||
|
||||
// Reload the provider
|
||||
if err := provider.Reload(); err != nil {
|
||||
t.Fatalf("Failed to reload: %v", err)
|
||||
}
|
||||
|
||||
// Read updated content
|
||||
file, err = provider.Open("test.txt")
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to open file after reload: %v", err)
|
||||
}
|
||||
data, _ = io.ReadAll(file)
|
||||
file.Close()
|
||||
|
||||
if string(data) != "updated content" {
|
||||
t.Errorf("Expected 'updated content', got %q", string(data))
|
||||
}
|
||||
}
|
||||
|
||||
func TestLocalFSProviderReload(t *testing.T) {
|
||||
// Create a temporary directory with test files
|
||||
tmpDir := t.TempDir()
|
||||
|
||||
testFile := filepath.Join(tmpDir, "test.txt")
|
||||
if err := os.WriteFile(testFile, []byte("original"), 0644); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
provider, err := NewLocalFSProvider(tmpDir)
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to create provider: %v", err)
|
||||
}
|
||||
defer provider.Close()
|
||||
|
||||
// Read initial content
|
||||
file, err := provider.Open("test.txt")
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to open file: %v", err)
|
||||
}
|
||||
data, _ := io.ReadAll(file)
|
||||
file.Close()
|
||||
|
||||
if string(data) != "original" {
|
||||
t.Errorf("Expected 'original', got %q", string(data))
|
||||
}
|
||||
|
||||
// Update the file
|
||||
if err := os.WriteFile(testFile, []byte("updated"), 0644); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
// Reload the provider
|
||||
if err := provider.Reload(); err != nil {
|
||||
t.Fatalf("Failed to reload: %v", err)
|
||||
}
|
||||
|
||||
// Read updated content
|
||||
file, err = provider.Open("test.txt")
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to open file after reload: %v", err)
|
||||
}
|
||||
data, _ = io.ReadAll(file)
|
||||
file.Close()
|
||||
|
||||
if string(data) != "updated" {
|
||||
t.Errorf("Expected 'updated', got %q", string(data))
|
||||
}
|
||||
}
|
||||
|
||||
func TestEmbedFSProvider(t *testing.T) {
|
||||
// Test with a mock embed.FS
|
||||
mockFS := &mockEmbedFS{
|
||||
files: map[string][]byte{
|
||||
"test.txt": []byte("test content"),
|
||||
},
|
||||
}
|
||||
|
||||
provider, err := NewEmbedFSProvider(mockFS, "")
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to create provider: %v", err)
|
||||
}
|
||||
defer provider.Close()
|
||||
|
||||
// Test type
|
||||
if provider.Type() != "embed" {
|
||||
t.Errorf("Expected type 'embed', got %q", provider.Type())
|
||||
}
|
||||
|
||||
// Test opening a file
|
||||
file, err := provider.Open("test.txt")
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to open file: %v", err)
|
||||
}
|
||||
defer file.Close()
|
||||
|
||||
// Read the file
|
||||
data, err := io.ReadAll(file)
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to read file: %v", err)
|
||||
}
|
||||
|
||||
if string(data) != "test content" {
|
||||
t.Errorf("Expected 'test content', got %q", string(data))
|
||||
}
|
||||
}
|
||||
|
||||
func TestEmbedFSProviderWithZip(t *testing.T) {
|
||||
// Create an embedded-like FS with a zip file
|
||||
// For simplicity, we'll use a mock embed.FS
|
||||
tmpDir := t.TempDir()
|
||||
zipPath := filepath.Join(tmpDir, "test.zip")
|
||||
|
||||
// Create zip file
|
||||
zipFile, err := os.Create(zipPath)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
zipWriter := zip.NewWriter(zipFile)
|
||||
fileWriter, err := zipWriter.Create("test.txt")
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
_, err = fileWriter.Write([]byte("embedded zip content"))
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
zipWriter.Close()
|
||||
zipFile.Close()
|
||||
|
||||
// Read the zip file
|
||||
zipData, err := os.ReadFile(zipPath)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
// Create a mock embed.FS
|
||||
mockFS := &mockEmbedFS{
|
||||
files: map[string][]byte{
|
||||
"test.zip": zipData,
|
||||
},
|
||||
}
|
||||
|
||||
provider, err := NewEmbedFSProvider(mockFS, "test.zip")
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to create provider: %v", err)
|
||||
}
|
||||
defer provider.Close()
|
||||
|
||||
// Test opening a file
|
||||
file, err := provider.Open("test.txt")
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to open file: %v", err)
|
||||
}
|
||||
defer file.Close()
|
||||
|
||||
// Read the file
|
||||
data, err := io.ReadAll(file)
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to read file: %v", err)
|
||||
}
|
||||
|
||||
if string(data) != "embedded zip content" {
|
||||
t.Errorf("Expected 'embedded zip content', got %q", string(data))
|
||||
}
|
||||
|
||||
// Test type
|
||||
if provider.Type() != "embed-zip" {
|
||||
t.Errorf("Expected type 'embed-zip', got %q", provider.Type())
|
||||
}
|
||||
}
|
||||
|
||||
// mockEmbedFS is a mock embed.FS for testing
|
||||
type mockEmbedFS struct {
|
||||
files map[string][]byte
|
||||
}
|
||||
|
||||
func (m *mockEmbedFS) Open(name string) (fs.File, error) {
|
||||
data, ok := m.files[name]
|
||||
if !ok {
|
||||
return nil, os.ErrNotExist
|
||||
}
|
||||
|
||||
return &mockFile{
|
||||
name: name,
|
||||
reader: bytes.NewReader(data),
|
||||
size: int64(len(data)),
|
||||
}, nil
|
||||
}
|
||||
|
||||
func (m *mockEmbedFS) ReadFile(name string) ([]byte, error) {
|
||||
data, ok := m.files[name]
|
||||
if !ok {
|
||||
return nil, os.ErrNotExist
|
||||
}
|
||||
return data, nil
|
||||
}
|
||||
|
||||
type mockFile struct {
|
||||
name string
|
||||
reader *bytes.Reader
|
||||
size int64
|
||||
}
|
||||
|
||||
func (f *mockFile) Stat() (fs.FileInfo, error) {
|
||||
return &mockFileInfo{name: f.name, size: f.size}, nil
|
||||
}
|
||||
|
||||
func (f *mockFile) Read(p []byte) (int, error) {
|
||||
return f.reader.Read(p)
|
||||
}
|
||||
|
||||
func (f *mockFile) Close() error {
|
||||
return nil
|
||||
}
|
||||
|
||||
type mockFileInfo struct {
|
||||
name string
|
||||
size int64
|
||||
}
|
||||
|
||||
func (fi *mockFileInfo) Name() string { return fi.name }
|
||||
func (fi *mockFileInfo) Size() int64 { return fi.size }
|
||||
func (fi *mockFileInfo) Mode() fs.FileMode { return 0644 }
|
||||
func (fi *mockFileInfo) ModTime() time.Time { return time.Now() }
|
||||
func (fi *mockFileInfo) IsDir() bool { return false }
|
||||
func (fi *mockFileInfo) Sys() interface{} { return nil }
|
||||
Reference in New Issue
Block a user