diff --git a/x/registration/internal/keeper/genesis_test.go b/x/registration/internal/keeper/genesis_test.go index 3e300bfa1..6e514f326 100644 --- a/x/registration/internal/keeper/genesis_test.go +++ b/x/registration/internal/keeper/genesis_test.go @@ -10,9 +10,7 @@ import ( ) func TestInitGenesisNoMaster(t *testing.T) { - tempDir, err := os.MkdirTemp("", "wasm") - require.NoError(t, err) - defer os.RemoveAll(tempDir) + tempDir := t.TempDir() ctx, keeper := CreateTestInput(t, false, tempDir, true) data := types.GenesisState{ @@ -25,9 +23,7 @@ func TestInitGenesisNoMaster(t *testing.T) { } func TestInitGenesis(t *testing.T) { - tempDir, err := os.MkdirTemp("", "wasm") - require.NoError(t, err) - defer os.RemoveAll(tempDir) + tempDir := t.TempDir() ctx, keeper := CreateTestInput(t, false, tempDir, true) cert, err := os.ReadFile("../../testdata/attestation_cert_sw") @@ -46,9 +42,7 @@ func TestInitGenesis(t *testing.T) { } func TestExportGenesis(t *testing.T) { - tempDir, err := os.MkdirTemp("", "wasm") - require.NoError(t, err) - defer os.RemoveAll(tempDir) + tempDir := t.TempDir() ctx, keeper := CreateTestInput(t, false, tempDir, true) cert, err := os.ReadFile("../../testdata/attestation_cert_sw") diff --git a/x/registration/internal/keeper/keeper_test.go b/x/registration/internal/keeper/keeper_test.go index a65b5bfff..c3c9568f7 100644 --- a/x/registration/internal/keeper/keeper_test.go +++ b/x/registration/internal/keeper/keeper_test.go @@ -21,17 +21,13 @@ func init() { } func TestNewKeeper(t *testing.T) { - tempDir, err := os.MkdirTemp("", "reg") - require.NoError(t, err) - defer os.RemoveAll(tempDir) + tempDir := t.TempDir() _, regKeeper := CreateTestInput(t, false, tempDir, true) require.NotNil(t, regKeeper) } func TestNewKeeper_Node(t *testing.T) { - tempDir, err := os.MkdirTemp("", "reg") - require.NoError(t, err) - defer os.RemoveAll(tempDir) + tempDir := t.TempDir() seedPath := filepath.Join(tempDir, types.SecretNodeCfgFolder, types.SecretNodeSeedNewConfig) @@ -46,9 +42,7 @@ func TestNewKeeper_Node(t *testing.T) { } func TestKeeper_RegisterationStore(t *testing.T) { - tempDir, err := os.MkdirTemp("", "wasm") - require.NoError(t, err) - defer os.RemoveAll(tempDir) + tempDir := t.TempDir() ctx, regKeeper := CreateTestInput(t, false, tempDir, true) cert, err := os.ReadFile("../../testdata/attestation_cert_sw") @@ -72,9 +66,7 @@ func TestKeeper_RegisterationStore(t *testing.T) { } func TestKeeper_RegisterNode(t *testing.T) { - tempDir, err := os.MkdirTemp("", "wasm") - require.NoError(t, err) - defer os.RemoveAll(tempDir) + tempDir := t.TempDir() ctx, regKeeper := CreateTestInput(t, false, tempDir, true) cert, err := os.ReadFile("../../testdata/attestation_cert_sw.combined") diff --git a/x/registration/internal/keeper/querier_test.go b/x/registration/internal/keeper/querier_test.go index 338cb6fb5..56853f630 100644 --- a/x/registration/internal/keeper/querier_test.go +++ b/x/registration/internal/keeper/querier_test.go @@ -16,9 +16,7 @@ import ( // //// func TestNewQuerier_MasterKey(t *testing.T) { - tempDir, err := os.MkdirTemp("", "wasm") - require.NoError(t, err) - defer os.RemoveAll(tempDir) + tempDir := t.TempDir() ctx, keeper := CreateTestInput(t, false, tempDir, true) querier := NewQuerier(keeper) @@ -45,9 +43,7 @@ func TestNewQuerier_MasterKey(t *testing.T) { } func TestNewQuerier_MalformedNodeID(t *testing.T) { - tempDir, err := os.MkdirTemp("", "wasm") - require.NoError(t, err) - defer os.RemoveAll(tempDir) + tempDir := t.TempDir() ctx, keeper := CreateTestInput(t, false, tempDir, true) nodeIdInvalid := "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa" @@ -75,9 +71,7 @@ func TestNewQuerier_MalformedNodeID(t *testing.T) { } func TestNewQuerier_ValidNodeID(t *testing.T) { - tempDir, err := os.MkdirTemp("", "wasm") - require.NoError(t, err) - defer os.RemoveAll(tempDir) + tempDir := t.TempDir() ctx, keeper := CreateTestInput(t, false, tempDir, true) querier := NewQuerier(keeper) // TODO: Should test NewQuerier() as well