Skip to content

Commit

Permalink
Merge pull request #2497 from jhaals/jhaals/tests
Browse files Browse the repository at this point in the history
Increase test coverage
  • Loading branch information
jhaals authored Oct 11, 2024
2 parents bdcae51 + 0b20839 commit c8049b4
Show file tree
Hide file tree
Showing 2 changed files with 38 additions and 4 deletions.
37 changes: 37 additions & 0 deletions cmd/yopass/main_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -139,6 +139,43 @@ func TestDecryptWithoutCustomKey(t *testing.T) {
}
}

func TestDecryptWithInvalidUrl(t *testing.T) {
viper.Set("decrypt", "https://yopass.se")
err := decrypt(nil)
if err == nil {
t.Fatal("expected invalid url error, got none")
}
want := `Invalid yopass decrypt URL: unexpected URL: "https://yopass.se"`
if err.Error() != want {
t.Fatalf("expected %s, got %s", want, err.Error())
}
}

func TestDecryptWithUnconfiguredUrl(t *testing.T) {
viper.Set("decrypt", "")
err := decrypt(nil)
if err == nil {
t.Fatal("expected unconfigured url error, got none")
}
want := `Unconfigured yopass decrypt URL, set --api and --url`
if err.Error() != want {
t.Fatalf("expected %s, got %s", want, err.Error())
}
}

func TestSecretNotFoundError(t *testing.T) {
viper.Set("decrypt", "https://yopass.se/#/c/21701b28-fb3f-451d-8a52-3e6c9094e7")
viper.Set("key", "woo")
err := decrypt(nil)
if err == nil {
t.Fatal("expected error, got none")
}
want := `Failed to fetch secret: yopass server error: unexpected response 404 Not Found: 404 page not found`
if strings.TrimRight(err.Error(), "\n") != want {
t.Fatalf("expected %s, got %s", want, err.Error())
}
}

func TestExpiration(t *testing.T) {
tests := []struct {
input string
Expand Down
5 changes: 1 addition & 4 deletions pkg/server/redis.go
Original file line number Diff line number Diff line change
Expand Up @@ -37,13 +37,10 @@ func (r *Redis) Get(key string) (yopass.Secret, error) {
}

if s.OneTime {
res, err := r.client.Del(key).Result()
_, err := r.Delete(key)
if err != nil {
return s, err
}
if res != 1 {
return s, fmt.Errorf("expected to delete 1 key, but deleted %d keys", res)
}
}
return s, nil
}
Expand Down

0 comments on commit c8049b4

Please sign in to comment.