51 lines
911 B
Go
51 lines
911 B
Go
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"reflect"
|
||
|
|
"testing"
|
||
|
|
"time"
|
||
|
|
)
|
||
|
|
|
||
|
|
func mockWebsiteChecker(url string) bool {
|
||
|
|
if url == "waat://furhurterwe.geds" {
|
||
|
|
return false
|
||
|
|
}
|
||
|
|
return true
|
||
|
|
}
|
||
|
|
|
||
|
|
func slowStubWebsiteChecker(_ string) bool {
|
||
|
|
time.Sleep(20 * time.Millisecond)
|
||
|
|
return true
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestCheckWebsites(t *testing.T) {
|
||
|
|
websites := []string{
|
||
|
|
"http://google.com",
|
||
|
|
"http://blog.gypsydave5.com",
|
||
|
|
"waat://furhurterwe.geds",
|
||
|
|
}
|
||
|
|
|
||
|
|
want := map[string]bool{
|
||
|
|
"http://google.com": true,
|
||
|
|
"http://blog.gypsydave5.com": true,
|
||
|
|
"waat://furhurterwe.geds": false,
|
||
|
|
}
|
||
|
|
|
||
|
|
got := CheckWebsites(mockWebsiteChecker, websites)
|
||
|
|
|
||
|
|
if !reflect.DeepEqual(want, got) {
|
||
|
|
t.Fatalf("Wanted %v, got %v", want, got)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func BenchmarkCheckWebsites(b *testing.B) {
|
||
|
|
urls := make([]string, 100)
|
||
|
|
for i := 0; i < len(urls); i++ {
|
||
|
|
urls[i] = "a url"
|
||
|
|
}
|
||
|
|
|
||
|
|
for i := 0; i < b.N; i++ {
|
||
|
|
CheckWebsites(slowStubWebsiteChecker, urls)
|
||
|
|
}
|
||
|
|
}
|