mirror of
https://github.com/status-im/status-go.git
synced 2025-02-23 04:08:27 +00:00
2df9df10ab
interface for initializing db, which is implemented for appdatabase and walletdatabase. TBD for multiaccounts DB. Unified DB initializion for all tests using helpers and new interface. Reduced sqlcipher kdf iterations for all tests to 1.
122 lines
3.6 KiB
Go
122 lines
3.6 KiB
Go
package collectibles
|
|
|
|
import (
|
|
"database/sql"
|
|
"math/big"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/suite"
|
|
|
|
"github.com/status-im/status-go/protocol/communities/token"
|
|
"github.com/status-im/status-go/protocol/protobuf"
|
|
"github.com/status-im/status-go/protocol/sqlite"
|
|
"github.com/status-im/status-go/services/wallet/bigint"
|
|
"github.com/status-im/status-go/t/helpers"
|
|
"github.com/status-im/status-go/walletdatabase"
|
|
)
|
|
|
|
func TestDatabaseSuite(t *testing.T) {
|
|
suite.Run(t, new(DatabaseSuite))
|
|
}
|
|
|
|
type DatabaseSuite struct {
|
|
suite.Suite
|
|
|
|
db *Database
|
|
}
|
|
|
|
func (s *DatabaseSuite) addCommunityToken(db *sql.DB, token *token.CommunityToken) error {
|
|
_, err := db.Exec(`INSERT INTO community_tokens (community_id, address, type, name, symbol, description, supply_str,
|
|
infinite_supply, transferable, remote_self_destruct, chain_id, deploy_state, image_base64, decimals, deployer, privileges_level)
|
|
VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`, token.CommunityID, token.Address, token.TokenType, token.Name,
|
|
token.Symbol, token.Description, token.Supply.String(), token.InfiniteSupply, token.Transferable, token.RemoteSelfDestruct,
|
|
token.ChainID, token.DeployState, token.Base64Image, token.Decimals, token.Deployer, token.PrivilegesLevel)
|
|
return err
|
|
}
|
|
|
|
func (s *DatabaseSuite) setupDatabase(db *sql.DB) error {
|
|
token721 := &token.CommunityToken{
|
|
CommunityID: "123",
|
|
TokenType: protobuf.CommunityTokenType_ERC721,
|
|
Address: "0x123",
|
|
Name: "StatusToken",
|
|
Symbol: "STT",
|
|
Description: "desc",
|
|
Supply: &bigint.BigInt{Int: big.NewInt(123)},
|
|
InfiniteSupply: false,
|
|
Transferable: true,
|
|
RemoteSelfDestruct: true,
|
|
ChainID: 1,
|
|
DeployState: token.InProgress,
|
|
Base64Image: "ABCD",
|
|
Deployer: "0xDEP1",
|
|
PrivilegesLevel: token.OwnerLevel,
|
|
}
|
|
|
|
token20 := &token.CommunityToken{
|
|
CommunityID: "345",
|
|
TokenType: protobuf.CommunityTokenType_ERC20,
|
|
Address: "0x345",
|
|
Name: "StatusToken",
|
|
Symbol: "STT",
|
|
Description: "desc",
|
|
Supply: &bigint.BigInt{Int: big.NewInt(345)},
|
|
InfiniteSupply: false,
|
|
Transferable: true,
|
|
RemoteSelfDestruct: true,
|
|
ChainID: 2,
|
|
DeployState: token.Failed,
|
|
Base64Image: "QWERTY",
|
|
Decimals: 21,
|
|
Deployer: "0xDEP2",
|
|
PrivilegesLevel: token.CommunityLevel,
|
|
}
|
|
|
|
err := s.addCommunityToken(db, token721)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return s.addCommunityToken(db, token20)
|
|
}
|
|
|
|
func (s *DatabaseSuite) SetupTest() {
|
|
s.db = nil
|
|
|
|
db, err := helpers.SetupTestMemorySQLDB(walletdatabase.DbInitializer{})
|
|
s.NoError(err, "creating sqlite db instance")
|
|
|
|
err = sqlite.Migrate(db)
|
|
s.NoError(err, "protocol migrate")
|
|
|
|
s.db = &Database{db: db}
|
|
|
|
err = s.setupDatabase(db)
|
|
s.NoError(err, "setting up database")
|
|
}
|
|
|
|
func (s *DatabaseSuite) TestGetTokenType() {
|
|
contractType, err := s.db.GetTokenType(1, "0x123")
|
|
s.Require().NoError(err)
|
|
s.Equal(contractType, protobuf.CommunityTokenType_ERC721)
|
|
|
|
contractType, err = s.db.GetTokenType(2, "0x345")
|
|
s.Require().NoError(err)
|
|
s.Equal(contractType, protobuf.CommunityTokenType_ERC20)
|
|
|
|
_, err = s.db.GetTokenType(10, "0x777")
|
|
s.Require().Error(err)
|
|
}
|
|
|
|
func (s *DatabaseSuite) TestGetPrivilegesLevel() {
|
|
privLevel, err := s.db.GetTokenPrivilegesLevel(1, "0x123")
|
|
s.Require().NoError(err)
|
|
s.Equal(privLevel, token.OwnerLevel)
|
|
|
|
privLevel, err = s.db.GetTokenPrivilegesLevel(2, "0x345")
|
|
s.Require().NoError(err)
|
|
s.Equal(privLevel, token.CommunityLevel)
|
|
|
|
_, err = s.db.GetTokenType(10, "0x777")
|
|
s.Require().Error(err)
|
|
}
|