pytorch/tools/test/test_extract_scripts.py

import unittest

from tools import extract_scripts

requirements_sh = """
#!/usr/bin/env bash
set -eo pipefail
pip install -r requirements.txt
""".strip()

hello_sh = """
#!/usr/bin/env sh
set -e
echo hello world
""".strip()


class TestExtractScripts(unittest.TestCase):
    def test_extract_none(self) -> None:
        self.assertEqual(
            extract_scripts.extract(
                {
                    "name": "Checkout PyTorch",
                    "uses": "zhouzhuojie/checkout@05b13c9a0d21f08f6d5e64a1d5042246d13619d9",
                }
            ),
            None,
        )

    def test_extract_run_default_bash(self) -> None:
        self.assertEqual(
            extract_scripts.extract(
                {
                    "name": "Install requirements",
                    "run": "pip install -r requirements.txt",
                }
            ),
            {
                "extension": ".sh",
                "script": requirements_sh,
            },
        )

    def test_extract_run_sh(self) -> None:
        self.assertEqual(
            extract_scripts.extract(
                {
                    "name": "Hello world",
                    "run": "echo hello world",
                    "shell": "sh",
                }
            ),
            {
                "extension": ".sh",
                "script": hello_sh,
            },
        )

    def test_extract_run_py(self) -> None:
        self.assertEqual(
            extract_scripts.extract(
                {
                    "name": "Hello world",
                    "run": 'print("Hello!")',
                    "shell": "python",
                }
            ),
            {
                "extension": ".py",
                "script": 'print("Hello!")',
            },
        )

    def test_extract_github_script(self) -> None:
        self.assertEqual(
            # https://github.com/actions/github-script/tree/v3.1.1#reading-step-results
            extract_scripts.extract(
                {
                    "uses": "actions/github-script@v3",
                    "id": "set-result",
                    "with": {
                        "script": 'return "Hello!"',
                        "result-encoding": "string",
                    },
                }
            ),
            {
                "extension": ".js",
                "script": 'return "Hello!"',
            },
        )


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