Add more guarding against panics when doing indexing into split strings

This commit is contained in:
David Dworken
2023-11-12 00:55:53 -08:00
parent 37e0e6653d
commit e712d77f62
2 changed files with 12 additions and 7 deletions

View File

@ -351,7 +351,11 @@ func stripShellPrefix(out string) string {
return out
}
func stripTuiCommandPrefix(t *testing.T, out string) string {
require.Contains(t, out, "hishtory tquery")
return strings.TrimSpace(strings.Split(out, "hishtory tquery")[1])
func stripRequiredPrefix(t *testing.T, out, prefix string) string {
require.Contains(t, out, prefix)
return strings.TrimSpace(strings.Split(out, prefix)[1])
}
func stripTuiCommandPrefix(t *testing.T, out string) string {
return stripRequiredPrefix(t, out, "hishtory tquery")
}