178 lines
3.5 KiB
Go
178 lines
3.5 KiB
Go
package threema
|
|
|
|
import (
|
|
"errors"
|
|
"testing"
|
|
|
|
"github.com/o3ma/o3"
|
|
"github.com/stretchr/testify/assert"
|
|
"gosrc.io/xmpp"
|
|
)
|
|
|
|
const threemaID = "87654321"
|
|
|
|
var threemaIDByte o3.IDString
|
|
|
|
func init() {
|
|
threemaIDByte = o3.NewIDString(threemaID)
|
|
}
|
|
|
|
func createDummyAccount() Account {
|
|
a := Account{
|
|
deliveredMSG: make(map[uint64]string),
|
|
readedMSG: make(map[uint64]string),
|
|
}
|
|
a.TID = make([]byte, len(threemaIDByte))
|
|
copy(a.TID, threemaIDByte[:])
|
|
|
|
return a
|
|
}
|
|
|
|
func TestReceive(t *testing.T) {
|
|
assert := assert.New(t)
|
|
|
|
a := createDummyAccount()
|
|
|
|
// receiving/skip error
|
|
p, err := a.receiving(o3.ReceivedMsg{
|
|
Err: errors.New("dummy"),
|
|
})
|
|
assert.Nil(p)
|
|
assert.Error(err)
|
|
|
|
// nothing to receiving
|
|
p, err = a.receiving(o3.ReceivedMsg{})
|
|
assert.Nil(p)
|
|
assert.Error(err)
|
|
}
|
|
|
|
func TestReceiveText(t *testing.T) {
|
|
assert := assert.New(t)
|
|
|
|
a := createDummyAccount()
|
|
|
|
// receiving text
|
|
session := o3.SessionContext{
|
|
ID: o3.ThreemaID{
|
|
ID: o3.NewIDString("12345678"),
|
|
Nick: o3.NewPubNick("user"),
|
|
},
|
|
}
|
|
txtMsg, err := o3.NewTextMessage(&session, threemaID, "Oojoh0Ah")
|
|
assert.NoError(err)
|
|
p, err := a.receiving(o3.ReceivedMsg{
|
|
Msg: txtMsg,
|
|
})
|
|
assert.NoError(err)
|
|
xMSG, ok := p.(xmpp.Message)
|
|
assert.True(ok)
|
|
assert.Equal("Oojoh0Ah", xMSG.Body)
|
|
}
|
|
|
|
func TestReceiveAudio(t *testing.T) {
|
|
assert := assert.New(t)
|
|
|
|
a := createDummyAccount()
|
|
a.threema = &Threema{}
|
|
|
|
/* receiving image
|
|
session := o3.SessionContext{
|
|
ID: o3.ThreemaID{
|
|
ID: o3.NewIDString("12345678"),
|
|
Nick: o3.NewPubNick("user"),
|
|
},
|
|
}*/
|
|
dataMsg := o3.AudioMessage{}
|
|
_, err := a.receiving(o3.ReceivedMsg{
|
|
Msg: dataMsg,
|
|
})
|
|
assert.Error(err)
|
|
|
|
a.threema.httpUploadPath = "/tmp"
|
|
dataMsg = o3.AudioMessage{}
|
|
_, err = a.receiving(o3.ReceivedMsg{
|
|
Msg: dataMsg,
|
|
})
|
|
assert.Error(err)
|
|
}
|
|
func TestReceiveImage(t *testing.T) {
|
|
assert := assert.New(t)
|
|
|
|
a := createDummyAccount()
|
|
a.threema = &Threema{}
|
|
|
|
/* receiving image
|
|
session := o3.SessionContext{
|
|
ID: o3.ThreemaID{
|
|
ID: o3.NewIDString("12345678"),
|
|
Nick: o3.NewPubNick("user"),
|
|
},
|
|
}*/
|
|
imgMsg := o3.ImageMessage{}
|
|
_, err := a.receiving(o3.ReceivedMsg{
|
|
Msg: imgMsg,
|
|
})
|
|
assert.Error(err)
|
|
|
|
a.threema.httpUploadPath = "/tmp"
|
|
imgMsg = o3.ImageMessage{}
|
|
_, err = a.receiving(o3.ReceivedMsg{
|
|
Msg: imgMsg,
|
|
})
|
|
assert.Error(err)
|
|
}
|
|
|
|
func TestReceiveDeliveryReceipt(t *testing.T) {
|
|
assert := assert.New(t)
|
|
|
|
a := createDummyAccount()
|
|
|
|
// receiving delivered
|
|
session := o3.SessionContext{
|
|
ID: o3.ThreemaID{
|
|
ID: o3.NewIDString("12345678"),
|
|
Nick: o3.NewPubNick("user"),
|
|
},
|
|
}
|
|
msgID := o3.NewMsgID()
|
|
a.deliveredMSG[msgID] = "im4aeseeh1IbaQui"
|
|
a.readedMSG[msgID] = "im4aeseeh1IbaQui"
|
|
|
|
drm, err := o3.NewDeliveryReceiptMessage(&session, threemaID, msgID, o3.MSGDELIVERED)
|
|
assert.NoError(err)
|
|
p, err := a.receiving(o3.ReceivedMsg{
|
|
Msg: drm,
|
|
})
|
|
assert.NoError(err)
|
|
xMSG, ok := p.(xmpp.Message)
|
|
assert.True(ok)
|
|
rr := xMSG.Extensions[0].(xmpp.ReceiptReceived)
|
|
assert.Equal("im4aeseeh1IbaQui", rr.Id)
|
|
|
|
// receiving delivered -> not in cache
|
|
p, err = a.receiving(o3.ReceivedMsg{
|
|
Msg: drm,
|
|
})
|
|
assert.NoError(err)
|
|
assert.Nil(p)
|
|
|
|
// receiving readed
|
|
drm, err = o3.NewDeliveryReceiptMessage(&session, threemaID, msgID, o3.MSGREAD)
|
|
assert.NoError(err)
|
|
p, err = a.receiving(o3.ReceivedMsg{
|
|
Msg: drm,
|
|
})
|
|
assert.NoError(err)
|
|
xMSG, ok = p.(xmpp.Message)
|
|
assert.True(ok)
|
|
cmdd := xMSG.Extensions[0].(xmpp.ChatMarkerDisplayed)
|
|
assert.Equal("im4aeseeh1IbaQui", cmdd.Id)
|
|
|
|
// receiving delivered -> not in cache
|
|
p, err = a.receiving(o3.ReceivedMsg{
|
|
Msg: drm,
|
|
})
|
|
assert.NoError(err)
|
|
assert.Nil(p)
|
|
}
|