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.")