rhodecode/tests/vcs/test_getslice.py

# Copyright (C) 2010-2023 RhodeCode GmbH
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License, version 3
# (only), as published by the Free Software Foundation.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.
#
# This program is dual-licensed. If you wish to learn more about the
# RhodeCode Enterprise Edition, including its added features, Support services,
# and proprietary license terms, please see https://rhodecode.com/licenses/
import datetime

import pytest
from rhodecode.lib.vcs.nodes import FileNode
from rhodecode.tests.vcs.conftest import BackendTestMixin


@pytest.mark.usefixtures("vcs_repository_support")
class TestGetslice(BackendTestMixin):
    @classmethod
    def _get_commits(cls):
        start_date = datetime.datetime(2010, 1, 1, 20)
        for x in range(5):
            yield {
                "message": "Commit %d" % x,
                "author": "Joe Doe <[e-mail]>",
                "date": start_date + datetime.timedelta(hours=12 * x),
                "added": [
                    FileNode(b"file_%d.txt" % x, content="Foobar %d" % x),
                ],
            }

    def test__getslice__last_item_is_tip(self):
        assert list(self.repo[-1:])[0] == self.repo.get_commit()

    def test__getslice__respects_start_index(self):
        assert list(self.repo[2:]) == [self.repo.get_commit(commit_id) for commit_id in self.repo.commit_ids[2:]]

    def test__getslice__respects_negative_start_index(self):
        assert list(self.repo[-2:]) == [self.repo.get_commit(commit_id) for commit_id in self.repo.commit_ids[-2:]]

    def test__getslice__respects_end_index(self):
        assert list(self.repo[:2]) == [self.repo.get_commit(commit_id) for commit_id in self.repo.commit_ids[:2]]

    def test__getslice__respects_negative_end_index(self):
        assert list(self.repo[:-2]) == [self.repo.get_commit(commit_id) for commit_id in self.repo.commit_ids[:-2]]

    def test__getslice__start_grater_than_end(self):
        assert list(self.repo[10:0]) == []

    def test__getslice__negative_iteration(self):
        assert list(self.repo[::-1]) == [self.repo.get_commit(commit_id) for commit_id in self.repo.commit_ids[::-1]]

    def test__getslice__iterate_even(self):
        assert list(self.repo[0:10:2]) == [
            self.repo.get_commit(commit_id) for commit_id in self.repo.commit_ids[0:10:2]
        ]