package main

import (
	"testing"
)

func isExpressionAttributeValueURL(elementName, attrName string) bool {
	var elementAttributeLookup = map[string]string{
		"a":      "href",
		"form":   "action",
		"link":   "href",
		"object": "data",
	}
	expectedAttribute, exists := elementAttributeLookup[elementName]
	if !exists {
		return false
	}
	return attrName == expectedAttribute
}

func isExpressionAttributeValueURL2(elementName, attrName string) bool {
	switch elementName {
	case "a", "link":
		return attrName == "href"
	case "form":
		return attrName == "action"
	case "object":
		return attrName == "data"
	default:
		return false
	}
}

var elementAttributeLookup3 = map[string]string{
	"a":      "href",
	"form":   "action",
	"link":   "href",
	"object": "data",
}

func isExpressionAttributeValueURL3(elementName, attrName string) bool {
	expectedAttribute, exists := elementAttributeLookup3[elementName]
	if !exists {
		return false
	}
	return attrName == expectedAttribute
}

func BenchmarkIsExpressionAttributeValueURL(b *testing.B) {
	b.ReportAllocs()
	for b.Loop() {
		_ = isExpressionAttributeValueURL("a", "href")
	}
}

func BenchmarkIsExpressionAttributeValueURL2(b *testing.B) {
	b.ReportAllocs()
	for b.Loop() {
		_ = isExpressionAttributeValueURL2("a", "href")
	}
}

func BenchmarkIsExpressionAttributeValueURL3(b *testing.B) {
	b.ReportAllocs()
	for b.Loop() {
		_ = isExpressionAttributeValueURL3("a", "href")
	}
}