def test_commit(self): self.connection_mock.commit = MagicMock() db = Database('my_db') db.conn = self.connection_mock db.commit() self.connection_mock.commit.assert_called_once()
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()