Merge branch 'master' into sig-reuse

This commit is contained in:
Jeff Hodges 2015-10-06 17:37:30 -07:00
commit 17674cb3a7
2 changed files with 29 additions and 1 deletions

View File

@ -103,8 +103,8 @@ func (padb *PolicyAuthorityDatabaseImpl) DumpRules() (rs RuleSet, err error) {
}
for _, r := range bList {
r.Host = core.ReverseName(r.Host)
rs.Blacklist = append(rs.Blacklist, r)
}
rs.Blacklist = bList
var wList []WhitelistRule
_, err = padb.dbMap.Select(&wList, "SELECT * FROM whitelist")
if err != nil {

View File

@ -24,6 +24,34 @@ func padbImpl(t *testing.T) (*PolicyAuthorityDatabaseImpl, func()) {
return padb, cleanUp
}
func TestLoadAndDumpRules(t *testing.T) {
p, cleanup := padbImpl(t)
defer cleanup()
load := RuleSet{
Blacklist: []BlacklistRule{
BlacklistRule{
Host: "bad.com",
},
},
Whitelist: []WhitelistRule{
WhitelistRule{
Host: "good.bad.com",
},
},
}
err := p.LoadRules(load)
test.AssertNotError(t, err, "Couldn't load rules")
dumped, err := p.DumpRules()
test.AssertNotError(t, err, "Couldn't dump rules")
test.AssertEquals(t, len(dumped.Blacklist), 1)
test.AssertEquals(t, len(dumped.Whitelist), 1)
test.AssertEquals(t, dumped.Whitelist[0], load.Whitelist[0])
test.AssertEquals(t, dumped.Blacklist[0], load.Blacklist[0])
}
func TestBlacklist(t *testing.T) {
p, cleanup := padbImpl(t)
defer cleanup()