def test_fingerprint(self):
payload = Payload()
payload.add_field('foo', PrimitiveField())
fingerprint1 = payload.fingerprint()
self.assertEqual(fingerprint1, payload.fingerprint())
payload.add_field('bar', PrimitiveField())
fingerprint2 = payload.fingerprint()
self.assertNotEqual(fingerprint1, fingerprint2)
self.assertEqual(fingerprint2, payload.fingerprint())
payload.freeze()
self.assertEqual(fingerprint2, payload.fingerprint())
def payload_for_scope(self, scope):
"""Returns a payload representing the options for the given scope."""
payload = Payload()
for (name, _, kwargs) in self.registration_args_iter_for_scope(scope):
if not kwargs.get('fingerprint', False):
continue
val = self.for_scope(scope)[name]
val_type = kwargs.get('type', '')
if val_type == Options.file:
field = FileField(val)
elif val_type == Options.target_list:
field = TargetListField(val)
else:
field = PrimitiveField(val)
payload.add_field(name, field)
payload.freeze()
return payload
def test_field_duplication(self):
payload = Payload()
payload.add_field('foo', PrimitiveField())
payload.freeze()
with self.assertRaises(PayloadFieldAlreadyDefinedError):
payload.add_field('foo', PrimitiveField())
def test_freeze(self):
payload = Payload()
payload.add_field('foo', PrimitiveField())
payload.freeze()
with self.assertRaises(PayloadFrozenError):
payload.add_field('bar', PrimitiveField())
def _make_payload_from_sources(self, sources):
p = Payload()
p.add_field('sources', SourcesField('', sources))
p.freeze()
return p
def _make_payload_from_sources(self, sources):
p = Payload()
p.add_field('sources', SourcesField('', sources))
p.freeze()
return p