mirror of
https://github.com/square/certigo.git
synced 2024-04-21 12:32:40 +00:00
Matthew McPherrin
26cd65d9bd
This splits the logic into a cli, which handles parsing command line args and reading files, and a terminal abstraction for handling user input and ouput. All uses of os.Exit are removed in favor of returning errors. Overall this enables better testing and reuse of code. Previously we had to rely on external unit testing for CLI tests, which are harder to write tests.
95 lines
3.6 KiB
Go
95 lines
3.6 KiB
Go
package cli
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/square/certigo/cli/terminal"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
const testCert string = `
|
|
-----BEGIN CERTIFICATE-----
|
|
MIIE1DCCArygAwIBAgICEAAwDQYJKoZIhvcNAQELBQAwPzELMAkGA1UEBhMCVVMx
|
|
FzAVBgNVBAoMDnRlc3QxLmFjbWUuY29tMRcwFQYDVQQDDA5JbnRlcm1lZGlhZXRD
|
|
QTAeFw0xNzA3MTkxNjUwMjBaFw0xNzA3MjkxNjUwMjBaMDUxCzAJBgNVBAYTAlVT
|
|
MRcwFQYDVQQKDA50ZXN0MS5hY21lLmNvbTENMAsGA1UEAwwEYmxvZzCCASIwDQYJ
|
|
KoZIhvcNAQEBBQADggEPADCCAQoCggEBAKm8P47lABp4+rz2nN+QYrxedbaFVWoF
|
|
FuoSkqcHsafMwbMrN+kI6wJVtlbwviDvxWFJ92q0H71QNFybTsmof3KUN/kYCp7P
|
|
+LKhBrN0ttWI5q6v5eDrjN0VdtVdnlZOYmJFbvETOgfK/qXKNRRM8HYW0tdqrtEw
|
|
CR5dIu53xVUSViBdwXpuy2c5W2mFn1gxTpdW+3hbZsL1pHrU9qPWLtTgl/KY8kjs
|
|
I7KW1cIcinE4SJomhB5L/4emhxKGY+kEa2+fN9IPjjvKSMOw9kiBKk1GHZcIY5EA
|
|
O3TIfUk3fysPzi5qA0su/bNtPQy1uXgXS10xUlV7pqRPvHjiNzgFkXUCAwEAAaOB
|
|
4zCB4DAJBgNVHRMEAjAAMB0GA1UdDgQWBBRVQ91jSOONzVr1VGBdJOlPN+3XxTBg
|
|
BgNVHSMEWTBXgBQ13bfx50rDZO3y2CZdHPgleFUEoKE7pDkwNzELMAkGA1UEBhMC
|
|
VVMxFzAVBgNVBAoMDnRlc3QxLmFjbWUuY29tMQ8wDQYDVQQDDAZSb290Q0GCAhAA
|
|
MA4GA1UdDwEB/wQEAwIDqDATBgNVHSUEDDAKBggrBgEFBQcDATAtBgNVHREEJjAk
|
|
hiJzcGlmZmU6Ly9kZXYuYWNtZS5jb20vcGF0aC9zZXJ2aWNlMA0GCSqGSIb3DQEB
|
|
CwUAA4ICAQBp2+rtUxt1VmNM/vi6PwoSoYzWFmQ2nc4OM7bsOG4uppU54wRYZ+T7
|
|
c42EcrpyBgWn+rWHT1Hi6SNcmloKHydaUTZ4pq3IlKKnBNqwivU5BzIxYLDrhR/U
|
|
wd9s1tgmLvADqkQa1XjjSFn5Auoj1R640ry4qpw8IOusdm6wVhru4ssRnHX4E2uR
|
|
jQe7b3ws38aZhjtL78Ip0BB4yPxWJRp/WmEoT33QP+cZhA4IYWECxNODr6DSJeq2
|
|
VNu/6JACGrNfM2Sjt4Wxz+nIa3cKDNCA6PR8StTUTcoQ6ZBzpn+n/Q1xSRIOJz6N
|
|
hgfkyb9O7HAMdAP+TxehjqG3gh5Ky2DgYMCIZOztVzsuOb1DGJe/kGUKeRJLl2/O
|
|
QwkctwUOcVIxckNu6OvclriFzvoXObqO77XeCI2V1Vef0wGTWlWNOdbFa4708Y7f
|
|
5UdwInYQUi87RFDnc1SDU4Jrsv4KzZiv9FCfDg8pCBIdWpWT7DAuI0d7i7PZ+iFt
|
|
ZZ6sb/YDkyiDXU4ar/dja0FDE2r7jsN9D+FfW49+iDvXr4ELQyhZpW3Zr1Ojwm58
|
|
CJzjZwbRYiVwPBRsKmiYfO1E7esvw3CmjK5chfz8c40f6/APDro9ZmYNBRv2CnJy
|
|
t/DtcM/GpAhBbLP9Tk7kPB41v5fRIxVDo50Iz/qvkr37pQ4RsejSFg==
|
|
-----END CERTIFICATE-----
|
|
`
|
|
|
|
const expectedVerbose string = `** CERTIFICATE 1 **
|
|
Serial: 4096
|
|
Valid: 2017-07-19 16:50 UTC to 2017-07-29 16:50 UTC
|
|
Signature: SHA256-RSA
|
|
Subject Info:
|
|
Country: US
|
|
Organization: test1.acme.com
|
|
CommonName: blog
|
|
Issuer Info:
|
|
Country: US
|
|
Organization: test1.acme.com
|
|
CommonName: IntermediaetCA
|
|
Subject Key ID: 55:43:DD:63:48:E3:8D:CD:5A:F5:54:60:5D:24:E9:4F:37:ED:D7:C5
|
|
Authority Key ID: 35:DD:B7:F1:E7:4A:C3:64:ED:F2:D8:26:5D:1C:F8:25:78:55:04:A0
|
|
Basic Constraints: CA:false
|
|
Key Usage:
|
|
Digital Signature
|
|
Key Encipherment
|
|
Key Agreement
|
|
Extended Key Usage:
|
|
Server Auth
|
|
URI Names:
|
|
spiffe://dev.acme.com/path/service
|
|
|
|
`
|
|
|
|
// Test basic dump functionality: Dump a cert
|
|
func TestDump(t *testing.T) {
|
|
tmpfile, err := ioutil.TempFile("", t.Name())
|
|
require.NoError(t, err)
|
|
defer os.Remove(tmpfile.Name())
|
|
|
|
_, err = tmpfile.Write([]byte(testCert))
|
|
require.NoError(t, err)
|
|
|
|
args := []string{"dump", "--verbose", "--format", "PEM", tmpfile.Name()}
|
|
testTerminal := terminal.TestTerminal{Width: 80}
|
|
|
|
assert.EqualValues(t, 0, Run(args, &testTerminal), "process should exit 0")
|
|
assert.Empty(t, testTerminal.ErrorBuf.Bytes(), "no error output expected")
|
|
assert.EqualValues(t, expectedVerbose, testTerminal.OutputBuf.String())
|
|
}
|
|
|
|
func TestDumpMissingFile(t *testing.T) {
|
|
testTerminal := terminal.TestTerminal{Width: 80}
|
|
args := []string{"dump", "this-is-a-file-that-definitely-does-not-exist1111.pem"}
|
|
assert.EqualValues(t, 2, Run(args, &testTerminal), "process should exit 0")
|
|
const expected = "path 'this-is-a-file-that-definitely-does-not-exist1111.pem' does not exist, try --help\n"
|
|
assert.Equal(t, expected, testTerminal.ErrorBuf.String())
|
|
assert.Empty(t, testTerminal.OutputBuf.Bytes())
|
|
}
|