def test_move_contents_together(self):
# Make some test pages.
p1 = Page()
p1.title = 'Test Page 1'
p1.save()
p2 = Page()
p2.title = 'Test Page 2'
p2.save()
# Make two test boxes.
b1 = Box()
b1.save()
b2 = Box()
b2.save()
# Put two pages in box 1.
b1.add_content( (p1, p2) )
# Put two pages in the same box into an envelope.
e.add_content( (p1, p2) )
# Move the envelope to box 2.
e.move_to_box(b2)
self.assertFalse(b1.contains_content(p1), "Page 1 did not leave box 1.")
self.assertFalse(b1.contains_content(p2), "Page 2 did not leave box 1.")
self.assertTrue(b2.contains_content(p1), "Page 1 did not move to box 2.")
self.assertTrue(b2.contains_content(p2), "Page 2 did not move to box 2.")
def test_box_contents(self):
b = Box()
b.save()
self.assertEqual(len(b.contents.all()), 0, "Box should start out empty.")
p1 = Page()
p1.title = 'Test Page 1'
b.add_content(p1)
p2 = Page()
p2.title = 'Test Page 2'
b.add_content(p2)
self.assertTrue(b.contains_content(p1), "Page 1 not found in box.")
self.assertTrue(b.contains_content(p2), "Page 2 not found in box.")
b.remove_content(p1)
self.assertFalse(b.contains_content(p1), "Page 1 not removed from box.")
b.remove_content(p2)
self.assertFalse(b.contains_content(p2), "Page 2 not removed from box.")
self.assertEqual(len(b.contents.all()), 0, "Box should now be empty.")
