def test___init__(self, mock_warning):
Persistence()
mock_warning.assert_called_with(
"This persistence can`t access the database.")
mock_session = mock.Mock()
Persistence(mock_session)
mock_session.get_session.assert_called_with()
def test_add_db(self, mock_flush):
p = Persistence()
self.assertFalse(p.add_db())
mock_session = mock.Mock()
mock_sessionobj = mock.Mock()
mock_sessionobj.get_session.return_value = mock_session
p = Persistence(mock_sessionobj)
self.assertTrue(p.add_db())
mock_flush.assert_called_with()
mock_session.add.assert_called_with(Base)
mock_session.commit.assert_called_with()
mock_session.add.side_effect = IntegrityError(1, 2, 3)
self.assertFalse(p.add_db())
mock_session.rollback.assert_called_with()
def setUp(self):
mock_session = mock.Mock()
mock_session.get_session.return_value = "session"
self.persistence = Persistence(mock_session)
