106 lines
2.6 KiB
Go
106 lines
2.6 KiB
Go
package artifact
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"os"
|
|
"testing"
|
|
|
|
"gitea.maximumdirect.net/eric/scriptorium/internal/domain"
|
|
)
|
|
|
|
func TestCompositeReader_Read(t *testing.T) {
|
|
reader := NewCompositeReader()
|
|
ctx := context.Background()
|
|
|
|
t.Run("inline artifact", func(t *testing.T) {
|
|
ref := domain.ArtifactRef{
|
|
Type: domain.ArtifactRefInline,
|
|
Body: "hello world",
|
|
}
|
|
art, err := reader.Read(ctx, ref)
|
|
if err != nil {
|
|
t.Fatalf("unexpected error: %v", err)
|
|
}
|
|
if string(art.Body) != "hello world" {
|
|
t.Errorf("expected 'hello world', got %s", string(art.Body))
|
|
}
|
|
if art.ContentType != "text/plain" {
|
|
t.Errorf("expected text/plain content type, got %q", art.ContentType)
|
|
}
|
|
if art.Hash != "b94d27b9934d3e08a52e52d7da7dabfac484efe37a5380ee9088f7ace2efcde9" {
|
|
t.Errorf("unexpected hash: %s", art.Hash)
|
|
}
|
|
})
|
|
|
|
t.Run("inline artifact missing body", func(t *testing.T) {
|
|
ref := domain.ArtifactRef{
|
|
Type: domain.ArtifactRefInline,
|
|
Body: "",
|
|
}
|
|
_, err := reader.Read(ctx, ref)
|
|
if !errors.Is(err, ErrMissingInlineBody) {
|
|
t.Errorf("expected ErrMissingInlineBody, got %v", err)
|
|
}
|
|
})
|
|
|
|
t.Run("unsupported ref type", func(t *testing.T) {
|
|
ref := domain.ArtifactRef{
|
|
Type: domain.ArtifactRefS3,
|
|
URI: "s3://bucket/key",
|
|
}
|
|
_, err := reader.Read(ctx, ref)
|
|
if !errors.Is(err, ErrUnsupportedRefType) {
|
|
t.Error("expected error for unsupported type")
|
|
}
|
|
})
|
|
}
|
|
|
|
func TestFileReader_Read(t *testing.T) {
|
|
content := []byte("test file content")
|
|
tmpFile, err := os.CreateTemp("", "artifact_test_*.txt")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer os.Remove(tmpFile.Name())
|
|
|
|
if _, err := tmpFile.Write(content); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
tmpFile.Close()
|
|
|
|
reader := NewCompositeReader()
|
|
ctx := context.Background()
|
|
|
|
t.Run("file artifact loading", func(t *testing.T) {
|
|
ref := domain.ArtifactRef{
|
|
Type: domain.ArtifactRefFile,
|
|
URI: tmpFile.Name(),
|
|
}
|
|
art, err := reader.Read(ctx, ref)
|
|
if err != nil {
|
|
t.Fatalf("unexpected error: %v", err)
|
|
}
|
|
if string(art.Body) != string(content) {
|
|
t.Errorf("expected %s, got %s", string(content), string(art.Body))
|
|
}
|
|
if art.Name == "" {
|
|
t.Error("expected name to be inferred from filename")
|
|
}
|
|
if art.Hash != "60f5237ed4049f0382661ef009d2bc42e48c3ceb3edb6600f7024e7ab3b838f3" {
|
|
t.Errorf("unexpected hash: %s", art.Hash)
|
|
}
|
|
})
|
|
|
|
t.Run("missing file path", func(t *testing.T) {
|
|
ref := domain.ArtifactRef{
|
|
Type: domain.ArtifactRefFile,
|
|
URI: "",
|
|
}
|
|
_, err := reader.Read(ctx, ref)
|
|
if !errors.Is(err, ErrMissingFilePath) {
|
|
t.Errorf("expected ErrMissingFilePath, got %v", err)
|
|
}
|
|
})
|
|
}
|