status-im/status-go

package encryption

import (
    "testing"

    dr "github.com/status-im/doubleratchet"
    "github.com/stretchr/testify/suite"

    "github.com/status-im/status-go/appdatabase"
    "github.com/status-im/status-go/protocol/sqlite"
    "github.com/status-im/status-go/t/helpers"
)

var (
    pubKey1 = dr.Key{0xe3, 0xbe, 0xb9, 0x4e, 0x70, 0x17, 0x37, 0xc, 0x1, 0x8f, 0xa9, 0x7e, 0xef, 0x4, 0xfb, 0x23, 0xac, 0xea, 0x28, 0xf7, 0xa9, 0x56, 0xcc, 0x1d, 0x46, 0xf3, 0xb5, 0x1d, 0x7d, 0x7d, 0x5e, 0x2c}
    pubKey2 = dr.Key{0xec, 0x8, 0x10, 0x7c, 0x33, 0x54, 0x0, 0x20, 0xe9, 0x4f, 0x6c, 0x84, 0xe4, 0x39, 0x50, 0x5a, 0x2f, 0x60, 0xbe, 0x81, 0xa, 0x78, 0x8b, 0xeb, 0x1e, 0x2c, 0x9, 0x8d, 0x4b, 0x4d, 0xc1, 0x40}
    mk1     = dr.Key{0x00, 0x8, 0x10, 0x7c, 0x33, 0x54, 0x0, 0x20, 0xe9, 0x4f, 0x6c, 0x84, 0xe4, 0x39, 0x50, 0x5a, 0x2f, 0x60, 0xbe, 0x81, 0xa, 0x78, 0x8b, 0xeb, 0x1e, 0x2c, 0x9, 0x8d, 0x4b, 0x4d, 0xc1, 0x40}
    mk2     = dr.Key{0x01, 0x8, 0x10, 0x7c, 0x33, 0x54, 0x0, 0x20, 0xe9, 0x4f, 0x6c, 0x84, 0xe4, 0x39, 0x50, 0x5a, 0x2f, 0x60, 0xbe, 0x81, 0xa, 0x78, 0x8b, 0xeb, 0x1e, 0x2c, 0x9, 0x8d, 0x4b, 0x4d, 0xc1, 0x40}
    mk3     = dr.Key{0x02, 0x8, 0x10, 0x7c, 0x33, 0x54, 0x0, 0x20, 0xe9, 0x4f, 0x6c, 0x84, 0xe4, 0x39, 0x50, 0x5a, 0x2f, 0x60, 0xbe, 0x81, 0xa, 0x78, 0x8b, 0xeb, 0x1e, 0x2c, 0x9, 0x8d, 0x4b, 0x4d, 0xc1, 0x40}
    mk4     = dr.Key{0x03, 0x8, 0x10, 0x7c, 0x33, 0x54, 0x0, 0x20, 0xe9, 0x4f, 0x6c, 0x84, 0xe4, 0x39, 0x50, 0x5a, 0x2f, 0x60, 0xbe, 0x81, 0xa, 0x78, 0x8b, 0xeb, 0x1e, 0x2c, 0x9, 0x8d, 0x4b, 0x4d, 0xc1, 0x40}
    mk5     = dr.Key{0x04, 0x8, 0x10, 0x7c, 0x33, 0x54, 0x0, 0x20, 0xe9, 0x4f, 0x6c, 0x84, 0xe4, 0x39, 0x50, 0x5a, 0x2f, 0x60, 0xbe, 0x81, 0xa, 0x78, 0x8b, 0xeb, 0x1e, 0x2c, 0x9, 0x8d, 0x4b, 0x4d, 0xc1, 0x40}
)

func TestSQLLitePersistenceKeysStorageTestSuite(t *testing.T) {
    suite.Run(t, new(SQLLitePersistenceKeysStorageTestSuite))
}

type SQLLitePersistenceKeysStorageTestSuite struct {
    suite.Suite
    service dr.KeysStorage
}

func (s *SQLLitePersistenceKeysStorageTestSuite) SetupTest() {
    db, err := helpers.SetupTestMemorySQLDB(appdatabase.DbInitializer{})
    s.Require().NoError(err)
    err = sqlite.Migrate(db)
    s.Require().NoError(err)

    p := newSQLitePersistence(db)
    s.service = p.KeysStorage()
}

func (s *SQLLitePersistenceKeysStorageTestSuite) TestKeysStorageSqlLiteGetMissing() {
    // Act.
    _, ok, err := s.service.Get(pubKey1, 0)

    // Assert.
    s.NoError(err)
    s.False(ok, "It returns false")
}

func (s *SQLLitePersistenceKeysStorageTestSuite) TestKeysStorageSqlLite_Put() {
    // Act and assert.
    err := s.service.Put([]byte("session-id"), pubKey1, 0, mk1, 1)
    s.NoError(err)
}

