thrempp/component/threema/send_test.go

package threema

import (
	"testing"

	"github.com/o3ma/o3"
	"github.com/stretchr/testify/assert"
	"gosrc.io/xmpp"
)

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

	send := make(chan o3.Message)
	a := Account{
		Session:      o3.NewSessionContext(o3.ThreemaID{ID: o3.NewIDString("43218765")}),
		send:         send,
		deliveredMSG: make(map[uint64]string),
		readedMSG:    make(map[uint64]string),
	}

	go func() {
		a.Send("a", xmpp.Message{
			PacketAttrs: xmpp.PacketAttrs{From: "a@example.org"},
			Body:        "ohz8kai0ohNgohth",
		})
	}()
	p := <-send
	assert.NotNil(p)
	msg := p.(o3.TextMessage)
	assert.Contains(msg.Text(), "ohz8kai0ohNgohth")

	// test error
	err := a.Send("a", xmpp.Message{
		PacketAttrs: xmpp.PacketAttrs{From: "a@example.org"},
		Extensions: []xmpp.MsgExtension{
			xmpp.ReceiptReceived{ID: "blub"},
		},
	})
	assert.Error(err)
}

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

	a := Account{
		Session: o3.NewSessionContext(o3.ThreemaID{ID: o3.NewIDString("43218765")}),
	}

	// test error - threema send only int ids
	msg, err := a.sending("a", xmpp.Message{
		PacketAttrs: xmpp.PacketAttrs{From: "a@example.org"},
		Extensions: []xmpp.MsgExtension{
			xmpp.ReceiptReceived{ID: "blub"},
		},
	})
	assert.Error(err)
	assert.Nil(msg)

	// test delivered
	msg, err = a.sending("a", xmpp.Message{
		PacketAttrs: xmpp.PacketAttrs{From: "a@example.org"},
		Extensions: []xmpp.MsgExtension{
			xmpp.MarkReceived{ID: "3"},
		},
	})
	assert.NoError(err)
	drm, ok := msg.(o3.DeliveryReceiptMessage)
	assert.True(ok)
	assert.Equal(o3.MSGDELIVERED, drm.Status())

	// test read
	msg, err = a.sending("a", xmpp.Message{
		PacketAttrs: xmpp.PacketAttrs{From: "a@example.org"},
		Extensions: []xmpp.MsgExtension{
			xmpp.MarkDisplayed{ID: "5"},
		},
	})
	assert.NoError(err)
	drm, ok = msg.(o3.DeliveryReceiptMessage)
	assert.True(ok)
	assert.Equal(o3.MSGREAD, drm.Status())
}