X-Git-Url: http://git.scottworley.com/reliable-chat/blobdiff_plain/d4039f6319ebf5b91bcbf3128d488dfc6f7be3c3..b199796a3d2c82a84243e3a5ad3099315d8f1b55:/server/server_test.go?ds=sidebyside diff --git a/server/server_test.go b/server/server_test.go index fd4a4c8..0634b8f 100644 --- a/server/server_test.go +++ b/server/server_test.go @@ -2,23 +2,28 @@ package main import "testing" import "runtime" +import "strconv" import "time" func TestMessageInsertAndRetreive(t *testing.T) { + id := "1" say := "'Ello, Mister Polly Parrot!" at := time.Now() var zero_time time.Time store := start_store() - store.Add <- &Message{at, say} + store.Add <- &Message{at, id, say} messages_from_store := make(chan []Message, 1) store.Get <- &StoreRequest{zero_time, messages_from_store} messages := <-messages_from_store if len(messages) != 1 { - t.Fail() + t.FailNow() } if messages[0].Time != at { t.Fail() } + if messages[0].ID != id { + t.Fail() + } if messages[0].Text != say { t.Fail() } @@ -28,6 +33,7 @@ func TestMessageInsertAndRetreive(t *testing.T) { func TestFetchBlocksUntilSpeak(t *testing.T) { start_fetch_wait_count := fetch_wait_count.String() + id := "2" say := "I've got a lovely fresh cuttle fish for you" at := time.Now() var zero_time time.Time @@ -37,14 +43,17 @@ func TestFetchBlocksUntilSpeak(t *testing.T) { for start_fetch_wait_count == fetch_wait_count.String() { runtime.Gosched() } - store.Add <- &Message{at, say} + store.Add <- &Message{at, id, say} messages := <-messages_from_store if len(messages) != 1 { - t.Fail() + t.FailNow() } if messages[0].Time != at { t.Fail() } + if messages[0].ID != id { + t.Fail() + } if messages[0].Text != say { t.Fail() } @@ -53,6 +62,7 @@ func TestFetchBlocksUntilSpeak(t *testing.T) { } func TestMultipleListeners(t *testing.T) { + id := "3" say := "This is your nine o'clock alarm call!" at := time.Now() var zero_time time.Time @@ -63,15 +73,18 @@ func TestMultipleListeners(t *testing.T) { messages_from_store[i] = make(chan []Message, 1) store.Get <- &StoreRequest{zero_time, messages_from_store[i]} } - store.Add <- &Message{at, say} + store.Add <- &Message{at, id, say} for i := 0; i < num_clients; i++ { messages := <-messages_from_store[i] if len(messages) != 1 { - t.Fail() + t.FailNow() } if messages[0].Time != at { t.Fail() } + if messages[0].ID != id { + t.Fail() + } if messages[0].Text != say { t.Fail() } @@ -79,3 +92,171 @@ func TestMultipleListeners(t *testing.T) { close(store.Get) close(store.Add) } + +func parseDuration(s string) time.Duration { + d, err := time.ParseDuration(s) + if err != nil { + panic(err) + } + return d +} + +func atoi(s string) int { + i, err := strconv.Atoi(s) + if err != nil { + panic(err) + } + return i +} + +func TestPartialRetreive(t *testing.T) { + start_speak_count := atoi(speak_count.String()) + id1 := "4" + id2 := "5" + id3 := "6" + say1 := "No, no.....No, 'e's stunned!" + say2 := "You stunned him, just as he was wakin' up!" + say3 := "Norwegian Blues stun easily, major." + base := time.Now() + at1 := base.Add(parseDuration("-4m")) + since := base.Add(parseDuration("-3m")) + at2 := base.Add(parseDuration("-2m")) + at3 := base.Add(parseDuration("-1m")) + store := start_store() + store.Add <- &Message{at1, id1, say1} + store.Add <- &Message{at2, id2, say2} + store.Add <- &Message{at3, id3, say3} + for atoi(speak_count.String()) != start_speak_count+3 { + runtime.Gosched() + } + messages_from_store := make(chan []Message, 1) + store.Get <- &StoreRequest{since, messages_from_store} + messages := <-messages_from_store + if len(messages) != 2 { + t.FailNow() + } + if messages[0].Time != at2 { + t.Fail() + } + if messages[0].ID != id2 { + t.Fail() + } + if messages[0].Text != say2 { + t.Fail() + } + if messages[1].Time != at3 { + t.Fail() + } + if messages[1].ID != id3 { + t.Fail() + } + if messages[1].Text != say3 { + t.Fail() + } + close(store.Get) + close(store.Add) +} + +func TestPrecisePartialRetreive(t *testing.T) { + start_speak_count := atoi(speak_count.String()) + id1 := "7" + id2 := "8" + id3 := "9" + say1 := "Well, he's...he's, ah...probably pining for the fjords." + say2 := "PININ' for the FJORDS?!?!?!?" + say3 := "look, why did he fall flat on his back the moment I got 'im home?" + base := time.Now() + at1 := base.Add(parseDuration("-3m")) + at2 := base.Add(parseDuration("-2m")) + at3 := base.Add(parseDuration("-1m")) + since := at2 + store := start_store() + store.Add <- &Message{at1, id1, say1} + store.Add <- &Message{at2, id2, say2} + store.Add <- &Message{at3, id3, say3} + for atoi(speak_count.String()) != start_speak_count+3 { + runtime.Gosched() + } + messages_from_store := make(chan []Message, 1) + store.Get <- &StoreRequest{since, messages_from_store} + messages := <-messages_from_store + if len(messages) != 1 { + t.FailNow() + } + if messages[0].Time != at3 { + t.Fail() + } + if messages[0].ID != id3 { + t.Fail() + } + if messages[0].Text != say3 { + t.Fail() + } + close(store.Get) + close(store.Add) +} + +func TestTypicalFlow(t *testing.T) { + id1 := "10" + id2 := "11" + say1 := "The Norwegian Blue prefers kippin' on it's back!" + say2 := "Remarkable bird, innit, squire? Lovely plumage!" + store := start_store() + + // A waiting zero-time fetch. + var zero_time time.Time + prev_fetch_wait_count := fetch_wait_count.String() + fetch1 := make(chan []Message, 1) + store.Get <- &StoreRequest{zero_time, fetch1} + for prev_fetch_wait_count == fetch_wait_count.String() { + runtime.Gosched() + } + + // Someone speaks. This triggers delivery. + at1 := time.Now() + store.Add <- &Message{at1, id1, say1} + messages1 := <-fetch1 + if len(messages1) != 1 { + t.FailNow() + } + if messages1[0].Time != at1 { + t.Fail() + } + if messages1[0].ID != id1 { + t.Fail() + } + if messages1[0].Text != say1 { + t.Fail() + } + + // Upon recipt, client blocks on fetch with since=at1 + prev_fetch_wait_count = fetch_wait_count.String() + fetch2 := make(chan []Message, 1) + store.Get <- &StoreRequest{at1, fetch2} + for prev_fetch_wait_count == fetch_wait_count.String() { + runtime.Gosched() + } + + // Someone speaks again. This triggers another delivery. + at2 := time.Now() + if !at2.After(at1) { + t.Fail() + } + store.Add <- &Message{at2, id2, say2} + messages2 := <-fetch2 + if len(messages2) != 1 { + t.FailNow() + } + if messages2[0].Time != at2 { + t.Fail() + } + if messages2[0].ID != id2 { + t.Fail() + } + if messages2[0].Text != say2 { + t.Fail() + } + + close(store.Get) + close(store.Add) +}