101 lines
2.7 KiB
Go
101 lines
2.7 KiB
Go
package monitor
|
|
|
|
import (
|
|
"bytes"
|
|
"net/http"
|
|
"strconv"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/golang-collections/go-datastructures/queue"
|
|
)
|
|
|
|
const testmport = ":13152"
|
|
|
|
func Test_MonitorListen(t *testing.T) {
|
|
m, err := New(testmport)
|
|
if err != nil {
|
|
t.Fatalf("cannot create new monitor: %v", err)
|
|
}
|
|
if err := m.Start(); err != nil {
|
|
t.Fatalf("cannot start monitor: %v", err)
|
|
}
|
|
defer m.Stop()
|
|
|
|
if resp, err := http.Get("http://localhost" + testmport + "/mia"); err != nil {
|
|
t.Fatalf("GET error: %v", err)
|
|
} else if resp.StatusCode != http.StatusNotFound {
|
|
t.Errorf("GET /mia didn't 404: got %v", resp.StatusCode)
|
|
}
|
|
|
|
if resp, err := http.Get("http://localhost" + testmport + "/newfeed"); err != nil {
|
|
t.Fatalf("GET error: %v", err)
|
|
} else if resp.StatusCode != http.StatusNotFound {
|
|
t.Errorf("GET /newfeed didn't 404: got %v", resp.StatusCode)
|
|
}
|
|
|
|
if resp, err := http.Post("http://localhost"+testmport+"/newfeed", "application/json", bytes.NewBuffer([]byte(""))); err != nil {
|
|
t.Fatalf("POST error: %v", err)
|
|
} else if resp.StatusCode != http.StatusBadRequest {
|
|
t.Errorf("POST /newfeed didn't 200: got %v", resp.StatusCode)
|
|
}
|
|
|
|
if resp, err := http.Post("http://localhost"+testmport+"/newfeed", "application/json", bytes.NewBuffer([]byte(`{"URL":"hello", "Interval":"5m"}`))); err != nil {
|
|
t.Fatalf("POST error: %v", err)
|
|
} else if resp.StatusCode != http.StatusOK {
|
|
t.Errorf("POST /newfeed didn't 200: got %v", resp.StatusCode)
|
|
}
|
|
}
|
|
|
|
func Test_Monitor(t *testing.T) {
|
|
m, err := New(testmport)
|
|
if err != nil {
|
|
t.Fatalf("cannot create new monitor: %v", err)
|
|
}
|
|
itemsNew := make(chan Item, 1)
|
|
itemsDone := make(chan Item, 1)
|
|
m.newItems = itemsNew
|
|
m.triggeredItems = itemsDone
|
|
|
|
if err := m.Start(); err != nil {
|
|
t.Fatalf("cannot start monitor: %v", err)
|
|
}
|
|
|
|
for i := 0; i < 2; i++ {
|
|
item := NewItem("item"+strconv.Itoa(i), time.Second+time.Second*time.Duration(i)*10)
|
|
select {
|
|
case itemsNew <- *item:
|
|
case <-time.After(time.Second * 5):
|
|
t.Fatalf("could not add new item in time limit")
|
|
}
|
|
}
|
|
|
|
for i := 0; i < 2; i++ {
|
|
select {
|
|
case triggered := <-itemsDone:
|
|
if triggered.URL != "item"+strconv.Itoa(i) {
|
|
t.Fatalf("wrong item done order: %d was %v", i, triggered)
|
|
}
|
|
case <-time.After(time.Second * 5):
|
|
t.Fatalf("could not get done item in time limit")
|
|
}
|
|
}
|
|
|
|
if err := m.Stop(); err != nil {
|
|
t.Fatalf("could not stop monitor: %v", err)
|
|
}
|
|
}
|
|
|
|
func Test_NextEvent(t *testing.T) {
|
|
q := queue.NewPriorityQueue(1)
|
|
a := NewItem("a", time.Second*5)
|
|
b := NewItem("b", time.Second*500)
|
|
q.Put(a)
|
|
q.Put(b)
|
|
if time, err := nextEventTime(q); err != nil {
|
|
t.Fatalf("could not get next event time: %v", err)
|
|
} else if time != a.next {
|
|
t.Fatalf("got wrong next event time: %v, expected %v", time, a.next)
|
|
}
|
|
}
|