git.scottworley.com Git - reliable-chat/blob - server/server_test.go

   1 package main
   2
   3 import "testing"
   4 import "runtime"
   5 import "strconv"
   6 import "time"
   7
   8 func TestMessageInsertAndRetreive(t *testing.T) {
   9         id := "1"
  10         say := "'Ello, Mister Polly Parrot!"
  11         at := time.Now()
  12         var zero_time time.Time
  13         store := start_store()
  14         store.Add <- &Message{at, id, say}
  15         messages_from_store := make(chan []Message, 1)
  16         store.Get <- &StoreRequest{zero_time, messages_from_store}
  17         messages := <-messages_from_store
  18         if len(messages) != 1 {
  19                 t.FailNow()
  20         }
  21         if messages[0].Time != at {
  22                 t.Fail()
  23         }
  24         if messages[0].ID != id {
  25                 t.Fail()
  26         }
  27         if messages[0].Text != say {
  28                 t.Fail()
  29         }
  30         close(store.Get)
  31         close(store.Add)
  32 }
  33
  34 func TestFetchBlocksUntilSpeak(t *testing.T) {
  35         start_fetch_wait_count := fetch_wait_count.String()
  36         id := "2"
  37         say := "I've got a lovely fresh cuttle fish for you"
  38         at := time.Now()
  39         var zero_time time.Time
  40         store := start_store()
  41         messages_from_store := make(chan []Message, 1)
  42         store.Get <- &StoreRequest{zero_time, messages_from_store}
  43         for start_fetch_wait_count == fetch_wait_count.String() {
  44                 runtime.Gosched()
  45         }
  46         store.Add <- &Message{at, id, say}
  47         messages := <-messages_from_store
  48         if len(messages) != 1 {
  49                 t.FailNow()
  50         }
  51         if messages[0].Time != at {
  52                 t.Fail()
  53         }
  54         if messages[0].ID != id {
  55                 t.Fail()
  56         }
  57         if messages[0].Text != say {
  58                 t.Fail()
  59         }
  60         close(store.Get)
  61         close(store.Add)
  62 }
  63
  64 func TestMultipleListeners(t *testing.T) {
  65         id := "3"
  66         say := "This is your nine o'clock alarm call!"
  67         at := time.Now()
  68         var zero_time time.Time
  69         store := start_store()
  70         const num_clients = 13
  71         var messages_from_store [num_clients]chan []Message
  72         for i := 0; i < num_clients; i++ {
  73                 messages_from_store[i] = make(chan []Message, 1)
  74                 store.Get <- &StoreRequest{zero_time, messages_from_store[i]}
  75         }
  76         store.Add <- &Message{at, id, say}
  77         for i := 0; i < num_clients; i++ {
  78                 messages := <-messages_from_store[i]
  79                 if len(messages) != 1 {
  80                         t.FailNow()
  81                 }
  82                 if messages[0].Time != at {
  83                         t.Fail()
  84                 }
  85                 if messages[0].ID != id {
  86                         t.Fail()
  87                 }
  88                 if messages[0].Text != say {
  89                         t.Fail()
  90                 }
  91         }
  92         close(store.Get)
  93         close(store.Add)
  94 }
  95
  96 func parseDuration(s string) time.Duration {
  97         d, err := time.ParseDuration(s)
  98         if err != nil {
  99                 panic(err)
 100         }
 101         return d
 102 }
 103
 104 func atoi(s string) int {
 105         i, err := strconv.Atoi(s)
 106         if err != nil {
 107                 panic(err)
 108         }
 109         return i
 110 }
 111
 112 func TestPartialRetreive(t *testing.T) {
 113         start_speak_count := atoi(speak_count.String())
 114         id1 := "4"
 115         id2 := "5"
 116         id3 := "6"
 117         say1 := "No, no.....No, 'e's stunned!"
 118         say2 := "You stunned him, just as he was wakin' up!"
 119         say3 := "Norwegian Blues stun easily, major."
 120         base := time.Now()
 121         at1 := base.Add(parseDuration("-4m"))
 122         since := base.Add(parseDuration("-3m"))
 123         at2 := base.Add(parseDuration("-2m"))
 124         at3 := base.Add(parseDuration("-1m"))
 125         store := start_store()
 126         store.Add <- &Message{at1, id1, say1}
 127         store.Add <- &Message{at2, id2, say2}
 128         store.Add <- &Message{at3, id3, say3}
 129         for atoi(speak_count.String()) != start_speak_count+3 {
 130                 runtime.Gosched()
 131         }
 132         messages_from_store := make(chan []Message, 1)
 133         store.Get <- &StoreRequest{since, messages_from_store}
 134         messages := <-messages_from_store
 135         if len(messages) != 2 {
 136                 t.FailNow()
 137         }
 138         if messages[0].Time != at2 {
 139                 t.Fail()
 140         }
 141         if messages[0].ID != id2 {
 142                 t.Fail()
 143         }
 144         if messages[0].Text != say2 {
 145                 t.Fail()
 146         }
 147         if messages[1].Time != at3 {
 148                 t.Fail()
 149         }
 150         if messages[1].ID != id3 {
 151                 t.Fail()
 152         }
 153         if messages[1].Text != say3 {
 154                 t.Fail()
 155         }
 156         close(store.Get)
 157         close(store.Add)
 158 }
 159
 160 func TestPrecisePartialRetreive(t *testing.T) {
 161         start_speak_count := atoi(speak_count.String())
 162         id1 := "7"
 163         id2 := "8"
 164         id3 := "9"
 165         say1 := "Well, he's...he's, ah...probably pining for the fjords."
 166         say2 := "PININ' for the FJORDS?!?!?!?"
 167         say3 := "look, why did he fall flat on his back the moment I got 'im home?"
 168         base := time.Now()
 169         at1 := base.Add(parseDuration("-3m"))
 170         at2 := base.Add(parseDuration("-2m"))
 171         at3 := base.Add(parseDuration("-1m"))
 172         since := at2
 173         store := start_store()
 174         store.Add <- &Message{at1, id1, say1}
 175         store.Add <- &Message{at2, id2, say2}
 176         store.Add <- &Message{at3, id3, say3}
 177         for atoi(speak_count.String()) != start_speak_count+3 {
 178                 runtime.Gosched()
 179         }
 180         messages_from_store := make(chan []Message, 1)
 181         store.Get <- &StoreRequest{since, messages_from_store}
 182         messages := <-messages_from_store
 183         if len(messages) != 1 {
 184                 t.FailNow()
 185         }
 186         if messages[0].Time != at3 {
 187                 t.Fail()
 188         }
 189         if messages[0].ID != id3 {
 190                 t.Fail()
 191         }
 192         if messages[0].Text != say3 {
 193                 t.Fail()
 194         }
 195         close(store.Get)
 196         close(store.Add)
 197 }
 198
 199 func TestTypicalFlow(t *testing.T) {
 200         id1 := "10"
 201         id2 := "11"
 202         say1 := "The Norwegian Blue prefers kippin' on it's back!"
 203         say2 := "Remarkable bird, innit, squire?  Lovely plumage!"
 204         store := start_store()
 205
 206         // A waiting zero-time fetch.
 207         var zero_time time.Time
 208         prev_fetch_wait_count := fetch_wait_count.String()
 209         fetch1 := make(chan []Message, 1)
 210         store.Get <- &StoreRequest{zero_time, fetch1}
 211         for prev_fetch_wait_count == fetch_wait_count.String() {
 212                 runtime.Gosched()
 213         }
 214
 215         // Someone speaks.  This triggers delivery.
 216         at1 := time.Now()
 217         store.Add <- &Message{at1, id1, say1}
 218         messages1 := <-fetch1
 219         if len(messages1) != 1 {
 220                 t.FailNow()
 221         }
 222         if messages1[0].Time != at1 {
 223                 t.Fail()
 224         }
 225         if messages1[0].ID != id1 {
 226                 t.Fail()
 227         }
 228         if messages1[0].Text != say1 {
 229                 t.Fail()
 230         }
 231
 232         // Upon recipt, client blocks on fetch with since=at1
 233         prev_fetch_wait_count = fetch_wait_count.String()
 234         fetch2 := make(chan []Message, 1)
 235         store.Get <- &StoreRequest{at1, fetch2}
 236         for prev_fetch_wait_count == fetch_wait_count.String() {
 237                 runtime.Gosched()
 238         }
 239
 240         // Someone speaks again.  This triggers another delivery.
 241         at2 := time.Now()
 242         if !at2.After(at1) {
 243                 t.Fail()
 244         }
 245         store.Add <- &Message{at2, id2, say2}
 246         messages2 := <-fetch2
 247         if len(messages2) != 1 {
 248                 t.FailNow()
 249         }
 250         if messages2[0].Time != at2 {
 251                 t.Fail()
 252         }
 253         if messages2[0].ID != id2 {
 254                 t.Fail()
 255         }
 256         if messages2[0].Text != say2 {
 257                 t.Fail()
 258         }
 259
 260         close(store.Get)
 261         close(store.Add)
 262 }