def testParseHeader(self):
parser = self.Parser('header {\ntarget::Test\n}')
expected = policy_simple.Header()
expected.AddField(policy_simple.Target('Test'))
pol = parser.Parse()
self.assertEqual(expected, pol.members[0])
def testAddMember(self):
p = policy_simple.Policy('test')
good = [
policy_simple.Header(),
policy_simple.Term('test'),
policy_simple.BlankLine(),
policy_simple.CommentLine('test'),
policy_simple.Include('other_pol')
]
bad = ('', 3, lambda x: x, policy_simple.Field('test'))
for member in good:
try:
p.AddMember(member)
except TypeError:
self.fail('Policy should accept member "%s"' % member)
self.assertEqual(good, p.members)
for member in bad:
self.assertRaises(TypeError, p.AddMember, member)
def testRejectsNonField(self):
b = policy_simple.Block()
for t in ('', 3, lambda x: x, policy_simple.Header(),
policy_simple.Policy('test')):
self.assertRaises(TypeError, b.AddField, t)
