2024-07-05 22:20:13 -04:00
|
|
|
package keeper_test
|
|
|
|
|
|
|
|
|
|
import (
|
|
|
|
|
"testing"
|
|
|
|
|
|
2024-09-05 18:43:54 -04:00
|
|
|
"cosmossdk.io/core/store"
|
2024-07-05 22:20:13 -04:00
|
|
|
"cosmossdk.io/log"
|
|
|
|
|
storetypes "cosmossdk.io/store/types"
|
2024-09-14 12:47:25 -04:00
|
|
|
nftkeeper "cosmossdk.io/x/nft/keeper"
|
2024-07-05 22:20:13 -04:00
|
|
|
"github.com/cosmos/cosmos-sdk/runtime"
|
|
|
|
|
"github.com/cosmos/cosmos-sdk/testutil"
|
|
|
|
|
simtestutil "github.com/cosmos/cosmos-sdk/testutil/sims"
|
|
|
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
|
|
|
moduletestutil "github.com/cosmos/cosmos-sdk/types/module/testutil"
|
|
|
|
|
authcodec "github.com/cosmos/cosmos-sdk/x/auth/codec"
|
|
|
|
|
authkeeper "github.com/cosmos/cosmos-sdk/x/auth/keeper"
|
|
|
|
|
authtypes "github.com/cosmos/cosmos-sdk/x/auth/types"
|
|
|
|
|
bankkeeper "github.com/cosmos/cosmos-sdk/x/bank/keeper"
|
|
|
|
|
govtypes "github.com/cosmos/cosmos-sdk/x/gov/types"
|
|
|
|
|
mintkeeper "github.com/cosmos/cosmos-sdk/x/mint/keeper"
|
|
|
|
|
minttypes "github.com/cosmos/cosmos-sdk/x/mint/types"
|
|
|
|
|
stakingkeeper "github.com/cosmos/cosmos-sdk/x/staking/keeper"
|
|
|
|
|
stakingtypes "github.com/cosmos/cosmos-sdk/x/staking/types"
|
2024-09-05 18:43:54 -04:00
|
|
|
"github.com/strangelove-ventures/poa"
|
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
|
"github.com/stretchr/testify/suite"
|
2024-07-05 22:20:13 -04:00
|
|
|
|
Feature/update dockerfile (#6)
* chore: remove unused new.Dockerfile
* feat: add DID model definitions
* fix: Fix EncodePublicKey method in KeyInfo struct
* feat: Update `EncodePublicKey` to be the inverse of `DecodePublicKey`
* refactor: update AssetInfo protobuf definition
* fix: update default assets with correct asset types
* fix: Initialize IPFS client and check for mounted directories
* feat: Improve IPFS client initialization and mount checking
* feat: Add local filesystem check for IPFS and IPNS
* fix: Use Unixfs().Get() instead of Cat() for IPFS and IPNS content retrieval
* feat: Update GetCID and GetIPNS functions to read data from IPFS node
* fix: Ensure IPFS client is initialized before pinning CID
* feat: Add AddFile and AddFolder methods
* feat: add IPFS file system abstraction
* feat: Implement IPFS file, location, and filesystem abstractions
* refactor: remove unused functions and types
* refactor: remove unused FileSystem interface
* feat: add initial wasm entrypoint
* feat: add basic vault command operations
* docs: add vault module features
* test: remove test for MsgUpdateParams
* refactor: Replace PrimaryKey with Property struct in zkprop.go
* feat: Update the `CreateWitness` and `CreateAccumulator` and `VerifyWitness` and `UpdateAccumulator` to Use the new `Accumulator` and `Witness` types. Then Clean up the code in the file and refactor the marshalling methods
* <no value>
* feat: add KeyCurve and KeyType to KeyInfo in genesis
* feat: add WASM build step to devbox.json
* feat: Add zkgate.go file
* feat: Uncomment and modify zkgate code to work with Property struct
* feat: Merge zkgate.go and zkprop.go logic
* feat: implement API endpoints for profile management
* refactor: remove unused template file
* feat(orm): remove unused ORM models
* feat: add persistent SQLite database support in WASM
* fix: Update module names in protobuf files
* feat: Add method to initialize SQLite database
* fix: update go-sqlite3 dependency to version 1.14.23
* feat: introduce database layer
* feat: Implement database layer for Vault node
* feature/update-dockerfile
* feat: Add keyshares table
* fix: Reorder the SQL statements in the tables.go file
* feat: Update the `createCredentialsTable` method to match the proper Credential struct
* feat: Update createProfilesTable and add createPropertiesTable
* feat: Add constant SQL queries to queries.go and use prepared statements in db.go
* feat: Add createKeysharesTable to internal/db/db.go
* feat: Update `createPermissionsTable` to match Permissions struct
* feat: Add database enum types
* feat: Add DIDNamespace and PermissionScope enums
* feat: Add DBConfig and DBOption types
* feat: Update the db implementation to use the provided go library
* fix: update db implementation to use go-sqlite3 v0.18.2
* fix: Refactor database connection and statement handling
* feat: Simplify db.go implementation
* feat: Convert constant SQL queries to functions in queries.go and update db.go to use prepared statements
* feat: Add models.go file with database table structs
* fix: Remove unused statement map and prepare statements
diff --git a/internal/db/db.go b/internal/db/db.go
index 201d09b..d4d4d4e 100644
--- a/internal/db/db.go
+++ b/internal/db/db.go
@@ -32,11 +32,6 @@ func Open(config *DBConfig) (*DB, error) {
Conn: conn,
}
- if err := createTables(db); err != nil {
- conn.Close()
- return nil, fmt.Errorf("failed to create tables: %w", err)
- }
-
return db, nil
}
@@ -61,114 +56,3 @@ func createTables(db *DB) error {
return nil
}
-// AddAccount adds a new account to the database
-func (db *DB) AddAccount(name, address string) error {
- return db.Exec(insertAccountQuery(name, address))
-}
-
-// AddAsset adds a new asset to the database
-func (db *DB) AddAsset(name, symbol string, decimals int, chainID int64) error {
- return db.Exec(insertAssetQuery(name, symbol, decimals, chainID))
-}
-
-// AddChain adds a new chain to the database
-func (db *DB) AddChain(name, networkID string) error {
- return db.Exec(insertChainQuery(name, networkID))
-}
-
-// AddCredential adds a new credential to the database
-func (db *DB) AddCredential(
- handle, controller, attestationType, origin string,
- credentialID, publicKey []byte,
- transport string,
- signCount uint32,
- userPresent, userVerified, backupEligible, backupState, cloneWarning bool,
-) error {
- return db.Exec(insertCredentialQuery(
- handle,
- controller,
- attestationType,
- origin,
- credentialID,
- publicKey,
- transport,
- signCount,
- userPresent,
- userVerified,
- backupEligible,
- backupState,
- cloneWarning,
- ))
-}
-
-// AddProfile adds a new profile to the database
-func (db *DB) AddProfile(
- id, subject, controller, originURI, publicMetadata, privateMetadata string,
-) error {
- return db.statements["insertProfile"].Exec(
- id, subject, controller, originURI, publicMetadata, privateMetadata,
- )
-}
-
-// AddProperty adds a new property to the database
-func (db *DB) AddProperty(
- profileID, key, accumulator, propertyKey string,
-) error {
- return db.statements["insertProperty"].Exec(
- profileID, key, accumulator, propertyKey,
- )
-}
-
-// AddPermission adds a new permission to the database
-func (db *DB) AddPermission(
- serviceID string,
- grants []DIDNamespace,
- scopes []PermissionScope,
-) error {
- grantsJSON, err := json.Marshal(grants)
- if err != nil {
- return fmt.Errorf("failed to marshal grants: %w", err)
- }
-
- scopesJSON, err := json.Marshal(scopes)
- if err != nil {
- return fmt.Errorf("failed to marshal scopes: %w", err)
- }
-
- return db.statements["insertPermission"].Exec(
- serviceID, string(grantsJSON), string(scopesJSON),
- )
-}
-
-// GetPermission retrieves the permission for the given service ID
-func (db *DB) GetPermission(serviceID string) ([]DIDNamespace, []PermissionScope, error) {
- row := db.statements["getPermission"].QueryRow(serviceID)
-
- var grantsJSON, scopesJSON string
- if err := row.Scan(&grantsJSON, &scopesJSON); err != nil {
- return nil, nil, fmt.Errorf("failed to get permission: %w", err)
- }
-
- var grants []DIDNamespace
- if err := json.Unmarshal([]byte(grantsJSON), &grants); err != nil {
- return nil, nil, fmt.Errorf("failed to unmarshal grants: %w", err)
- }
-
- var scopes []PermissionScope
- if err := json.Unmarshal([]byte(scopesJSON), &scopes); err != nil {
- return nil, nil, fmt.Errorf("failed to unmarshal scopes: %w", err)
- }
-
- return grants, scopes, nil
-}
-
-// Close closes the database connection and finalizes all prepared statements
-func (db *DB) Close() error {
- for _, stmt := range db.statements {
- stmt.Finalize()
- }
- return db.Conn.Close()
-}
diff --git a/internal/db/queries.go b/internal/db/queries.go
index 807d701..e69de29 100644
--- a/internal/db/queries.go
+++ b/internal/db/queries.go
@@ -1,79 +0,0 @@
-package db
-
-import "fmt"
-
-// Account queries
-func insertAccountQuery(name, address string) string {
- return fmt.Sprintf(`INSERT INTO accounts (name, address) VALUES (%s, %s)`, name, address)
-}
-
-// Asset queries
-func insertAssetQuery(name, symbol string, decimals int, chainID int64) string {
- return fmt.Sprintf(
- `INSERT INTO assets (name, symbol, decimals, chain_id) VALUES (%s, %s, %d, %d)`,
- name,
- symbol,
- decimals,
- chainID,
- )
-}
-
-// Chain queries
-func insertChainQuery(name string, networkID string) string {
- return fmt.Sprintf(`INSERT INTO chains (name, network_id) VALUES (%s, %d)`, name, networkID)
-}
-
-// Credential queries
-func insertCredentialQuery(
- handle, controller, attestationType, origin string,
- credentialID, publicKey []byte,
- transport string,
- signCount uint32,
- userPresent, userVerified, backupEligible, backupState, cloneWarning bool,
-) string {
- return fmt.Sprintf(`INSERT INTO credentials (
- handle, controller, attestation_type, origin,
- credential_id, public_key, transport, sign_count,
- user_present, user_verified, backup_eligible,
- backup_state, clone_warning
- ) VALUES (%s, %s, %s, %s, %s, %s, %s, %d, %t, %t, %t, %t, %t)`,
- handle, controller, attestationType, origin,
- credentialID, publicKey, transport, signCount,
- userPresent, userVerified, backupEligible,
- backupState, cloneWarning)
-}
-
-// Profile queries
-func insertProfileQuery(
- id, subject, controller, originURI, publicMetadata, privateMetadata string,
-) string {
- return fmt.Sprintf(`INSERT INTO profiles (
- id, subject, controller, origin_uri,
- public_metadata, private_metadata
- ) VALUES (%s, %s, %s, %s, %s, %s)`,
- id, subject, controller, originURI,
- publicMetadata, privateMetadata)
-}
-
-// Property queries
-func insertPropertyQuery(profileID, key, accumulator, propertyKey string) string {
- return fmt.Sprintf(`INSERT INTO properties (
- profile_id, key, accumulator, property_key
- ) VALUES (%s, %s, %s, %s)`,
- profileID, key, accumulator, propertyKey)
-}
-
-// Permission queries
-func insertPermissionQuery(serviceID, grants, scopes string) string {
- return fmt.Sprintf(
- `INSERT INTO permissions (service_id, grants, scopes) VALUES (%s, %s, %s)`,
- serviceID,
- grants,
- scopes,
- )
-}
-
-// GetPermission query
-func getPermissionQuery(serviceID string) string {
- return fmt.Sprintf(`SELECT grants, scopes FROM permissions WHERE service_id = %s`, serviceID)
-}
* fix: update Makefile to use sonrd instead of wasmd
* feat: Add targets for templ and vault in Makefile and use only make in devbox.json
* feat: add SQLite database support
* bump: version 0.6.0 → 0.7.0
* refactor: upgrade actions to latest versions
2024-09-05 01:24:57 -04:00
|
|
|
module "github.com/onsonr/sonr/x/did"
|
|
|
|
|
"github.com/onsonr/sonr/x/did/keeper"
|
|
|
|
|
"github.com/onsonr/sonr/x/did/types"
|
2024-07-05 22:20:13 -04:00
|
|
|
)
|
|
|
|
|
|
|
|
|
|
var maccPerms = map[string][]string{
|
|
|
|
|
authtypes.FeeCollectorName: nil,
|
|
|
|
|
stakingtypes.BondedPoolName: {authtypes.Burner, authtypes.Staking},
|
|
|
|
|
stakingtypes.NotBondedPoolName: {authtypes.Burner, authtypes.Staking},
|
|
|
|
|
minttypes.ModuleName: {authtypes.Minter},
|
|
|
|
|
govtypes.ModuleName: {authtypes.Burner},
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
type testFixture struct {
|
|
|
|
|
suite.Suite
|
|
|
|
|
|
|
|
|
|
ctx sdk.Context
|
|
|
|
|
k keeper.Keeper
|
|
|
|
|
msgServer types.MsgServer
|
|
|
|
|
queryServer types.QueryServer
|
|
|
|
|
appModule *module.AppModule
|
|
|
|
|
|
|
|
|
|
accountkeeper authkeeper.AccountKeeper
|
|
|
|
|
bankkeeper bankkeeper.BaseKeeper
|
2024-09-14 12:47:25 -04:00
|
|
|
nftKeeper nftkeeper.Keeper
|
2024-07-05 22:20:13 -04:00
|
|
|
stakingKeeper *stakingkeeper.Keeper
|
|
|
|
|
mintkeeper mintkeeper.Keeper
|
|
|
|
|
|
|
|
|
|
addrs []sdk.AccAddress
|
|
|
|
|
govModAddr string
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func SetupTest(t *testing.T) *testFixture {
|
|
|
|
|
t.Helper()
|
|
|
|
|
f := new(testFixture)
|
|
|
|
|
require := require.New(t)
|
|
|
|
|
|
|
|
|
|
// Base setup
|
|
|
|
|
logger := log.NewTestLogger(t)
|
|
|
|
|
encCfg := moduletestutil.MakeTestEncodingConfig()
|
|
|
|
|
|
|
|
|
|
f.govModAddr = authtypes.NewModuleAddress(govtypes.ModuleName).String()
|
|
|
|
|
f.addrs = simtestutil.CreateIncrementalAccounts(3)
|
|
|
|
|
|
|
|
|
|
key := storetypes.NewKVStoreKey(poa.ModuleName)
|
|
|
|
|
storeService := runtime.NewKVStoreService(key)
|
|
|
|
|
testCtx := testutil.DefaultContextWithDB(t, key, storetypes.NewTransientStoreKey("transient_test"))
|
|
|
|
|
|
|
|
|
|
f.ctx = testCtx.Ctx
|
|
|
|
|
|
|
|
|
|
// Register SDK modules.
|
|
|
|
|
registerBaseSDKModules(f, encCfg, storeService, logger, require)
|
|
|
|
|
|
|
|
|
|
// Setup POA Keeper.
|
2024-10-02 01:40:49 -04:00
|
|
|
f.k = keeper.NewKeeper(encCfg.Codec, storeService, logger, f.govModAddr, f.accountkeeper, f.nftKeeper, f.stakingKeeper)
|
2024-07-05 22:20:13 -04:00
|
|
|
f.msgServer = keeper.NewMsgServerImpl(f.k)
|
|
|
|
|
f.queryServer = keeper.NewQuerier(f.k)
|
2024-09-14 12:47:25 -04:00
|
|
|
f.appModule = module.NewAppModule(encCfg.Codec, f.k, f.nftKeeper)
|
2024-07-05 22:20:13 -04:00
|
|
|
|
|
|
|
|
return f
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func registerModuleInterfaces(encCfg moduletestutil.TestEncodingConfig) {
|
|
|
|
|
authtypes.RegisterInterfaces(encCfg.InterfaceRegistry)
|
|
|
|
|
stakingtypes.RegisterInterfaces(encCfg.InterfaceRegistry)
|
|
|
|
|
|
|
|
|
|
types.RegisterInterfaces(encCfg.InterfaceRegistry)
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func registerBaseSDKModules(
|
|
|
|
|
f *testFixture,
|
|
|
|
|
encCfg moduletestutil.TestEncodingConfig,
|
|
|
|
|
storeService store.KVStoreService,
|
|
|
|
|
logger log.Logger,
|
|
|
|
|
require *require.Assertions,
|
|
|
|
|
) {
|
|
|
|
|
registerModuleInterfaces(encCfg)
|
|
|
|
|
|
|
|
|
|
// Auth Keeper.
|
|
|
|
|
f.accountkeeper = authkeeper.NewAccountKeeper(
|
|
|
|
|
encCfg.Codec, storeService,
|
|
|
|
|
authtypes.ProtoBaseAccount,
|
|
|
|
|
maccPerms,
|
|
|
|
|
authcodec.NewBech32Codec(sdk.Bech32MainPrefix), sdk.Bech32MainPrefix,
|
|
|
|
|
f.govModAddr,
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
// Bank Keeper.
|
|
|
|
|
f.bankkeeper = bankkeeper.NewBaseKeeper(
|
|
|
|
|
encCfg.Codec, storeService,
|
|
|
|
|
f.accountkeeper,
|
|
|
|
|
nil,
|
|
|
|
|
f.govModAddr, logger,
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
// Staking Keeper.
|
|
|
|
|
f.stakingKeeper = stakingkeeper.NewKeeper(
|
|
|
|
|
encCfg.Codec, storeService,
|
|
|
|
|
f.accountkeeper, f.bankkeeper, f.govModAddr,
|
|
|
|
|
authcodec.NewBech32Codec(sdk.Bech32PrefixValAddr),
|
|
|
|
|
authcodec.NewBech32Codec(sdk.Bech32PrefixConsAddr),
|
|
|
|
|
)
|
|
|
|
|
require.NoError(f.stakingKeeper.SetParams(f.ctx, stakingtypes.DefaultParams()))
|
|
|
|
|
f.accountkeeper.SetModuleAccount(f.ctx, f.stakingKeeper.GetNotBondedPool(f.ctx))
|
|
|
|
|
f.accountkeeper.SetModuleAccount(f.ctx, f.stakingKeeper.GetBondedPool(f.ctx))
|
|
|
|
|
|
|
|
|
|
// Mint Keeper.
|
|
|
|
|
f.mintkeeper = mintkeeper.NewKeeper(
|
|
|
|
|
encCfg.Codec, storeService,
|
|
|
|
|
f.stakingKeeper, f.accountkeeper, f.bankkeeper,
|
|
|
|
|
authtypes.FeeCollectorName, f.govModAddr,
|
|
|
|
|
)
|
|
|
|
|
f.accountkeeper.SetModuleAccount(f.ctx, f.accountkeeper.GetModuleAccount(f.ctx, minttypes.ModuleName))
|
|
|
|
|
f.mintkeeper.InitGenesis(f.ctx, f.accountkeeper, minttypes.DefaultGenesisState())
|
|
|
|
|
}
|