Function FuzzPolybius

cipher/polybius/polybius_test.go:154–190

(f *testing.F)

}

func FuzzPolybius(f *testing.F) {
	const (
		size       = 5
		characters = "HogeF"
		key        = "abcdefghijklmnopqrstuvwxy"
	)
	f.Add(size, characters, key)
	f.Fuzz(func(t *testing.T, size int, characters, key string) {
		p, err := NewPolybius(key, size, characters)
		switch {
		case err == nil:
		case strings.Contains(err.Error(), "cannot be negative"),
			strings.Contains(err.Error(), "is too small"),
			strings.Contains(err.Error(), "should only contain latin characters"),
			strings.Contains(err.Error(), "contains same character"),
			strings.Contains(err.Error(), "must be as long as size squared"):
			return
		default:
			t.Fatalf("unexpected error when creating a new polybius variable: %v", err)
		}
		encrypted, err := p.Encrypt(characters)
		switch {
		case err == nil:
		case strings.Contains(err.Error(), "does not exist in keys"):
			return
		default:
			t.Fatalf("unexpected error during encryption: %v", err)
		}
		decrypted, err := p.Decrypt(encrypted)
		if err != nil {
			t.Fatalf("unexpected error during decryption: %v", err)
		}
		if decrypted != strings.ToUpper(characters) {
			t.Errorf("Expecting output to match with %q but was %q", characters, decrypted)
		}
	})
}