mirror of
https://github.com/binwiederhier/ntfy.git
synced 2024-12-21 07:52:14 +03:00
118 lines
3.2 KiB
Go
118 lines
3.2 KiB
Go
package client_test
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/stretchr/testify/require"
|
|
"heckel.io/ntfy/client"
|
|
"heckel.io/ntfy/log"
|
|
"heckel.io/ntfy/test"
|
|
"os"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func TestMain(m *testing.M) {
|
|
log.SetLevel(log.ErrorLevel)
|
|
os.Exit(m.Run())
|
|
}
|
|
|
|
func TestClient_Publish_Subscribe(t *testing.T) {
|
|
s, port := test.StartServer(t)
|
|
defer test.StopServer(t, s, port)
|
|
c := client.New(newTestConfig(port))
|
|
|
|
subscriptionID, _ := c.Subscribe("mytopic")
|
|
time.Sleep(time.Second)
|
|
|
|
msg, err := c.Publish("mytopic", "some message")
|
|
require.Nil(t, err)
|
|
require.Equal(t, "some message", msg.Message)
|
|
|
|
msg, err = c.Publish("mytopic", "some other message",
|
|
client.WithTitle("some title"),
|
|
client.WithPriority("high"),
|
|
client.WithTags([]string{"tag1", "tag 2"}))
|
|
require.Nil(t, err)
|
|
require.Equal(t, "some other message", msg.Message)
|
|
require.Equal(t, "some title", msg.Title)
|
|
require.Equal(t, []string{"tag1", "tag 2"}, msg.Tags)
|
|
require.Equal(t, 4, msg.Priority)
|
|
|
|
msg, err = c.Publish("mytopic", "some delayed message",
|
|
client.WithDelay("25 hours"))
|
|
require.Nil(t, err)
|
|
require.Equal(t, "some delayed message", msg.Message)
|
|
require.True(t, time.Now().Add(24*time.Hour).Unix() < msg.Time)
|
|
|
|
time.Sleep(200 * time.Millisecond)
|
|
|
|
msg = nextMessage(c)
|
|
require.NotNil(t, msg)
|
|
require.Equal(t, "some message", msg.Message)
|
|
|
|
msg = nextMessage(c)
|
|
require.NotNil(t, msg)
|
|
require.Equal(t, "some other message", msg.Message)
|
|
require.Equal(t, "some title", msg.Title)
|
|
require.Equal(t, []string{"tag1", "tag 2"}, msg.Tags)
|
|
require.Equal(t, 4, msg.Priority)
|
|
|
|
msg = nextMessage(c)
|
|
require.Nil(t, msg)
|
|
|
|
c.Unsubscribe(subscriptionID)
|
|
time.Sleep(200 * time.Millisecond)
|
|
|
|
msg, err = c.Publish("mytopic", "a message that won't be received")
|
|
require.Nil(t, err)
|
|
require.Equal(t, "a message that won't be received", msg.Message)
|
|
|
|
msg = nextMessage(c)
|
|
require.Nil(t, msg)
|
|
}
|
|
|
|
func TestClient_Publish_Poll(t *testing.T) {
|
|
s, port := test.StartServer(t)
|
|
defer test.StopServer(t, s, port)
|
|
c := client.New(newTestConfig(port))
|
|
|
|
msg, err := c.Publish("mytopic", "some message", client.WithNoFirebase(), client.WithTagsList("tag1,tag2"))
|
|
require.Nil(t, err)
|
|
require.Equal(t, "some message", msg.Message)
|
|
require.Equal(t, []string{"tag1", "tag2"}, msg.Tags)
|
|
|
|
msg, err = c.Publish("mytopic", "this won't be cached", client.WithNoCache())
|
|
require.Nil(t, err)
|
|
require.Equal(t, "this won't be cached", msg.Message)
|
|
|
|
msg, err = c.Publish("mytopic", "some delayed message", client.WithDelay("20 min"))
|
|
require.Nil(t, err)
|
|
require.Equal(t, "some delayed message", msg.Message)
|
|
|
|
messages, err := c.Poll("mytopic")
|
|
require.Nil(t, err)
|
|
require.Equal(t, 1, len(messages))
|
|
require.Equal(t, "some message", messages[0].Message)
|
|
|
|
messages, err = c.Poll("mytopic", client.WithScheduled())
|
|
require.Nil(t, err)
|
|
require.Equal(t, 2, len(messages))
|
|
require.Equal(t, "some message", messages[0].Message)
|
|
require.Equal(t, "some delayed message", messages[1].Message)
|
|
}
|
|
|
|
func newTestConfig(port int) *client.Config {
|
|
c := client.NewConfig()
|
|
c.DefaultHost = fmt.Sprintf("http://127.0.0.1:%d", port)
|
|
return c
|
|
}
|
|
|
|
func nextMessage(c *client.Client) *client.Message {
|
|
select {
|
|
case m := <-c.Messages:
|
|
return m
|
|
default:
|
|
return nil
|
|
}
|
|
}
|