stringmanip_test.go

package utils

import (
	"bytes"
	"reflect"
	"testing"
)

func TestExtractKW1(t *testing.T) {
	_, kwMap := ExtractKeywords([]byte(`<!--
title: test
theme: fest
-->
# Headline
text
`), []string{"title", "theme", "beard"})

	if !bytes.Equal(kwMap["title"], []byte("test")) {
		t.Errorf("Expected 'test', got '%s'", kwMap["title"])
	}

	if !bytes.Equal(kwMap["theme"], []byte("fest")) {
		t.Errorf("Expected 'fest', got '%s'", kwMap["theme"])
	}

	if _, ok := kwMap["beard"]; ok {
		t.Errorf("Expected 'beard' key to be absent, but it exists.")
	}

	if kwMap["beard"] != nil && len(kwMap["beard"]) != 0 {
		t.Errorf("Expected empty byte slice, got '%s'", kwMap["beard"])
	}
}

func TestExtractKW2(t *testing.T) {
	data := `
	% Best Title
	<!--
	weather: nice
	-->
	<!-- boat: missing -->
	<!-- horse: ignored -->

	# Page contents goes here

	And also here, but this should be ignored:
	title: nope
	weather: nope
	boat: nope

	All done.
	`
	_, kwMap := ExtractKeywords([]byte(data), []string{"title", "weather", "boat"})

	if !bytes.Equal(kwMap["title"], []byte("Best Title")) {
		t.Errorf("Expected 'Best Title', got '%s'", kwMap["title"])
	}

	if !bytes.Equal(kwMap["weather"], []byte("nice")) {
		t.Errorf("Expected 'nice', got '%s'", kwMap["weather"])
	}

	if !bytes.Equal(kwMap["boat"], []byte("missing")) {
		t.Errorf("Expected 'missing', got '%s'", kwMap["boat"])
	}

	if _, ok := kwMap["horse"]; ok {
		t.Errorf("Expected 'horse' key to be absent, but it exists.")
	}

	if kwMap["horse"] != nil && len(kwMap["horse"]) != 0 {
		t.Errorf("Expected empty byte slice, got '%s'", kwMap["horse"])
	}
}

func TestExtractLocalImagePaths(t *testing.T) {
	tests := []struct {
		html     string
		expected []string
	}{
		{
			html:     `<img src="local1.jpg"> <img src="http://remote.com/remote1.jpg"> <img src="local2.png"> <img src="https://remote.com/remote2.png">`,
			expected: []string{"local1.jpg", "local2.png"},
		},
		{
			html:     `<img src="/path/to/image.jpg"> <img src="anotherLocalImage.png">`,
			expected: []string{"/path/to/image.jpg", "anotherLocalImage.png"},
		},
		{
			html:     `<img src="https://remote.com/image.jpg">`,
			expected: []string{},
		},
		{
			html:     `<img src="localWithoutExtension">`,
			expected: []string{"localWithoutExtension"},
		},
	}
	for _, test := range tests {
		result := ExtractLocalImagePaths(test.html)
		if !reflect.DeepEqual(result, test.expected) {
			t.Errorf("For HTML %q, expected %#v but got %#v", test.html, test.expected, result)
		}
	}
}