86 lines
2.0 KiB
Go
86 lines
2.0 KiB
Go
package chat
|
|
|
|
import (
|
|
"github.com/ethereum/go-ethereum/crypto"
|
|
"github.com/stretchr/testify/suite"
|
|
"os"
|
|
"testing"
|
|
)
|
|
|
|
func TestEncryptionServiceMultiDeviceSuite(t *testing.T) {
|
|
suite.Run(t, new(EncryptionServiceMultiDeviceSuite))
|
|
}
|
|
|
|
type EncryptionServiceMultiDeviceSuite struct {
|
|
suite.Suite
|
|
alice1 *EncryptionService
|
|
bob1 *EncryptionService
|
|
alice2 *EncryptionService
|
|
bob2 *EncryptionService
|
|
}
|
|
|
|
func (s *EncryptionServiceMultiDeviceSuite) SetupTest() {
|
|
const (
|
|
aliceDBPath1 = "/tmp/alice1.db"
|
|
aliceDBKey1 = "alice1"
|
|
aliceDBPath2 = "/tmp/alice2.db"
|
|
aliceDBKey2 = "alice2"
|
|
bobDBPath1 = "/tmp/bob1.db"
|
|
bobDBKey1 = "bob1"
|
|
bobDBPath2 = "/tmp/bob2.db"
|
|
bobDBKey2 = "bob2"
|
|
)
|
|
|
|
os.Remove(aliceDBPath1)
|
|
os.Remove(bobDBPath1)
|
|
os.Remove(aliceDBPath2)
|
|
os.Remove(bobDBPath2)
|
|
|
|
alicePersistence1, err := NewSQLLitePersistence(aliceDBPath1, aliceDBKey1)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
alicePersistence2, err := NewSQLLitePersistence(aliceDBPath2, aliceDBKey2)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
bobPersistence1, err := NewSQLLitePersistence(bobDBPath1, bobDBKey1)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
bobPersistence2, err := NewSQLLitePersistence(bobDBPath2, bobDBKey2)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
s.alice1 = NewEncryptionService(alicePersistence1, "alice1")
|
|
s.bob1 = NewEncryptionService(bobPersistence1, "bob1")
|
|
|
|
s.alice2 = NewEncryptionService(alicePersistence2, "alice2")
|
|
s.bob2 = NewEncryptionService(bobPersistence2, "bob2")
|
|
|
|
}
|
|
|
|
func (s *EncryptionServiceMultiDeviceSuite) TestProcessPublicBundle() {
|
|
aliceKey, err := crypto.GenerateKey()
|
|
s.Require().NoError(err)
|
|
|
|
_, err = s.alice1.CreateBundle(aliceKey)
|
|
s.Require().NoError(err)
|
|
|
|
alice2Bundle, err := s.alice2.CreateBundle(aliceKey)
|
|
s.Require().NoError(err)
|
|
|
|
err = s.alice1.ProcessPublicBundle(aliceKey, alice2Bundle)
|
|
s.Require().NoError(err)
|
|
|
|
alice1MergedBundle1, err := s.alice1.CreateBundle(aliceKey)
|
|
s.Require().NoError(err)
|
|
|
|
s.Require().NotNil(alice1MergedBundle1.GetSignedPreKeys()["alice1"])
|
|
s.Require().NotNil(alice1MergedBundle1.GetSignedPreKeys()["alice2"])
|
|
}
|