# This Source Code Form is subject to the terms of the Mozilla Public
# License, v. 2.0. If a copy of the MPL was not distributed with this
# file, You can obtain one at http://mozilla.org/MPL/2.0/.

import os
from importlib import reload
from unittest.mock import MagicMock

import mozunit
import pytest
from tryselect import push
from tryselect.selectors import again


@pytest.fixture(autouse=True)
def patch_history_path(tmpdir, monkeypatch):
    monkeypatch.setattr(push, "history_path", tmpdir.join("history.json").strpath)
    reload(again)


def test_try_again(monkeypatch):
    monkeypatch.setattr(push, "MACH_TRY_REMOTE", "ssh://hg.mozilla.org/try")

    metrics = MagicMock()
    push.push_to_try(
        "fuzzy",
        "Fuzzy message",
        metrics,
        try_task_config=push.generate_try_task_config(
            "fuzzy",
            ["foo", "bar"],
            {"try_task_config": {"use-artifact-builds": True}},
        ),
        # Push to VCS instead of Lando so the push fails but history is generated.
        push_to_vcs=True,
    )

    assert os.path.isfile(push.history_path)
    with open(push.history_path) as fh:
        assert len(fh.readlines()) == 1

    def fake_push_to_try(*args, **kwargs):
        return args, kwargs

    monkeypatch.setattr(push, "push_to_try", fake_push_to_try)
    reload(again)

    args, kwargs = again.run(metrics)

    assert args[0] == "again"
    assert args[1] == "Fuzzy message"

    try_task_config = kwargs["try_task_config"]["parameters"].pop("try_task_config")
    assert sorted(try_task_config.get("tasks")) == sorted(["foo", "bar"])
    assert try_task_config.get("env") == {"TRY_SELECTOR": "fuzzy"}
    assert try_task_config.get("use-artifact-builds")

    with open(push.history_path) as fh:
        assert len(fh.readlines()) == 1


def test_no_push_does_not_generate_history(tmpdir):
    assert not os.path.isfile(push.history_path)

    metrics = MagicMock()
    push.push_to_try(
        "fuzzy",
        "Fuzzy",
        metrics,
        try_task_config=push.generate_try_task_config(
            "fuzzy",
            ["foo", "bar"],
            {"use-artifact-builds": True},
        ),
        dry_run=True,
    )
    assert not os.path.isfile(push.history_path)
    assert again.run(metrics) == 1


if __name__ == "__main__":
    mozunit.main()