func (s *SQLLitePersistenceKeysStorageTestSuite) TestKeysStorageSqlLite_DeleteOldMks() {
    // Insert keys out-of-order
    err := s.service.Put([]byte("session-id"), pubKey1, 0, mk1, 1)
    s.NoError(err)
    err = s.service.Put([]byte("session-id"), pubKey1, 1, mk2, 2)
    s.NoError(err)
    err = s.service.Put([]byte("session-id"), pubKey1, 2, mk3, 20)
    s.NoError(err)
    err = s.service.Put([]byte("session-id"), pubKey1, 3, mk4, 21)
    s.NoError(err)
    err = s.service.Put([]byte("session-id"), pubKey1, 4, mk5, 22)
    s.NoError(err)

    err = s.service.DeleteOldMks([]byte("session-id"), 20)
    s.NoError(err)

    _, ok, err := s.service.Get(pubKey1, 0)
    s.NoError(err)
    s.False(ok)

    _, ok, err = s.service.Get(pubKey1, 1)
    s.NoError(err)
    s.False(ok)

    _, ok, err = s.service.Get(pubKey1, 2)
    s.NoError(err)
    s.False(ok)

    _, ok, err = s.service.Get(pubKey1, 3)
    s.NoError(err)
    s.True(ok)

    _, ok, err = s.service.Get(pubKey1, 4)
    s.NoError(err)
    s.True(ok)
}

func (s *SQLLitePersistenceKeysStorageTestSuite) TestKeysStorageSqlLite_TruncateMks() {
    // Insert keys out-of-order
    err := s.service.Put([]byte("session-id"), pubKey2, 2, mk5, 5)
    s.NoError(err)
    err = s.service.Put([]byte("session-id"), pubKey2, 0, mk3, 3)
    s.NoError(err)
    err = s.service.Put([]byte("session-id"), pubKey1, 1, mk2, 2)
    s.NoError(err)
    err = s.service.Put([]byte("session-id"), pubKey2, 1, mk4, 4)
    s.NoError(err)
    err = s.service.Put([]byte("session-id"), pubKey1, 0, mk1, 1)
    s.NoError(err)

    err = s.service.TruncateMks([]byte("session-id"), 2)
    s.NoError(err)

    _, ok, err := s.service.Get(pubKey1, 0)
    s.NoError(err)
    s.False(ok)

    _, ok, err = s.service.Get(pubKey1, 1)
    s.NoError(err)
    s.False(ok)

    _, ok, err = s.service.Get(pubKey2, 0)
    s.NoError(err)
    s.False(ok)

    _, ok, err = s.service.Get(pubKey2, 1)
    s.NoError(err)
    s.True(ok)

    _, ok, err = s.service.Get(pubKey2, 2)
    s.NoError(err)
    s.True(ok)
}

func (s *SQLLitePersistenceKeysStorageTestSuite) TestKeysStorageSqlLite_Count() {

    // Act.
    cnt, err := s.service.Count(pubKey1)

    // Assert.
    s.NoError(err)
    s.EqualValues(0, cnt, "It returns 0 when no keys are in the database")
}

func (s *SQLLitePersistenceKeysStorageTestSuite) TestKeysStorageSqlLite_Delete() {
    // Arrange.

    // Act and assert.
    err := s.service.DeleteMk(pubKey1, 0)
    s.NoError(err)
}

func (s *SQLLitePersistenceKeysStorageTestSuite) TestKeysStorageSqlLite_Flow() {

    // Act.
    err := s.service.Put([]byte("session-id"), pubKey1, 0, mk1, 1)
    s.NoError(err)

    k, ok, err := s.service.Get(pubKey1, 0)

    // Assert.
    s.NoError(err)
    s.True(ok, "It returns true")
    s.Equal(mk1, k, "It returns the message key")

    // Act.
    _, ok, err = s.service.Get(pubKey2, 0)

    // Assert.
    s.NoError(err)
    s.False(ok, "It returns false when querying non existing public key")

    // Act.
    _, ok, err = s.service.Get(pubKey1, 1)

    // Assert.
    s.NoError(err)
    s.False(ok, "It returns false when querying the wrong msg number")

    // Act.
    cnt, err := s.service.Count(pubKey1)

    // Assert.
    s.NoError(err)
    s.EqualValues(1, cnt)

    // Act and assert.
    err = s.service.DeleteMk(pubKey1, 1)
    s.NoError(err)

    // Act and assert.
    err = s.service.DeleteMk(pubKey2, 0)
    s.NoError(err)

    // Act.
    err = s.service.DeleteMk(pubKey1, 0)
    s.NoError(err)

    cnt, err = s.service.Count(pubKey1)

    // Assert.
    s.NoError(err)
    s.EqualValues(0, cnt)
}