128 lines
2.5 KiB
Go
128 lines
2.5 KiB
Go
package faye_test
|
|
|
|
import (
|
|
"github.com/pkg/errors"
|
|
"github.com/thesyncim/faye"
|
|
"github.com/thesyncim/faye/extensions"
|
|
"github.com/thesyncim/faye/message"
|
|
"log"
|
|
"os"
|
|
"os/exec"
|
|
"sync"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
type cancelFn func()
|
|
|
|
var unauthorizedErr = errors.New("500::unauthorized channel")
|
|
|
|
func setup(t *testing.T) (cancelFn, error) {
|
|
cmd := exec.Command("npm", "start")
|
|
cmd.Stdout = os.Stdout
|
|
cmd.Stderr = os.Stderr
|
|
|
|
err := cmd.Start()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
var cancel = func() {
|
|
cmd.Process.Signal(os.Kill)
|
|
log.Println("canceled")
|
|
}
|
|
go func() {
|
|
select {
|
|
case <-time.After(time.Second * 30):
|
|
cancel()
|
|
t.Fatal("test failed")
|
|
os.Exit(1)
|
|
|
|
}
|
|
}()
|
|
|
|
return cancel, nil
|
|
}
|
|
|
|
func TestServerSubscribeAndPublish10Messages(t *testing.T) {
|
|
shutdown, err := setup(t)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer shutdown()
|
|
|
|
debug := extensions.NewDebugExtension(os.Stdout)
|
|
|
|
client, err := fayec.NewClient("ws://localhost:8000/faye", fayec.WithExtension(debug.InExtension, debug.OutExtension))
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
var delivered int
|
|
var done sync.WaitGroup
|
|
done.Add(10)
|
|
client.OnPublishResponse("/test", func(message *message.Message) {
|
|
if !message.Successful {
|
|
t.Fatalf("failed to send message with id %s", message.Id)
|
|
}
|
|
delivered++
|
|
done.Done()
|
|
})
|
|
|
|
go func() {
|
|
client.Subscribe("/test", func(data message.Data) {
|
|
if data != "hello world" {
|
|
t.Fatalf("expecting: `hello world` got : %s", data)
|
|
}
|
|
})
|
|
}()
|
|
|
|
//give some time for setup
|
|
time.Sleep(time.Second)
|
|
for i := 0; i < 10; i++ {
|
|
id, err := client.Publish("/test", "hello world")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
log.Println(id, i)
|
|
}
|
|
|
|
done.Wait()
|
|
err = client.Unsubscribe("/test")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
//try to publish one more message
|
|
id, err := client.Publish("/test", "hello world")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
log.Println(id)
|
|
if delivered != 10 {
|
|
t.Fatal("message received after client unsubscribe")
|
|
}
|
|
}
|
|
|
|
func TestSubscribeUnauthorizedChannel(t *testing.T) {
|
|
shutdown, err := setup(t)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer shutdown()
|
|
|
|
debug := extensions.NewDebugExtension(os.Stdout)
|
|
|
|
client, err := fayec.NewClient("ws://localhost:8000/faye", fayec.WithExtension(debug.InExtension, debug.OutExtension))
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
err = client.Subscribe("/unauthorized", func(data message.Data) {
|
|
t.Fatal("received message on unauthorized channel")
|
|
})
|
|
if err.Error() != unauthorizedErr.Error() {
|
|
t.Fatalf("expecting `500::unauthorized channel` got : `%s`", err.Error())
|
|
}
|
|
|
|
}
|