コード例 #1
0
ファイル: test_document.py プロジェクト: HughP/nostaples
 def setUp(self):
     self.mock_application = Mock(spec=Application)
     self.document_model = DocumentModel(self.mock_application)
コード例 #2
0
ファイル: test_document.py プロジェクト: HughP/nostaples
class TestDocumentModel(unittest.TestCase):
    def setUp(self):
        self.mock_application = Mock(spec=Application)
        self.document_model = DocumentModel(self.mock_application)
    
    def tearDown(self):
        self.mock_application = None
        self.document_model = None
    
    def test_append(self):
        self.assertEqual(self.document_model.count, 0)
        
        self.document_model.append(PageModel(self.mock_application))
        self.document_model.append(PageModel(self.mock_application))
        p = PageModel(self.mock_application)
        self.document_model.append(p)
        
        self.assertEqual(self.document_model.count, 3)
        self.assertEqual(self.document_model[2][0], p)
        
    def test_prepend(self):
        self.assertEqual(self.document_model.count, 0)
        
        self.document_model.prepend(PageModel(self.mock_application))
        self.document_model.prepend(PageModel(self.mock_application))
        p = PageModel(self.mock_application)
        self.document_model.prepend(p)
        
        self.assertEqual(self.document_model.count, 3)
        self.assertEqual(self.document_model[0][0], p)
        
    def test_insert(self):
        self.assertEqual(self.document_model.count, 0)
        
        p0 = PageModel(self.mock_application)
        p1 = PageModel(self.mock_application)
        p2 = PageModel(self.mock_application)
        
        self.document_model.insert(0, p2)
        self.document_model.insert(0, p0)
        self.document_model.insert(1, p1)
        
        self.assertEqual(self.document_model.count, 3)
        self.assertEqual(self.document_model[0][0], p0)
        self.assertEqual(self.document_model[1][0], p1)
        self.assertEqual(self.document_model[2][0], p2)
        
    def test_insert_before(self):
        self.assertEqual(self.document_model.count, 0)
        
        p0 = PageModel(self.mock_application)
        p1 = PageModel(self.mock_application)
        p2 = PageModel(self.mock_application)
                
        iter = self.document_model.get_iter_root()
        self.document_model.insert_before(iter, p2)
        iter = self.document_model.get_iter(0)
        self.document_model.insert_before(iter, p0)
        iter = self.document_model.get_iter(1)
        self.document_model.insert_before(iter, p1)
        
        self.assertEqual(self.document_model.count, 3)
        self.assertEqual(self.document_model[0][0], p0)
        self.assertEqual(self.document_model[1][0], p1)
        self.assertEqual(self.document_model[2][0], p2)
        
    def test_insert_after(self):
        self.assertEqual(self.document_model.count, 0)
        
        p0 = PageModel(self.mock_application)
        p1 = PageModel(self.mock_application)
        p2 = PageModel(self.mock_application)
                
        iter = self.document_model.get_iter_root()
        self.document_model.insert_after(iter, p0)
        iter = self.document_model.get_iter(0)
        self.document_model.insert_after(iter, p2)
        iter = self.document_model.get_iter(0)
        self.document_model.insert_after(iter, p1)
        
        self.assertEqual(self.document_model.count, 3)
        self.assertEqual(self.document_model[0][0], p0)
        self.assertEqual(self.document_model[1][0], p1)
        self.assertEqual(self.document_model[2][0], p2)
        
    def test_remove(self):
        self.assertEqual(self.document_model.count, 0)
        
        p0 = PageModel(self.mock_application)
        p1 = PageModel(self.mock_application)
        p2 = PageModel(self.mock_application)
                
        self.document_model.append(p0)
        self.document_model.append(p1)
        self.document_model.append(p2)
        
        iter = self.document_model.get_iter(1)
        self.document_model.remove(iter)
        
        self.assertEqual(self.document_model.count, 2)
        self.assertEqual(self.document_model[0][0], p0)
        self.assertEqual(self.document_model[1][0], p2)
        
    def test_clear(self):
        self.assertEqual(self.document_model.count, 0)
        
        p0 = PageModel(self.mock_application)
        p1 = PageModel(self.mock_application)
        p2 = PageModel(self.mock_application)
                
        self.document_model.append(p0)
        self.document_model.append(p1)
        self.document_model.append(p2)
        
        self.document_model.clear()
        
        self.assertEqual(self.document_model.count, 0)
        self.assertRaises(ValueError, self.document_model.get_iter, 0)