Work with berkeleydb and try importing it first.

This commit is contained in:
Stuart D. Gathman
2024-05-29 18:08:30 -04:00
parent 35416dfc46
commit 5675adeb3c
3 changed files with 8 additions and 3 deletions
+4 -1
View File
@@ -1,5 +1,8 @@
try: try:
from bsddb3 import db try:
from berkeleydb import db
except:
from bsddb3 import db
class DB(object): class DB(object):
def open(self,fname,mode): def open(self,fname,mode):
if mode == 'r': flags = db.DB_RDONLY if mode == 'r': flags = db.DB_RDONLY
+1
View File
@@ -7,3 +7,4 @@ SMTP-Auth:good@example.com OK
SMTP-Auth:example.com REJECT SMTP-Auth:example.com REJECT
SMTP-Auth:bad@localhost.localdomain REJECT SMTP-Auth:bad@localhost.localdomain REJECT
SMTP-Test: REJECT SMTP-Test: REJECT
SMTP-Test:.baz.com WILDCARD
+3 -2
View File
@@ -8,6 +8,7 @@ class Config(object):
def __init__(self): def __init__(self):
self.access_file='test/access.db' self.access_file='test/access.db'
self.access_file_nulls=True self.access_file_nulls=True
self.access_file_colon = False
class PolicyTestCase(unittest.TestCase): class PolicyTestCase(unittest.TestCase):
@@ -23,8 +24,8 @@ class PolicyTestCase(unittest.TestCase):
print("Missing test/access") print("Missing test/access")
def testPolicy(self): def testPolicy(self):
self.config.use_colon = True self.config.access_file_colon = False
self.config.use_nulls = True self.config.access_file_nulls = True
with MTAPolicy('good@example.com',conf=self.config) as p: with MTAPolicy('good@example.com',conf=self.config) as p:
pol = p.getPolicy('smtp-auth') pol = p.getPolicy('smtp-auth')
self.assertEqual(pol,'OK') self.assertEqual(pol,'OK')