factory test

x/wasm/keeper/factory_test.go

@@ -0,0 +1,44 @@
+package keeper
+
+import (
+	"encoding/json"
+	"testing"
+
+	sdk "github.com/cosmos/cosmos-sdk/types"
+	"github.com/stretchr/testify/require"
+)
+
+type FactoryInitMsg struct {
+	// Contract can instantiate itself through a boolean flag
+	Factory bool `json:"factory"`
+	// CodeID is the child contract to instantiate
+	CodeID uint64 `json:"code_id"`
+
+	Controller bool `json:"controller"`
+}
+
+func (m FactoryInitMsg) GetBytes(t testing.TB) []byte {
+	initMsgBz, err := json.Marshal(m)
+	require.NoError(t, err)
+	return initMsgBz
+}
+
+func TestFactoryContracts(t *testing.T) {
+	ctx, keepers := CreateTestInput(t, false, AvailableCapabilities)
+	contract := StoreFactoryContract(t, ctx, keepers)
+	// contract2 := StoreFactoryContract(t, ctx, keepers)
+
+	initMsgBz := FactoryInitMsg{
+		Factory:    true,
+		CodeID:     contract.CodeID,
+		Controller: false,
+	}.GetBytes(t)
+	initialAmount := sdk.NewCoins(sdk.NewInt64Coin("denom", 100))
+
+	adminAddr := contract.CreatorAddr
+	label := "demo contract to query"
+	_, _, err := keepers.ContractKeeper.Instantiate(ctx, contract.CodeID,
+		contract.CreatorAddr, adminAddr, initMsgBz, label, initialAmount)
+	require.NoError(t, err)
+
+}

x/wasm/keeper/test_common.go

@@ -5,7 +5,11 @@ import (
 	"encoding/binary"
 	"encoding/json"
 	"fmt"
+	"io"
+	"net/http"
 	"os"
+	"strings"
+	"sync"
 	"testing"
 	"time"
 
@@ -560,12 +564,16 @@ func StoreReflectContract(t testing.TB, ctx sdk.Context, keepers TestKeepers) Ex
 	return StoreExampleContract(t, ctx, keepers, "./testdata/reflect.wasm")
 }
 
+func StoreFactoryContract(t testing.TB, ctx sdk.Context, keepers TestKeepers) ExampleContract {
+	return StoreExampleContract(t, ctx, keepers, "https://github.com/jhernandezb/factory-test/releases/download/v0.1.0/factory_test.wasm")
+}
+
 func StoreExampleContract(t testing.TB, ctx sdk.Context, keepers TestKeepers, wasmFile string) ExampleContract {
 	anyAmount := sdk.NewCoins(sdk.NewInt64Coin("denom", 1000))
 	creator, _, creatorAddr := keyPubAddr()
 	fundAccounts(t, ctx, keepers.AccountKeeper, keepers.BankKeeper, creatorAddr, anyAmount)
 
-	wasmCode, err := os.ReadFile(wasmFile)
+	wasmCode, err := GetContractBytes(wasmFile)
 	require.NoError(t, err)
 
 	codeID, _, err := keepers.ContractKeeper.Create(ctx, creatorAddr, wasmCode, nil)
@@ -758,3 +766,40 @@ func keyPubAddr() (crypto.PrivKey, crypto.PubKey, sdk.AccAddress) {
 	addr := sdk.AccAddress(pub.Address())
 	return key, pub, addr
 }
+
+type storeCache struct {
+	sync.Mutex
+	contracts map[string][]byte
+}
+
+var contractsCache = storeCache{contracts: make(map[string][]byte)}
+
+func GetContractBytes(contract string) ([]byte, error) {
+	contractsCache.Lock()
+	bz, found := contractsCache.contracts[contract]
+	contractsCache.Unlock()
+	if found {
+		return bz, nil
+	}
+	contractsCache.Lock()
+	defer contractsCache.Unlock()
+	if strings.HasPrefix(contract, "https://") {
+		resp, err := http.Get(contract)
+		if err != nil {
+			return nil, err
+		}
+		defer resp.Body.Close()
+		bz, err = io.ReadAll(resp.Body)
+		if err != nil {
+			return nil, err
+		}
+	} else {
+		var err error
+		bz, err = os.ReadFile(contract)
+		if err != nil {
+			return nil, err
+		}
+	}
+	contractsCache.contracts[contract] = bz
+	return bz, nil
+}