Fix issues from review feedback.
This commit is contained in:
parent
cf95b6d412
commit
9b9b09d35d
|
@ -85,7 +85,7 @@ func main() {
|
|||
cmd.FailOnError(err, "Couldn't create PA")
|
||||
|
||||
if c.CA.HostnamePolicyFile != "" {
|
||||
err = pa.SetHostnamePolicyFile(c.RA.HostnamePolicyFile)
|
||||
err = pa.SetHostnamePolicyFile(c.CA.HostnamePolicyFile)
|
||||
cmd.FailOnError(err, "Couldn't load hostname policy file")
|
||||
}
|
||||
|
||||
|
|
|
@ -37,14 +37,16 @@ type AuthorityImpl struct {
|
|||
}
|
||||
|
||||
// New constructs a Policy Authority.
|
||||
// TODO(https://github.com/letsencrypt/boulder/issues/1616): Remove the _ bool
|
||||
// argument (used to be enforceWhitelist). Update all callers.
|
||||
func New(dbMap *gorp.DbMap, _ bool, challengeTypes map[string]bool) (*AuthorityImpl, error) {
|
||||
logger := blog.GetAuditLogger()
|
||||
logger.Notice("Policy Authority Starting")
|
||||
|
||||
var padb *AuthorityDatabaseImpl
|
||||
var err error
|
||||
if dbMap != nil {
|
||||
// Setup policy db
|
||||
var err error
|
||||
padb, err = NewAuthorityDatabaseImpl(dbMap)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
|
|
|
@ -65,6 +65,9 @@ func New(filename string, callback func([]byte, error) error) (*Reloader, error)
|
|||
}
|
||||
}
|
||||
err = callback(b, nil)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
go loop()
|
||||
return &Reloader{stopChan}, err
|
||||
}
|
||||
|
|
|
@ -17,7 +17,7 @@ func TestNoStat(t *testing.T) {
|
|||
filename := os.TempDir() + "/doesntexist.123456789"
|
||||
_, err := New(filename, noop)
|
||||
if err == nil {
|
||||
t.Errorf("Expected New to return error when the file doesn't exist.")
|
||||
t.Fatalf("Expected New to return error when the file doesn't exist.")
|
||||
}
|
||||
}
|
||||
|
||||
|
@ -27,7 +27,7 @@ func TestNoRead(t *testing.T) {
|
|||
f.Chmod(0) // no read permissions
|
||||
_, err := New(f.Name(), noop)
|
||||
if err == nil {
|
||||
t.Errorf("Expected New to return error when permission denied.")
|
||||
t.Fatalf("Expected New to return error when permission denied.")
|
||||
}
|
||||
}
|
||||
|
||||
|
@ -38,19 +38,20 @@ func TestFirstError(t *testing.T) {
|
|||
return fmt.Errorf("i die")
|
||||
})
|
||||
if err == nil {
|
||||
t.Errorf("Expected New to return error when the callback returned error the first time.")
|
||||
t.Fatalf("Expected New to return error when the callback returned error the first time.")
|
||||
}
|
||||
}
|
||||
|
||||
func TestFirstSuccess(t *testing.T) {
|
||||
f, _ := ioutil.TempFile("", "test-first-success.txt")
|
||||
defer os.Remove(f.Name())
|
||||
_, err := New(f.Name(), func([]byte, error) error {
|
||||
r, err := New(f.Name(), func([]byte, error) error {
|
||||
return nil
|
||||
})
|
||||
if err != nil {
|
||||
t.Errorf("Expected New to succeed, got %s", err)
|
||||
}
|
||||
r.Stop()
|
||||
}
|
||||
|
||||
// Override makeTicker for testing.
|
||||
|
@ -70,20 +71,18 @@ func makeFakeMakeTicker() (chan<- time.Time, func()) {
|
|||
func TestReload(t *testing.T) {
|
||||
// Mock out makeTicker
|
||||
fakeTick, restoreMakeTicker := makeFakeMakeTicker()
|
||||
defer restoreMakeTicker()
|
||||
|
||||
f, _ := ioutil.TempFile("", "test-reload.txt")
|
||||
filename := f.Name()
|
||||
defer func() {
|
||||
restoreMakeTicker()
|
||||
os.Remove(filename)
|
||||
}()
|
||||
defer os.Remove(filename)
|
||||
|
||||
f.Write([]byte("first body"))
|
||||
f.Close()
|
||||
|
||||
var bodies []string
|
||||
reloads := make(chan []byte, 1)
|
||||
_, err := New(filename, func(b []byte, err error) error {
|
||||
r, err := New(filename, func(b []byte, err error) error {
|
||||
if err != nil {
|
||||
t.Fatalf("Got error in callback: %s", err)
|
||||
}
|
||||
|
@ -92,8 +91,9 @@ func TestReload(t *testing.T) {
|
|||
return nil
|
||||
})
|
||||
if err != nil {
|
||||
t.Errorf("Expected New to succeed, got %s", err)
|
||||
t.Fatalf("Expected New to succeed, got %s", err)
|
||||
}
|
||||
defer r.Stop()
|
||||
expected := []string{"first body"}
|
||||
if !reflect.DeepEqual(bodies, expected) {
|
||||
t.Errorf("Expected bodies = %#v, got %#v", expected, bodies)
|
||||
|
|
Loading…
Reference in New Issue