def test_envelope_descriptor(self):
e = Envelope()
self.assertTrue(e.descriptor is None, "Descriptor not initialized to None.")
p = Page()
p.title = 'TestEnvelope 1'
p.data = 'This Page is used as the descriptor for TestEnvelope 1.'
p.save()
e.descriptor = p
e.save()
self.assertTrue(e.descriptor == p, "Descriptor not saved.")
self.assertTrue(p.describes_envelope == b, "Descriptor does not point back to envelope.")
def test_envelope_contents(self):
e = Envelope()
e.save()
self.assertEqual(len(e.contents), 0, "Envelope should start out empty.")
p1 = Page()
p1.title = 'Test Page 1'
e.add_content(p1)
p2 = Page()
p2.title = 'Test Page 2'
e.add_content(p2)
self.assertTrue(e.contains_content(p1), "Page 1 not found in envelope.")
self.assertTrue(e.contains_content(p2), "Page 2 not found in envelope.")
e.remove_content(p1)
self.assertFalse(e.contains_content(p1), "Page 1 not removed from envelope.")
e.remove_content(p2)
self.assertFalse(e.contains_content(p2), "Page 2 not removed from envelope.")
self.assertEqual(len(e.contents), 0, "Envelope should now be empty.")
def test_new_envelope(self):
e = Envelope()
e.save()
self.assertTrue(e.pk > 0, "Envelope not created.")
