Skip to content

Commit aefb96f

Browse files
Hakob23Hakob23
committed
fmt: fix
1 parent 9c8e07f commit aefb96f

File tree

2 files changed

+137
-137
lines changed

2 files changed

+137
-137
lines changed

‎cmd/utils/resign_utils/resign_utils_test.go‎

Lines changed: 69 additions & 69 deletions
Original file line numberDiff line numberDiff line change
@@ -36,85 +36,85 @@ func TestReadResignUtilsInput(t *testing.T) {
3636
})
3737
}
3838

39-
func TestParseResignUtilsInput(t *testing.T) {
39+
func TestParseResignUtilsInputRequiredFields(t *testing.T) {
4040
parse := func() {}
41+
envFile := writeTempEnvFile(t, "")
42+
t.Setenv("LPS_UTILS_ENV_FILE", envFile)
43+
44+
input := &ResignUtilsInput{
45+
BaseInput: scripts.BaseInput{
46+
Network: "",
47+
RskEndpoint: "",
48+
SecretSource: "",
49+
},
50+
Resign: true,
51+
}
52+
err := ParseResignUtilsInput(parse, input)
53+
require.Error(t, err)
54+
assert.Contains(t, err.Error(), "invalid input")
55+
}
4156

42-
t.Run("should validate required fields", func(t *testing.T) {
43-
envFile := writeTempEnvFile(t, "")
44-
t.Setenv("LPS_UTILS_ENV_FILE", envFile)
45-
46-
input := &ResignUtilsInput{
47-
BaseInput: scripts.BaseInput{
48-
Network: "",
49-
RskEndpoint: "",
50-
SecretSource: "",
51-
},
52-
Resign: true,
53-
}
54-
err := ParseResignUtilsInput(parse, input)
55-
require.Error(t, err)
56-
assert.Contains(t, err.Error(), "invalid input")
57-
})
58-
59-
t.Run("should reject missing action", func(t *testing.T) {
60-
envFile := writeTempEnvFile(t, "")
61-
t.Setenv("LPS_UTILS_ENV_FILE", envFile)
62-
63-
input := &ResignUtilsInput{
64-
BaseInput: scripts.BaseInput{
65-
Network: "regtest",
66-
RskEndpoint: "http://localhost:4444",
67-
SecretSource: "env",
68-
AwsLocalEndpoint: "http://localhost:4566",
69-
},
70-
}
71-
err := ParseResignUtilsInput(parse, input)
72-
require.Error(t, err)
73-
assert.Contains(t, err.Error(), "select exactly one action")
74-
})
57+
func TestParseResignUtilsInputMissingAction(t *testing.T) {
58+
parse := func() {}
59+
envFile := writeTempEnvFile(t, "")
60+
t.Setenv("LPS_UTILS_ENV_FILE", envFile)
61+
62+
input := &ResignUtilsInput{
63+
BaseInput: scripts.BaseInput{
64+
Network: "regtest",
65+
RskEndpoint: "http://localhost:4444",
66+
SecretSource: "env",
67+
AwsLocalEndpoint: "http://localhost:4566",
68+
},
69+
}
70+
err := ParseResignUtilsInput(parse, input)
71+
require.Error(t, err)
72+
assert.Contains(t, err.Error(), "select exactly one action")
73+
}
7574

76-
t.Run("should reject multiple actions", func(t *testing.T) {
77-
envFile := writeTempEnvFile(t, "")
78-
t.Setenv("LPS_UTILS_ENV_FILE", envFile)
79-
80-
input := &ResignUtilsInput{
81-
BaseInput: scripts.BaseInput{
82-
Network: "regtest",
83-
RskEndpoint: "http://localhost:4444",
84-
SecretSource: "env",
85-
AwsLocalEndpoint: "http://localhost:4566",
86-
},
87-
Resign: true,
88-
WithdrawCollateral: true,
89-
}
90-
err := ParseResignUtilsInput(parse, input)
91-
require.Error(t, err)
92-
assert.Contains(t, err.Error(), "select exactly one action")
93-
})
75+
func TestParseResignUtilsInputMultipleActions(t *testing.T) {
76+
parse := func() {}
77+
envFile := writeTempEnvFile(t, "")
78+
t.Setenv("LPS_UTILS_ENV_FILE", envFile)
79+
80+
input := &ResignUtilsInput{
81+
BaseInput: scripts.BaseInput{
82+
Network: "regtest",
83+
RskEndpoint: "http://localhost:4444",
84+
SecretSource: "env",
85+
AwsLocalEndpoint: "http://localhost:4566",
86+
},
87+
Resign: true,
88+
WithdrawCollateral: true,
89+
}
90+
err := ParseResignUtilsInput(parse, input)
91+
require.Error(t, err)
92+
assert.Contains(t, err.Error(), "select exactly one action")
93+
}
9494

95-
t.Run("should load defaults from env file", func(t *testing.T) {
96-
envFile, err := os.CreateTemp(t.TempDir(), "lps-utils-*.env")
97-
require.NoError(t, err)
98-
require.NoError(t, os.WriteFile(envFile.Name(), []byte(`
95+
func TestParseResignUtilsInputDefaultsFromEnvFile(t *testing.T) {
96+
parse := func() {}
97+
envFile, err := os.CreateTemp(t.TempDir(), "lps-utils-*.env")
98+
require.NoError(t, err)
99+
require.NoError(t, os.WriteFile(envFile.Name(), []byte(`
99100
LPS_STAGE=regtest
100101
RSK_ENDPOINT=http://localhost:4444
101102
AWS_LOCAL_ENDPOINT=http://localhost:4566
102103
SECRET_SRC=env
103104
WALLET_FILE=./keystore.json
104105
`), 0o600))
105-
t.Setenv("LPS_UTILS_ENV_FILE", envFile.Name())
106+
t.Setenv("LPS_UTILS_ENV_FILE", envFile.Name())
106107

107-
input := &ResignUtilsInput{
108-
BaseInput: scripts.BaseInput{},
109-
Resign: true,
110-
}
111-
err = ParseResignUtilsInput(parse, input)
112-
require.NoError(t, err)
113-
assert.Equal(t, "regtest", input.Network)
114-
assert.Equal(t, "http://localhost:4444", input.RskEndpoint)
115-
assert.Equal(t, "env", input.SecretSource)
116-
assert.Equal(t, "./keystore.json", input.KeystoreFile)
117-
})
108+
input := &ResignUtilsInput{
109+
BaseInput: scripts.BaseInput{},
110+
Resign: true,
111+
}
112+
err = ParseResignUtilsInput(parse, input)
113+
require.NoError(t, err)
114+
assert.Equal(t, "regtest", input.Network)
115+
assert.Equal(t, "http://localhost:4444", input.RskEndpoint)
116+
assert.Equal(t, "env", input.SecretSource)
117+
assert.Equal(t, "./keystore.json", input.KeystoreFile)
118118
}
119119

120120
func writeTempEnvFile(t *testing.T, contents string) string {
@@ -150,7 +150,7 @@ func TestExecuteWithdrawCollateral(t *testing.T) {
150150

151151
err := ExecuteWithdrawCollateral(contract)
152152
require.Error(t, err)
153-
assert.ErrorIs(t, err, liquidity_provider.ProviderNotResignedError)
153+
require.ErrorIs(t, err, liquidity_provider.ProviderNotResignedError)
154154
contract.AssertExpectations(t)
155155
})
156156
}

‎cmd/utils/withdraw/withdraw_test.go‎

Lines changed: 68 additions & 68 deletions
Original file line numberDiff line numberDiff line change
@@ -38,85 +38,85 @@ func TestReadWithdrawScriptInput(t *testing.T) {
3838
})
3939
}
4040

41-
func TestParseWithdrawScriptInput(t *testing.T) {
41+
func TestParseWithdrawScriptInputRequiredFields(t *testing.T) {
4242
parse := func() {}
43+
envFile := writeTempEnvFile(t, "")
44+
t.Setenv("LPS_UTILS_ENV_FILE", envFile)
45+
46+
input := &WithdrawScriptInput{
47+
BaseInput: scripts.BaseInput{
48+
Network: "",
49+
RskEndpoint: "",
50+
SecretSource: "",
51+
},
52+
Amount: "1000",
53+
}
54+
err := ParseWithdrawScriptInput(parse, input)
55+
require.Error(t, err)
56+
assert.Contains(t, err.Error(), "invalid input")
57+
}
4358

44-
t.Run("should validate required fields", func(t *testing.T) {
45-
envFile := writeTempEnvFile(t, "")
46-
t.Setenv("LPS_UTILS_ENV_FILE", envFile)
47-
48-
input := &WithdrawScriptInput{
49-
BaseInput: scripts.BaseInput{
50-
Network: "",
51-
RskEndpoint: "",
52-
SecretSource: "",
53-
},
54-
Amount: "1000",
55-
}
56-
err := ParseWithdrawScriptInput(parse, input)
57-
require.Error(t, err)
58-
assert.Contains(t, err.Error(), "invalid input")
59-
})
60-
61-
t.Run("should reject missing action", func(t *testing.T) {
62-
envFile := writeTempEnvFile(t, "")
63-
t.Setenv("LPS_UTILS_ENV_FILE", envFile)
64-
65-
input := &WithdrawScriptInput{
66-
BaseInput: scripts.BaseInput{
67-
Network: "regtest",
68-
RskEndpoint: "http://localhost:4444",
69-
SecretSource: "env",
70-
AwsLocalEndpoint: "http://localhost:4566",
71-
},
72-
}
73-
err := ParseWithdrawScriptInput(parse, input)
74-
require.Error(t, err)
75-
assert.Contains(t, err.Error(), "select exactly one action")
76-
})
59+
func TestParseWithdrawScriptInputMissingAction(t *testing.T) {
60+
parse := func() {}
61+
envFile := writeTempEnvFile(t, "")
62+
t.Setenv("LPS_UTILS_ENV_FILE", envFile)
63+
64+
input := &WithdrawScriptInput{
65+
BaseInput: scripts.BaseInput{
66+
Network: "regtest",
67+
RskEndpoint: "http://localhost:4444",
68+
SecretSource: "env",
69+
AwsLocalEndpoint: "http://localhost:4566",
70+
},
71+
}
72+
err := ParseWithdrawScriptInput(parse, input)
73+
require.Error(t, err)
74+
assert.Contains(t, err.Error(), "select exactly one action")
75+
}
7776

78-
t.Run("should reject multiple actions", func(t *testing.T) {
79-
envFile := writeTempEnvFile(t, "")
80-
t.Setenv("LPS_UTILS_ENV_FILE", envFile)
81-
82-
input := &WithdrawScriptInput{
83-
BaseInput: scripts.BaseInput{
84-
Network: "regtest",
85-
RskEndpoint: "http://localhost:4444",
86-
SecretSource: "env",
87-
AwsLocalEndpoint: "http://localhost:4566",
88-
},
89-
All: true,
90-
Amount: "1000",
91-
}
92-
err := ParseWithdrawScriptInput(parse, input)
93-
require.Error(t, err)
94-
assert.Contains(t, err.Error(), "select exactly one action")
95-
})
77+
func TestParseWithdrawScriptInputMultipleActions(t *testing.T) {
78+
parse := func() {}
79+
envFile := writeTempEnvFile(t, "")
80+
t.Setenv("LPS_UTILS_ENV_FILE", envFile)
81+
82+
input := &WithdrawScriptInput{
83+
BaseInput: scripts.BaseInput{
84+
Network: "regtest",
85+
RskEndpoint: "http://localhost:4444",
86+
SecretSource: "env",
87+
AwsLocalEndpoint: "http://localhost:4566",
88+
},
89+
All: true,
90+
Amount: "1000",
91+
}
92+
err := ParseWithdrawScriptInput(parse, input)
93+
require.Error(t, err)
94+
assert.Contains(t, err.Error(), "select exactly one action")
95+
}
9696

97-
t.Run("should load defaults from env file", func(t *testing.T) {
98-
envFile, err := os.CreateTemp(t.TempDir(), "lps-utils-*.env")
99-
require.NoError(t, err)
100-
require.NoError(t, os.WriteFile(envFile.Name(), []byte(`
97+
func TestParseWithdrawScriptInputDefaultsFromEnvFile(t *testing.T) {
98+
parse := func() {}
99+
envFile, err := os.CreateTemp(t.TempDir(), "lps-utils-*.env")
100+
require.NoError(t, err)
101+
require.NoError(t, os.WriteFile(envFile.Name(), []byte(`
101102
LPS_STAGE=regtest
102103
RSK_ENDPOINT=http://localhost:4444
103104
AWS_LOCAL_ENDPOINT=http://localhost:4566
104105
SECRET_SRC=env
105106
WALLET_FILE=./keystore.json
106107
`), 0o600))
107-
t.Setenv("LPS_UTILS_ENV_FILE", envFile.Name())
108+
t.Setenv("LPS_UTILS_ENV_FILE", envFile.Name())
108109

109-
input := &WithdrawScriptInput{
110-
BaseInput: scripts.BaseInput{},
111-
All: true,
112-
}
113-
err = ParseWithdrawScriptInput(parse, input)
114-
require.NoError(t, err)
115-
assert.Equal(t, "regtest", input.Network)
116-
assert.Equal(t, "http://localhost:4444", input.RskEndpoint)
117-
assert.Equal(t, "env", input.SecretSource)
118-
assert.Equal(t, "./keystore.json", input.KeystoreFile)
119-
})
110+
input := &WithdrawScriptInput{
111+
BaseInput: scripts.BaseInput{},
112+
All: true,
113+
}
114+
err = ParseWithdrawScriptInput(parse, input)
115+
require.NoError(t, err)
116+
assert.Equal(t, "regtest", input.Network)
117+
assert.Equal(t, "http://localhost:4444", input.RskEndpoint)
118+
assert.Equal(t, "env", input.SecretSource)
119+
assert.Equal(t, "./keystore.json", input.KeystoreFile)
120120
}
121121

122122
func writeTempEnvFile(t *testing.T, contents string) string {

0 commit comments

Comments
 (0)