Esempio n. 1
0
    def test_combine_envelope_contents(self):
        # Make some pages.
        p1 = Page()
        p1.save()
        p2 = Page()
        p2.save()
        p3 = Page()
        p3.save()
        p4 = Page()
        p4.save()

        # Make a box for them.
        b = Box()
        b.save()

        # Put them in the box.
        box.add_content( (p1, p2, p3, p4) )

        # Make two envelopes and put pages in them.
        e1 = Envelope()
        e1.add_content( (p1, p2) )
        e2 = Envelope()
        e2.add_content( (p3, p4) )

        # Combine envelope contents.
        e2.send_contents_to_envelope(e1)

        self.assertEqual(len(e2.contents.all()), 0,
                         "Some content is still left in sending envelope.")
        self.assertEqual(len(e1.contents.all()), 4,
                         "Content in receiving envelope is not correct.")
        for page in [p1, p2, p3, p4]:
            self.assertTrue(page in e1.get_content(),
                            "Page not found in receiving envelope.")
Esempio n. 2
0
    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.")