Revision: 2634
Author: janne.t.harkonen
Date: Mon Mar 15 13:54:40 2010
Log: Added tests for get employee
http://code.google.com/p/robotframework/source/detail?r=2634
Modified:
/trunk/proto/atdd_example/test_vacalc.py
/trunk/proto/atdd_example/vacalc.py
=======================================
--- /trunk/proto/atdd_example/test_vacalc.py Mon Mar 15 13:37:36 2010
+++ /trunk/proto/atdd_example/test_vacalc.py Mon Mar 15 13:54:40 2010
@@ -26,6 +26,16 @@
employee = store.add_employee('test', '2000-12-24')
self.assertRaises(VacalcError,
store.add_employee, 'test', '2001-01-24')
+ def test_getting_employee(self):
+ store = EmployeeStore(None)
+ employee = store.add_employee('Mr Foo Bar', '1990-02-03')
+ self.assertEquals(store.get_employee('Mr Foo Bar'), employee)
+
+ def test_get_missing_employee(self):
+ store = EmployeeStore(None)
+ self.assertRaises(VacalcError, store.get_employee, 'I am not here')
+
+
if __name__ == '__main__':
unittest.main()
=======================================
--- /trunk/proto/atdd_example/vacalc.py Mon Mar 15 13:49:01 2010
+++ /trunk/proto/atdd_example/vacalc.py Mon Mar 15 13:54:40 2010
@@ -26,7 +26,10 @@
return employees
def get_employee(self, name):
- return self._employees[name]
+ try:
+ return self._employees[name]
+ except KeyError:
+ raise VacalcError("Employee '%s' not found" % name)
def add_employee(self, name, startdate):
if name in self._employees:
