From 5c5526d34bfdc978ec5b39927e3ca8d87f684e3c Mon Sep 17 00:00:00 2001
From: Martin Baeuml <baeuml@kit.edu>
Date: Thu, 5 May 2011 17:33:10 +0200
Subject: [PATCH] add py.test testing functions

---
 sceneitems/annotationitem.py | 77 ++++++++++++++++++++++++++++++++++++
 1 file changed, 77 insertions(+)

diff --git a/sceneitems/annotationitem.py b/sceneitems/annotationitem.py
index 030b69b..eeb28be 100644
--- a/sceneitems/annotationitem.py
+++ b/sceneitems/annotationitem.py
@@ -360,4 +360,81 @@ ItemFactory.register('rect',     AnnotationGraphicsRectItem,  RectItemInserter)
 ItemFactory.register('point',    AnnotationGraphicsPointItem, PointItemInserter)
 ItemFactory.register('polygon',  AnnotationGraphicsRectItem,  PolygonItemInserter)
 
+### testing
+import pytest
+class MockupRectItem:    pass
+class MockupPointItem:   pass
+class MockupPolygonItem: pass
+class MockupRectItemInserter:    pass
+class MockupPointItemInserter:   pass
+class MockupPolygonItemInserter: pass
+
+def _create_factory():
+    itemfactory = AnnotationGraphicsItemFactory()
+    itemfactory.register('rect',     MockupRectItem,  MockupRectItemInserter)
+    itemfactory.register('point',    MockupPointItem, MockupPointItemInserter)
+    itemfactory.register('polygon',  MockupPolygonItem,  MockupPolygonItemInserter)
+    return itemfactory
+
+def test_register():
+    itemfactory = _create_factory()
+
+    item = itemfactory.createItem('rect')
+    assert isinstance(item, MockupRectItem)
+    item = itemfactory.createItem('point')
+    assert isinstance(item, MockupPointItem)
+    item = itemfactory.createItem('polygon')
+    assert isinstance(item, MockupPolygonItem)
+    item = itemfactory.createItem('polygon2')
+    assert item is None
+
+    inserter = itemfactory.createItemInserter('rect')
+    assert isinstance(inserter, MockupRectItemInserter)
+    inserter = itemfactory.createItemInserter('point')
+    assert isinstance(inserter, MockupPointItemInserter)
+    inserter = itemfactory.createItemInserter('polygon')
+    assert isinstance(inserter, MockupPolygonItemInserter)
+    inserter = itemfactory.createItemInserter('polygon2')
+    assert inserter is None
+
+def test_register_fail():
+    itemfactory = _create_factory()
+    with pytest.raises(Exception):
+        itemfactory.register('rect', MockupRectItem, MockupRectItemInserter)
+    with pytest.raises(Exception):
+        itemfactory.register('polygon', MockupPolygonItem)
+
+def test_register_replace():
+    itemfactory = _create_factory()
+
+    itemfactory.register('rect', MockupPolygonItem, MockupPolygonItemInserter, replace=True)
+    item = itemfactory.createItem('rect')
+    assert isinstance(item, MockupPolygonItem)
+    inserter = itemfactory.createItemInserter('rect')
+    assert isinstance(inserter, MockupPolygonItemInserter)
+
+    itemfactory.register('rect', MockupRectItem, MockupRectItemInserter, replace=True)
+    item = itemfactory.createItem('rect')
+    assert isinstance(item, MockupRectItem)
+    inserter = itemfactory.createItemInserter('rect')
+    assert isinstance(inserter, MockupRectItemInserter)
+
+def test_clear():
+    itemfactory = _create_factory()
+
+    item = itemfactory.createItem('rect')
+    assert isinstance(item, MockupRectItem)
+    itemfactory.clear('rect')
+    item = itemfactory.createItem('rect')
+    assert item is None
+    inserter = itemfactory.createItemInserter('rect')
+    assert inserter is None
+
+    item = itemfactory.createItem('point')
+    assert isinstance(item, MockupPointItem)
+    item = itemfactory.createItem('polygon')
+    assert isinstance(item, MockupPolygonItem)
+    itemfactory.clear()
+    assert itemfactory.createItem('point') is None
+    assert itemfactory.createItem('polygon') is None