helper_test.go

package sns

import (
	"testing"

	"github.com/stretchr/testify/assert"
)

func TestComposeMessageGCM(t *testing.T) {
	assert := assert.New(t)

	opt := make(map[string]interface{})
	msg, err := composeMessageGCM("test", opt)
	assert.NoError(err)
	assert.Equal(`{"android":{"priority":"high"},"data":{"message":"test"}}`, msg)

	opt["sound"] = "jazz"
	msg, err = composeMessageGCM("test", opt)
	assert.NoError(err)
	assert.Equal(`{"android":{"priority":"high"},"data":{"message":"test","sound":"jazz"}}`, msg)

	delete(opt, "sound")
	opt["badge"] = 5
	msg, err = composeMessageGCM("test", opt)
	assert.NoError(err)
	assert.Equal(`{"android":{"priority":"high"},"data":{"badge":5,"message":"test"}}`, msg)

	opt["sound"] = "jazz"
	opt["badge"] = 5
	msg, err = composeMessageGCM("test", opt)
	assert.NoError(err)
	assert.Equal(`{"android":{"priority":"high"},"data":{"badge":5,"message":"test","sound":"jazz"}}`, msg)

	opt["x-option"] = "foo"
	msg, err = composeMessageGCM("test", opt)
	assert.NoError(err)
	assert.Equal(`{"android":{"priority":"high"},"data":{"badge":5,"message":"test","sound":"jazz","x-option":"foo"}}`, msg)
}

func TestComposeMessageAPNS(t *testing.T) {
	assert := assert.New(t)

	opt := make(map[string]interface{})
	msg, err := composeMessageAPNS("test", opt)
	assert.NoError(err)
	assert.Equal(`{"aps":{"alert":"test","sound":"default"}}`, msg)

	opt["sound"] = "jazz"
	msg, err = composeMessageAPNS("test", opt)
	assert.NoError(err)
	assert.Equal(`{"aps":{"alert":"test","sound":"jazz"}}`, msg)

	delete(opt, "sound")
	opt["badge"] = 5
	msg, err = composeMessageAPNS("test", opt)
	assert.NoError(err)
	assert.Equal(`{"aps":{"alert":"test","badge":5,"sound":"default"}}`, msg)

	delete(opt, "badge")
	opt["mutable-content"] = 1
	msg, err = composeMessageAPNS("test", opt)
	assert.NoError(err)
	assert.Equal(`{"aps":{"alert":"test","mutable-content":1,"sound":"default"}}`, msg)

	opt["sound"] = "jazz"
	opt["category"] = "new_message"
	opt["badge"] = 5
	opt["mutable-content"] = 1
	msg, err = composeMessageAPNS("test", opt)
	assert.NoError(err)
	assert.Equal(`{"aps":{"alert":"test","badge":5,"category":"new_message","mutable-content":1,"sound":"jazz"}}`, msg)

	opt["x-option"] = "foo"
	msg, err = composeMessageAPNS("test", opt)
	assert.NoError(err)
	assert.Equal(`{"aps":{"alert":"test","badge":5,"category":"new_message","mutable-content":1,"sound":"jazz"},"x-option":"foo"}`, msg)
}