"""Unit tests for the fetch_test_lifecycle.py script."""

import unittest

import buildscripts.fetch_test_lifecycle as fetch


class TestFetchTestLifecycle(unittest.TestCase):
    def test_get_metadata_revision(self):
        metadata_repo = MockMetadataRepository([("metadata_revision_05", "mongo_revision_06"),
                                                ("metadata_revision_04", "mongo_revision_06"),
                                                ("metadata_revision_03", "mongo_revision_02"),
                                                ("metadata_revision_02", "mongo_revision_02"),
                                                ("metadata_revision_01", None)])

        mongo_repo = MockMongoRepository(["mongo_revision_07",
                                          "mongo_revision_06",
                                          "mongo_revision_05",
                                          "mongo_revision_04",
                                          "mongo_revision_03",
                                          "mongo_revision_02",
                                          "mongo_revision_01"])

        self._check_metadata_revision(metadata_repo, mongo_repo,
                                      "mongo_revision_07",
                                      "metadata_revision_05")

        self._check_metadata_revision(metadata_repo, mongo_repo,
                                      "mongo_revision_06",
                                      "metadata_revision_05")

        self._check_metadata_revision(metadata_repo, mongo_repo,
                                      "mongo_revision_05",
                                      "metadata_revision_03")

        self._check_metadata_revision(metadata_repo, mongo_repo,
                                      "mongo_revision_04",
                                      "metadata_revision_03")

        self._check_metadata_revision(metadata_repo, mongo_repo,
                                      "mongo_revision_03",
                                      "metadata_revision_03")

        self._check_metadata_revision(metadata_repo, mongo_repo,
                                      "mongo_revision_02",
                                      "metadata_revision_03")

        self._check_metadata_revision(metadata_repo, mongo_repo,
                                      "mongo_revision_01",
                                      None)

    def _check_metadata_revision(self, metadata_repo, mongo_repo, mongo_revision,
                                 expected_metadata_revision):
            metadata_revision = fetch._get_metadata_revision(metadata_repo, mongo_repo, "project",
                                                             mongo_revision)
            self.assertEqual(expected_metadata_revision, metadata_revision)


class MockMongoRepository(object):
    def __init__(self, revisions):
        self.revisions = revisions

    def is_ancestor(self, parent, child):
        return (parent in self.revisions and child in self.revisions and
                self.revisions.index(parent) >= self.revisions.index(child))


class MockMetadataRepository(object):
    def __init__(self, references_revisions):
        self.references_revisions = references_revisions

    def list_revisions(self):
        return [r[0] for r in self.references_revisions]

    def get_reference(self, revision, project):
        for (metadata_revision, mongo_revision) in self.references_revisions:
            if metadata_revision == revision:
                return mongo_revision
        return None