def test_GetCount(self):
# checked IOError exception already so not checking that
user = User(None, "Name", "123456", "*****@*****.**", "h 43, Delhi", False)
count = 0
with mock.patch("FileHandling.readFirstLine") as mockRead, \
mock.patch("FileHandling.writeFirstLine") as mockWrite:
mockRead.return_value = count
self.assertEqual(0, user._getCount())
mockRead.assert_called_with("users/count")
mockWrite.assert_called_with("1", "users/count")
count = count + 1
mockRead.return_value = count
self.assertEqual(1, user._getCount())
mockRead.assert_called_with("users/count")
mockWrite.assert_called_with("2", "users/count")
def test__AddingUser(self):
user = User(None, "Name", "123456", "*****@*****.**", "h 43, Delhi", False)
with mock.patch("FileHandling.readFirstLine", side_effect=IOError) as mockRead, \
mock.patch("FileHandling.writeFirstLine") as mockWrite:
user.userId = user._getCount()
self.assertEqual(0, user.userId)
mockRead.assert_called_once_with("users/count")
mockWrite.assert_called_once_with("0", "users/count")
with mock.patch("FileHandling.writeFirstLine") as mockWrite:
user._writeFirstLine()
string = "[email protected]====h 43, Delhi====0"
mockWrite.assert_called_once_with(string, "users/0-Name")
