package linkinfo

import (
	"strings"
	"testing"
)

func Test_detectContentType(t *testing.T) {
	contentType := detectContentType("http://example.com", "")

	if contentType == "" {
		t.Fatal("Unable to test example.com, returned empty type")
	}

	if idx := strings.Index(contentType, ";"); idx != -1 {
		contentType = contentType[:idx]
	}

	if contentType != ContentTypeHtml {
		t.Fatal("Unexpected content type:", contentType)
	}
}

func Test_defaultLinkHandler(t *testing.T) {
	type expectedData struct {
		Link  string
		Title string
		Type  string
	}

	testLinks := []expectedData{
		{"https://paste.ee", "Paste.ee", "text/html"},
		{"http://techslides.com/demos/sample-videos/small.mp4", "", "video/mp4"},
	}

	for _, link := range testLinks {
		l, err := defaultLinkHandler(link.Link)

		if err != nil {
			t.Fatal("Unable to retrieve link info:", err)
		}

		if link.Title != "" && link.Title != l.Title {
			t.Fatal("Unexpected title, expected:", link.Title, "got:", l.Title)
		}

		if link.Type != "" && link.Type != l.ContentType {
			t.Fatal("Unexpected content type, expected:", link.Type, "got:", l.ContentType)
		}
	}
}