Revision: 2632
Author: janne.t.harkonen
Date: Mon Mar 15 13:37:36 2010
Log: Fix unit tests
http://code.google.com/p/robotframework/source/detail?r=2632
Modified:
/trunk/proto/atdd_example/test_vacalc.py
/trunk/proto/atdd_example/vacalc.py
=======================================
--- /trunk/proto/atdd_example/test_vacalc.py Mon Mar 15 06:45:32 2010
+++ /trunk/proto/atdd_example/test_vacalc.py Mon Mar 15 13:37:36 2010
@@ -1,30 +1,30 @@
import datetime
import unittest
-from vacalc import User, UserStore, DuplicateUser
+from vacalc import Employee, EmployeeStore, VacalcError
-class TestUser(unittest.TestCase):
-
- def test_creating_user(self):
- user = User('Juan von Rantanen', '2010-3-15')
- self.assertEquals(user.name, 'Juan von Rantanen')
- self.assertEquals(user.startdate, datetime.date(2010, 3, 15))
+class TestEmployee(unittest.TestCase):
+
+ def test_creating_employee(self):
+ employee = Employee('Juan von Rantanen', '2010-3-15')
+ self.assertEquals(employee.name, 'Juan von Rantanen')
+ self.assertEquals(employee.startdate, datetime.date(2010, 3, 15))
-class TestUserStore(unittest.TestCase):
-
- def test_adding_user(self):
- store = UserStore()
- user = store.add_user('Test User Store', '2000-12-24')
- self.assertEquals(user.name, 'Test User Store')
- self.assertEquals(user.startdate, datetime.date(2000, 12, 24))
- self.assertEquals(store._users, {user.name: user})
-
- def test_adding_duplicate_user(self):
- store = UserStore()
- user = store.add_user('test', '2000-12-24')
- self.assertRaises(DuplicateUser,
store.add_user, 'test', '2001-01-24')
+class TestEmployeeStore(unittest.TestCase):
+
+ def test_adding_employee(self):
+ store = EmployeeStore(None)
+ employee = store.add_employee('Test Employee Store', '2000-12-24')
+ self.assertEquals(employee.name, 'Test Employee Store')
+ self.assertEquals(employee.startdate, datetime.date(2000, 12, 24))
+ self.assertEquals(store._employees, {employee.name: employee})
+
+ def test_adding_duplicate_employee(self):
+ store = EmployeeStore(None)
+ employee = store.add_employee('test', '2000-12-24')
+ self.assertRaises(VacalcError,
store.add_employee, 'test', '2001-01-24')
if __name__ == '__main__':
=======================================
--- /trunk/proto/atdd_example/vacalc.py Mon Mar 15 13:37:29 2010
+++ /trunk/proto/atdd_example/vacalc.py Mon Mar 15 13:37:36 2010
@@ -39,6 +39,8 @@
return employee
def _serialize(self, employee):
+ if not self._db_file:
+ return
with open(self._db_file, 'a') as db:
writer = csv.writer(db, lineterminator='\n')
writer.writerow([employee.name,
employee.startdate.isoformat()])
