From 1dc90ad1abeb752713aa5fe7a3a3074156e941f6 Mon Sep 17 00:00:00 2001 From: Cedric Kienzler Date: Tue, 22 Mar 2022 21:26:33 +0100 Subject: [PATCH] Add tests for single key Adding tests to verify we do not break the usage of a single AGE key Signed-off-by: Cedric Kienzler --- age/keysource_test.go | 26 +++++++++++++++++++++++++- 1 file changed, 25 insertions(+), 1 deletion(-) diff --git a/age/keysource_test.go b/age/keysource_test.go index a19fa3c2f..57e53f7dd 100644 --- a/age/keysource_test.go +++ b/age/keysource_test.go @@ -21,7 +21,19 @@ func TestMasterKeysFromRecipientsEmpty(t *testing.T) { assert.Equal(recipients, make([]*MasterKey, 0)) } -func TestMasterKeyFromRecipientWithLeadingAndTrailingSpaces(t *testing.T) { +func TestMasterKeyFromRecipientWithLeadingAndTrailingSpacesSingle(t *testing.T) { + assert := assert.New(t) + + commaSeparatedRecipients := " age1yt3tfqlfrwdwx0z0ynwplcr6qxcxfaqycuprpmy89nr83ltx74tqdpszlw " + keys, err := MasterKeysFromRecipients(commaSeparatedRecipients) + + assert.NoError(err) + + assert.Equal(len(keys), 1) + assert.Equal(keys[0].Recipient, "age1yt3tfqlfrwdwx0z0ynwplcr6qxcxfaqycuprpmy89nr83ltx74tqdpszlw") +} + +func TestMasterKeyFromRecipientWithLeadingAndTrailingSpacesMultiple(t *testing.T) { assert := assert.New(t) commaSeparatedRecipients := " age1yt3tfqlfrwdwx0z0ynwplcr6qxcxfaqycuprpmy89nr83ltx74tqdpszlw , age1tmaae3ld5vpevmsh5yacsauzx8jetg300mpvc4ugp5zr5l6ssq9sla97ep " @@ -34,6 +46,18 @@ func TestMasterKeyFromRecipientWithLeadingAndTrailingSpaces(t *testing.T) { assert.Equal(keys[1].Recipient, "age1tmaae3ld5vpevmsh5yacsauzx8jetg300mpvc4ugp5zr5l6ssq9sla97ep") } +func TestMasterKeysFromRecipientsWithSingle(t *testing.T) { + assert := assert.New(t) + + commaSeparatedRecipients := "age1yt3tfqlfrwdwx0z0ynwplcr6qxcxfaqycuprpmy89nr83ltx74tqdpszlw" + keys, err := MasterKeysFromRecipients(commaSeparatedRecipients) + + assert.NoError(err) + + assert.Equal(len(keys), 1) + assert.Equal(keys[0].Recipient, "age1yt3tfqlfrwdwx0z0ynwplcr6qxcxfaqycuprpmy89nr83ltx74tqdpszlw") +} + func TestMasterKeysFromRecipientsWithMultiple(t *testing.T) { assert := assert.New(t)