Exemple #1
0
    def test_close(self):
        close_mock = MagicMock()
        self.connection_mock.close = close_mock
        db = Database('my_db')
        db.conn = self.connection_mock

        db.close()

        close_mock.assert_called_once()