mirror of
https://github.com/aptly-dev/aptly.git
synced 2026-05-06 22:18:28 +00:00
341 lines
13 KiB
Go
341 lines
13 KiB
Go
package api
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
|
|
. "gopkg.in/check.v1"
|
|
)
|
|
|
|
type GPGSuite struct {
|
|
APISuite
|
|
}
|
|
|
|
var _ = Suite(&GPGSuite{})
|
|
|
|
// TestParseGPGOutputEmpty tests parsing of empty GPG output
|
|
func (s *GPGSuite) TestParseGPGOutputEmpty(c *C) {
|
|
output := ""
|
|
keys := parseGPGOutput(output)
|
|
c.Check(keys, HasLen, 0)
|
|
}
|
|
|
|
// TestParseGPGOutputSingleKeyMinimal tests parsing a single key with minimal fields
|
|
func (s *GPGSuite) TestParseGPGOutputSingleKeyMinimal(c *C) {
|
|
// Minimal valid GPG output with one key
|
|
output := `pub:u:4096:1:8B48AD6246925553:1611864000:1643400000:uidhash:::scESC:::::::23::0:
|
|
uid:u::::1611864000::1234567890::John Doe <john@example.com>::::::::::0:
|
|
fpr:::::::::D8E8F5A516E7A2C4F3E4B5A6C7D8E9F0:`
|
|
|
|
keys := parseGPGOutput(output)
|
|
c.Check(keys, HasLen, 1)
|
|
|
|
key := keys[0]
|
|
c.Check(key.KeyID, Equals, "8B48AD6246925553")
|
|
c.Check(key.Validity, Equals, "u")
|
|
c.Check(key.CreatedAt, Equals, "1611864000")
|
|
c.Check(key.Fingerprint, Equals, "D8E8F5A516E7A2C4F3E4B5A6C7D8E9F0")
|
|
c.Check(key.UserIDs, DeepEquals, []string{"John Doe <john@example.com>"})
|
|
}
|
|
|
|
// TestParseGPGOutputMultipleKeys tests parsing multiple keys
|
|
func (s *GPGSuite) TestParseGPGOutputMultipleKeys(c *C) {
|
|
output := `pub:u:4096:1:8B48AD6246925553:1611864000:1643400000:uidhash:::scESC:::::::23::0:
|
|
uid:u::::1611864000::1234567890::John Doe <john@example.com>::::::::::0:
|
|
fpr:::::::::D8E8F5A516E7A2C4F3E4B5A6C7D8E9F0:
|
|
pub:f:2048:1:A1B2C3D4E5F67890:1580592000:1612128000:uidhash:::scESC:::::::23::0:
|
|
uid:f::::1580592000::0987654321::Jane Smith <jane@example.com>::::::::::0:
|
|
fpr:::::::::E9F0A1B2C3D4E5F6A7B8C9D0E1F2A3B4:`
|
|
|
|
keys := parseGPGOutput(output)
|
|
c.Check(keys, HasLen, 2)
|
|
|
|
// First key
|
|
c.Check(keys[0].KeyID, Equals, "8B48AD6246925553")
|
|
c.Check(keys[0].Validity, Equals, "u")
|
|
c.Check(keys[0].UserIDs, DeepEquals, []string{"John Doe <john@example.com>"})
|
|
|
|
// Second key
|
|
c.Check(keys[1].KeyID, Equals, "A1B2C3D4E5F67890")
|
|
c.Check(keys[1].Validity, Equals, "f")
|
|
c.Check(keys[1].UserIDs, DeepEquals, []string{"Jane Smith <jane@example.com>"})
|
|
}
|
|
|
|
// TestParseGPGOutputMultipleUIDs tests a key with multiple user IDs
|
|
func (s *GPGSuite) TestParseGPGOutputMultipleUIDs(c *C) {
|
|
output := `pub:u:4096:1:8B48AD6246925553:1611864000:1643400000:uidhash:::scESC:::::::23::0:
|
|
uid:u::::1611864000::1234567890::John Doe <john@example.com>::::::::::0:
|
|
uid:u::::1611864000::1234567891::John Doe <john.doe@company.com>::::::::::0:
|
|
fpr:::::::::D8E8F5A516E7A2C4F3E4B5A6C7D8E9F0:`
|
|
|
|
keys := parseGPGOutput(output)
|
|
c.Check(keys, HasLen, 1)
|
|
|
|
key := keys[0]
|
|
c.Check(key.UserIDs, HasLen, 2)
|
|
c.Check(key.UserIDs, DeepEquals, []string{
|
|
"John Doe <john@example.com>",
|
|
"John Doe <john.doe@company.com>",
|
|
})
|
|
}
|
|
|
|
// TestParseGPGOutputMalformedLines tests that malformed lines are skipped
|
|
func (s *GPGSuite) TestParseGPGOutputMalformedLines(c *C) {
|
|
// Mix of valid and invalid lines (too few fields)
|
|
output := `pub:u:4096:1:8B48AD6246925553:1611864000:1643400000:uidhash:::scESC:::::::23::0:
|
|
invalid:line:with:only:three:fields
|
|
uid:u::::1611864000::1234567890::John Doe <john@example.com>::::::::::0:
|
|
fpr:::::::::D8E8F5A516E7A2C4F3E4B5A6C7D8E9F0:`
|
|
|
|
keys := parseGPGOutput(output)
|
|
c.Check(keys, HasLen, 1)
|
|
c.Check(keys[0].KeyID, Equals, "8B48AD6246925553")
|
|
}
|
|
|
|
// TestParseGPGOutputEmptyLines tests that empty lines are skipped
|
|
func (s *GPGSuite) TestParseGPGOutputEmptyLines(c *C) {
|
|
output := `pub:u:4096:1:8B48AD6246925553:1611864000:1643400000:uidhash:::scESC:::::::23::0:
|
|
|
|
uid:u::::1611864000::1234567890::John Doe <john@example.com>::::::::::0:
|
|
|
|
fpr:::::::::D8E8F5A516E7A2C4F3E4B5A6C7D8E9F0:`
|
|
|
|
keys := parseGPGOutput(output)
|
|
c.Check(keys, HasLen, 1)
|
|
c.Check(keys[0].KeyID, Equals, "8B48AD6246925553")
|
|
}
|
|
|
|
// TestParseGPGOutputKeyWithoutUID tests a public key without user ID
|
|
func (s *GPGSuite) TestParseGPGOutputKeyWithoutUID(c *C) {
|
|
// Key without uid record (should still be included)
|
|
output := `pub:u:4096:1:8B48AD6246925553:1611864000:1643400000:uidhash:::scESC:::::::23::0:
|
|
fpr:::::::::D8E8F5A516E7A2C4F3E4B5A6C7D8E9F0:`
|
|
|
|
keys := parseGPGOutput(output)
|
|
c.Check(keys, HasLen, 1)
|
|
|
|
key := keys[0]
|
|
c.Check(key.KeyID, Equals, "8B48AD6246925553")
|
|
c.Check(key.UserIDs, HasLen, 0)
|
|
c.Check(key.Fingerprint, Equals, "D8E8F5A516E7A2C4F3E4B5A6C7D8E9F0")
|
|
}
|
|
|
|
// TestParseGPGOutputVariousValidity tests different validity values
|
|
func (s *GPGSuite) TestParseGPGOutputVariousValidity(c *C) {
|
|
output := `pub:u:4096:1:KEY1111111111111:1611864000:1643400000:uidhash:::scESC:::::::23::0:
|
|
uid:u::::1611864000::1234567890::Key1::::::::::0:
|
|
fpr:::::::::1111111111111111111111111111111111111111:
|
|
pub:f:4096:1:KEY2222222222222:1611864000:1643400000:uidhash:::scESC:::::::23::0:
|
|
uid:f::::1611864000::1234567891::Key2::::::::::0:
|
|
fpr:::::::::2222222222222222222222222222222222222222:
|
|
pub:m:4096:1:KEY3333333333333:1611864000:1643400000:uidhash:::scESC:::::::23::0:
|
|
uid:m::::1611864000::1234567892::Key3::::::::::0:
|
|
fpr:::::::::3333333333333333333333333333333333333333:
|
|
pub:n:4096:1:KEY4444444444444:1611864000:1643400000:uidhash:::scESC:::::::23::0:
|
|
uid:n::::1611864000::1234567893::Key4::::::::::0:
|
|
fpr:::::::::4444444444444444444444444444444444444444:`
|
|
|
|
keys := parseGPGOutput(output)
|
|
c.Check(keys, HasLen, 4)
|
|
|
|
validities := []string{"u", "f", "m", "n"}
|
|
for i, validity := range validities {
|
|
c.Check(keys[i].Validity, Equals, validity)
|
|
}
|
|
}
|
|
|
|
// TestParseGPGOutputShortKeyID tests that key IDs are shortened to 16 chars
|
|
func (s *GPGSuite) TestParseGPGOutputShortKeyID(c *C) {
|
|
// 40-character key ID that should be shortened to last 16 chars
|
|
longKeyID := "0123456789ABCDEF0123456789ABCDEF8B48AD62"
|
|
output := `pub:u:4096:1:` + longKeyID + `:1611864000:1643400000:uidhash:::scESC:::::::23::0:
|
|
uid:u::::1611864000::1234567890::John Doe <john@example.com>::::::::::0:
|
|
fpr:::::::::D8E8F5A516E7A2C4F3E4B5A6C7D8E9F0:`
|
|
|
|
keys := parseGPGOutput(output)
|
|
c.Check(keys, HasLen, 1)
|
|
// Should extract the last 16 characters: 89ABCDEF8B48AD62
|
|
c.Check(keys[0].KeyID, Equals, "89ABCDEF8B48AD62")
|
|
}
|
|
|
|
// TestParseGPGOutputSpecialCharactersInUID tests user IDs with special characters
|
|
func (s *GPGSuite) TestParseGPGOutputSpecialCharactersInUID(c *C) {
|
|
// UID with Unicode characters and special formatting
|
|
output := `pub:u:4096:1:8B48AD6246925553:1611864000:1643400000:uidhash:::scESC:::::::23::0:
|
|
uid:u::::1611864000::1234567890::J\xc3\xb6hn D\xc3\xb6\xc3\xa9 (D\xc3\xbcss) <john@example.com>::::::::::0:
|
|
fpr:::::::::D8E8F5A516E7A2C4F3E4B5A6C7D8E9F0:`
|
|
|
|
keys := parseGPGOutput(output)
|
|
c.Check(keys, HasLen, 1)
|
|
// Should preserve the encoded special characters
|
|
c.Check(keys[0].UserIDs, HasLen, 1)
|
|
}
|
|
|
|
// TestAPIGPGListKeysDefaultKeyring tests the HTTP endpoint with default keyring
|
|
func (s *GPGSuite) TestAPIGPGListKeysDefaultKeyring(c *C) {
|
|
c.Skip("Requires GPG binary and test key installation. Run manually if needed.")
|
|
|
|
response, err := s.HTTPRequest("GET", "/api/gpg/keys", nil)
|
|
c.Assert(err, IsNil)
|
|
c.Check(response.Code, Equals, 200)
|
|
|
|
// Verify response is valid JSON
|
|
var result gpgKeyListResponse
|
|
err = json.NewDecoder(response.Body).Decode(&result)
|
|
c.Assert(err, IsNil)
|
|
c.Check(result.Keys, NotNil)
|
|
}
|
|
|
|
// TestAPIGPGListKeysWithKeyringParam tests the HTTP endpoint with custom keyring parameter
|
|
func (s *GPGSuite) TestAPIGPGListKeysWithKeyringParam(c *C) {
|
|
c.Skip("Requires GPG binary and test key installation. Run manually if needed.")
|
|
|
|
response, err := s.HTTPRequest("GET", "/api/gpg/keys?keyring=custom.gpg", nil)
|
|
c.Assert(err, IsNil)
|
|
// May fail if custom.gpg doesn't exist, but endpoint should handle gracefully
|
|
// Accept either 200 (success) or 400 (file not found)
|
|
code := response.Code
|
|
c.Check(code == 200 || code == 400, Equals, true)
|
|
}
|
|
|
|
// TestAPIGPGListKeysResponseFormat tests that the response has the correct structure
|
|
func (s *GPGSuite) TestAPIGPGListKeysResponseFormat(c *C) {
|
|
c.Skip("Requires GPG binary and test key installation. Run manually if needed.")
|
|
|
|
response, err := s.HTTPRequest("GET", "/api/gpg/keys", nil)
|
|
c.Assert(err, IsNil)
|
|
|
|
if response.Code == 200 {
|
|
var result gpgKeyListResponse
|
|
err = json.NewDecoder(response.Body).Decode(&result)
|
|
c.Assert(err, IsNil)
|
|
|
|
// If there are keys, verify their structure
|
|
for _, key := range result.Keys {
|
|
c.Check(key.KeyID, Not(Equals), "")
|
|
c.Check(key.Validity, Not(Equals), "")
|
|
c.Check(key.CreatedAt, Not(Equals), "")
|
|
}
|
|
}
|
|
}
|
|
|
|
// TestParseGPGOutputEdgeCaseUIDWithoutFields tests UID record with missing fields
|
|
func (s *GPGSuite) TestParseGPGOutputEdgeCaseUIDWithoutFields(c *C) {
|
|
// UID record with fewer than 10 fields
|
|
output := `pub:u:4096:1:8B48AD6246925553:1611864000:1643400000:uidhash:::scESC:::::::23::0:
|
|
uid:u::::1611864000::1234567890:
|
|
fpr:::::::::D8E8F5A516E7A2C4F3E4B5A6C7D8E9F0:`
|
|
|
|
keys := parseGPGOutput(output)
|
|
c.Check(keys, HasLen, 1)
|
|
// Should not have user ID since it's in field 9 and this record is too short
|
|
c.Check(keys[0].UserIDs, HasLen, 0)
|
|
}
|
|
|
|
// TestParseGPGOutputFingerprintWithoutCurrentKey tests FPR record appearing before any PUB
|
|
func (s *GPGSuite) TestParseGPGOutputFingerprintWithoutCurrentKey(c *C) {
|
|
// FPR record without a preceding PUB (should be ignored)
|
|
output := `fpr:::::::::D8E8F5A516E7A2C4F3E4B5A6C7D8E9F0:
|
|
pub:u:4096:1:8B48AD6246925553:1611864000:1643400000:uidhash:::scESC:::::::23::0:
|
|
uid:u::::1611864000::1234567890::John Doe <john@example.com>::::::::::0:
|
|
fpr:::::::::D8E8F5A516E7A2C4F3E4B5A6C7D8E9F0:`
|
|
|
|
keys := parseGPGOutput(output)
|
|
c.Check(keys, HasLen, 1)
|
|
// Should only have one key with the correct fingerprint
|
|
c.Check(keys[0].Fingerprint, Equals, "D8E8F5A516E7A2C4F3E4B5A6C7D8E9F0")
|
|
}
|
|
|
|
// TestParseGPGOutputComplexRealWorldExample tests real-world-like GPG output
|
|
func (s *GPGSuite) TestParseGPGOutputComplexRealWorldExample(c *C) {
|
|
// Real-world GPG output with multiple keys, UIDs, and other record types (sig, sub)
|
|
// Note: sub and sig records are skipped as we only care about pub/uid/fpr
|
|
realWorldOutput := `tru::1:1611864000:0:3:1:5
|
|
pub:u:4096:1:8B48AD6246925553:1611864000:2023-01-15T00:00:00:::::scESC:::::::23::0:
|
|
fpr:::::::::D8E8F5A516E7A2C4F3E4B5A6C7D8E9F0:
|
|
uid:u::::1611864000::1234567890::John Doe <john@example.com>::::::::::0:
|
|
uid:u::::1611864100::1234567891::John Doe <john@work.com>::::::::::0:
|
|
pub:f:2048:1:1234567890123456:1580592000:2022-12-31T00:00:00::u:::scESC:::::::23::0:
|
|
fpr:::::::::F4E3D2C1B0A9F8E7D6C5B4A3F2E1D0C9:
|
|
uid:f::::1580592000::0987654321::Maintainer Key <maint@example.com>::::::::::0:`
|
|
|
|
keys := parseGPGOutput(realWorldOutput)
|
|
c.Check(keys, HasLen, 2)
|
|
|
|
// First key should have 2 UIDs
|
|
c.Check(keys[0].KeyID, Equals, "8B48AD6246925553")
|
|
c.Check(keys[0].UserIDs, HasLen, 2)
|
|
c.Check(keys[0].Fingerprint, Equals, "D8E8F5A516E7A2C4F3E4B5A6C7D8E9F0")
|
|
|
|
// Second key should have 1 UID
|
|
c.Check(keys[1].KeyID, Equals, "1234567890123456")
|
|
c.Check(keys[1].UserIDs, HasLen, 1)
|
|
c.Check(keys[1].Fingerprint, Equals, "F4E3D2C1B0A9F8E7D6C5B4A3F2E1D0C9")
|
|
}
|
|
|
|
// TestParseGPGOutputConsecutiveEmptyUIDs tests handling of consecutive empty user ID fields
|
|
func (s *GPGSuite) TestParseGPGOutputConsecutiveEmptyUIDs(c *C) {
|
|
output := `pub:u:4096:1:8B48AD6246925553:1611864000:1643400000:uidhash:::scESC:::::::23::0:
|
|
uid:u::::1611864000::1234567890:::::::::::0:
|
|
uid:u::::1611864000::1234567891::John Doe <john@example.com>::::::::::0:
|
|
fpr:::::::::D8E8F5A516E7A2C4F3E4B5A6C7D8E9F0:`
|
|
|
|
keys := parseGPGOutput(output)
|
|
c.Check(keys, HasLen, 1)
|
|
// Should skip empty UID but include the non-empty one
|
|
c.Check(keys[0].UserIDs, HasLen, 1)
|
|
c.Check(keys[0].UserIDs[0], Equals, "John Doe <john@example.com>")
|
|
}
|
|
|
|
// TestGPGDeleteKeyParamsValidation tests gpgDeleteKeyParams validation
|
|
func (s *GPGSuite) TestGPGDeleteKeyParamsValidation(c *C) {
|
|
// This is a unit test that validates parameter structure (no HTTP needed)
|
|
params := gpgDeleteKeyParams{
|
|
Keyring: "custom.gpg",
|
|
GpgKeyID: "8B48AD6246925553",
|
|
}
|
|
c.Check(params.Keyring, Equals, "custom.gpg")
|
|
c.Check(params.GpgKeyID, Equals, "8B48AD6246925553")
|
|
}
|
|
|
|
// TestAPIGPGDeleteKeyMissingKeyID tests delete with missing key ID parameter
|
|
func (s *GPGSuite) TestAPIGPGDeleteKeyMissingKeyID(c *C) {
|
|
c.Skip("Requires GPG binary. Run manually if needed.")
|
|
|
|
body, err := json.Marshal(map[string]string{
|
|
"Keyring": "trustedkeys.gpg",
|
|
// GpgKeyID is missing
|
|
})
|
|
c.Assert(err, IsNil)
|
|
|
|
response, err := s.HTTPRequest("DELETE", "/api/gpg/key", bytes.NewReader(body))
|
|
c.Assert(err, IsNil)
|
|
c.Check(response.Code, Equals, 400)
|
|
}
|
|
|
|
// TestAPIGPGDeleteKeyInvalidJSON tests delete with invalid JSON request
|
|
func (s *GPGSuite) TestAPIGPGDeleteKeyInvalidJSON(c *C) {
|
|
c.Skip("Requires GPG binary. Run manually if needed.")
|
|
|
|
response, err := s.HTTPRequest("DELETE", "/api/gpg/key", bytes.NewReader([]byte("invalid json")))
|
|
c.Assert(err, IsNil)
|
|
c.Check(response.Code, Equals, 400)
|
|
}
|
|
|
|
// TestAPIGPGDeleteKeySuccess tests successful key deletion
|
|
func (s *GPGSuite) TestAPIGPGDeleteKeySuccess(c *C) {
|
|
c.Skip("Requires GPG binary and test key installation. Run manually if needed.")
|
|
|
|
body, err := json.Marshal(gpgDeleteKeyParams{
|
|
Keyring: "trustedkeys.gpg",
|
|
GpgKeyID: "8B48AD6246925553",
|
|
})
|
|
c.Assert(err, IsNil)
|
|
|
|
response, err := s.HTTPRequest("DELETE", "/api/gpg/key", bytes.NewReader(body))
|
|
c.Assert(err, IsNil)
|
|
// Should succeed (200) or fail gracefully (400) if key doesn't exist
|
|
code := response.Code
|
|
c.Check(code == 200 || code == 400, Equals, true)
|
|
}
|