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