alexvanin/certificates
1
0
Fork 0
forked from TrueCloudLab/certificates
Code Pull requests Activity

Complete AuthDB interface.

Browse source
This commit is contained in:
Mariano Cano 2019-11-14 10:49:13 -08:00
parent a4fd76f1a8
commit efc2180c4a
1 changed files with 33 additions and 20 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

53
authority/db_test.go
View file

@ -8,26 +8,18 @@ import (
) )
type MockAuthDB struct { type MockAuthDB struct {
err error err error
ret1 interface{} ret1 interface{}
init func(*db.Config) (db.AuthDB, error) isRevoked func(string) (bool, error)
isRevoked func(string) (bool, error) isSSHRevoked func(string) (bool, error)
revoke func(rci *db.RevokedCertificateInfo) error revoke func(rci *db.RevokedCertificateInfo) error
storeCertificate func(crt *x509.Certificate) error revokeSSH func(rci *db.RevokedCertificateInfo) error
useToken func(id, tok string) (bool, error) storeCertificate func(crt *x509.Certificate) error
isSSHHost func(principal string) (bool, error) useToken func(id, tok string) (bool, error)
storeSSHCertificate func(crt *ssh.Certificate) error isSSHHost func(principal string) (bool, error)
shutdown func() error storeSSHCertificate func(crt *ssh.Certificate) error
} getSSHHostPrincipals func() ([]string, error)
shutdown func() error
func (m *MockAuthDB) Init(c *db.Config) (db.AuthDB, error) {
if m.init != nil {
return m.init(c)
}
if m.ret1 == nil {
return nil, m.err
}
return m.ret1.(*db.DB), m.err
} }
func (m *MockAuthDB) IsRevoked(sn string) (bool, error) { func (m *MockAuthDB) IsRevoked(sn string) (bool, error) {
@ -37,6 +29,13 @@ func (m *MockAuthDB) IsRevoked(sn string) (bool, error) {
return m.ret1.(bool), m.err return m.ret1.(bool), m.err
} }
func (m *MockAuthDB) IsSSHRevoked(sn string) (bool, error) {
if m.isSSHRevoked != nil {
return m.isSSHRevoked(sn)
}
return m.ret1.(bool), m.err
}
func (m *MockAuthDB) UseToken(id, tok string) (bool, error) { func (m *MockAuthDB) UseToken(id, tok string) (bool, error) {
if m.useToken != nil { if m.useToken != nil {
return m.useToken(id, tok) return m.useToken(id, tok)
@ -54,6 +53,13 @@ func (m *MockAuthDB) Revoke(rci *db.RevokedCertificateInfo) error {
return m.err return m.err
} }
func (m *MockAuthDB) RevokeSSH(rci *db.RevokedCertificateInfo) error {
if m.revokeSSH != nil {
return m.revokeSSH(rci)
}
return m.err
}
func (m *MockAuthDB) StoreCertificate(crt *x509.Certificate) error { func (m *MockAuthDB) StoreCertificate(crt *x509.Certificate) error {
if m.storeCertificate != nil { if m.storeCertificate != nil {
return m.storeCertificate(crt) return m.storeCertificate(crt)
@ -75,6 +81,13 @@ func (m *MockAuthDB) StoreSSHCertificate(crt *ssh.Certificate) error {
return m.err return m.err
} }
func (m *MockAuthDB) GetSSHHostPrincipals() ([]string, error) {
if m.getSSHHostPrincipals != nil {
return m.getSSHHostPrincipals()
}
return m.ret1.([]string), m.err
}
func (m *MockAuthDB) Shutdown() error { func (m *MockAuthDB) Shutdown() error {
if m.shutdown != nil { if m.shutdown != nil {
return m.shutdown() return m.shutdown()