1 files changed, 139 insertions, 0 deletions
diff --git a/js/src/tests/test/run.py b/js/src/tests/test/run.py
new file mode 100755
index 0000000000..a6f7a94932
--- /dev/null
+++ b/js/src/tests/test/run.py
@@ -0,0 +1,139 @@
+#!/usr/bin/env python
+# Adapted from https://github.com/tc39/test262/blob/main/tools/generation/test/run.py
+
+import contextlib
+import os
+import shutil
+import subprocess
+import tempfile
+import unittest
+
+testDir = os.path.dirname(os.path.relpath(__file__))
+OUT_DIR = os.path.join(testDir, "out")
+EXPECTED_DIR = os.path.join(testDir, "expected")
+ex = os.path.join(testDir, "..", "test262-export.py")
+importExec = os.path.join(testDir, "..", "test262-update.py")
+test262Url = "git://github.com/tc39/test262.git"
+
+
+@contextlib.contextmanager
+def TemporaryDirectory():
+    tmpDir = tempfile.mkdtemp()
+    try:
+        yield tmpDir
+    finally:
+        shutil.rmtree(tmpDir)
+
+
+class TestExport(unittest.TestCase):
+    maxDiff = None
+
+    def exportScript(self):
+        relpath = os.path.relpath(os.path.join(testDir, "fixtures", "export"))
+        sp = subprocess.Popen([ex, relpath, "--out", OUT_DIR], stdout=subprocess.PIPE)
+        stdout, stderr = sp.communicate()
+        return dict(stdout=stdout, stderr=stderr, returncode=sp.returncode)
+
+    def importLocal(self):
+        with TemporaryDirectory() as cloneDir:
+            branch = "smTempBranch"
+            # Clone Test262 to a local branch
+            subprocess.check_call(["git", "clone", "--depth=1", test262Url, cloneDir])
+            # Checkout to a new branch
+            subprocess.check_call(["git", "-C", cloneDir, "checkout", "-b", branch])
+            # Make changes on the new branch
+            # Remove test/language/export/escaped-from.js
+            subprocess.check_call(
+                ["git", "-C", cloneDir, "rm", "test/language/export/escaped-from.js"]
+            )
+            # Rename test/language/export/escaped-default.js
+            subprocess.check_call(
+                [
+                    "git",
+                    "-C",
+                    cloneDir,
+                    "mv",
+                    "test/language/export/escaped-default.js",
+                    "test/language/export/escaped-foobarbaz.js",
+                ]
+            )
+            # Copy fixtures files
+            fixturesDir = os.path.join(testDir, "fixtures", "import", "files")
+            shutil.copytree(fixturesDir, os.path.join(cloneDir, "test", "temp42"))
+            # Stage and Commit changes
+            subprocess.check_call(["git", "-C", cloneDir, "add", "."])
+            subprocess.check_call(
+                ["git", "-C", cloneDir, "commit", "-m", '"local foo"']
+            )
+
+            # Run import script
+            print("%s --local %s --out %s" % (importExec, cloneDir, OUT_DIR))
+            sp = subprocess.Popen(
+                [importExec, "--local", cloneDir, "--out", OUT_DIR],
+                stdout=subprocess.PIPE,
+                stderr=subprocess.PIPE,
+            )
+            stdoutdata, _ = sp.communicate()
+
+            return stdoutdata, sp.returncode, cloneDir
+
+    def isTestFile(self, filename):
+        return not (
+            filename.startswith(".")
+            or filename.startswith("#")
+            or filename.endswith("~")
+        )
+
+    def getFiles(self, path):
+        names = []
+        for root, _, fileNames in os.walk(path):
+            for fileName in filter(self.isTestFile, fileNames):
+                names.append(os.path.join(root, fileName))
+        names.sort()
+        return names
+
+    def compareTrees(self, targetName):
+        expectedPath = os.path.join(EXPECTED_DIR, targetName)
+        actualPath = OUT_DIR
+
+        expectedFiles = self.getFiles(expectedPath)
+        actualFiles = self.getFiles(actualPath)
+
+        self.assertListEqual(
+            map(lambda x: os.path.relpath(x, expectedPath), expectedFiles),
+            map(lambda x: os.path.relpath(x, actualPath), actualFiles),
+        )
+
+        for expectedFile, actualFile in zip(expectedFiles, actualFiles):
+            with open(expectedFile) as expectedHandle:
+                with open(actualFile) as actualHandle:
+                    self.assertMultiLineEqual(
+                        expectedHandle.read(), actualHandle.read()
+                    )
+
+    def compareContents(self, output, filePath, folder):
+        with open(filePath, "rb") as file:
+            expected = file.read()
+
+        expected = expected.replace("{{folder}}", folder)
+        self.assertMultiLineEqual(output, expected)
+
+    def tearDown(self):
+        shutil.rmtree(OUT_DIR, ignore_errors=True)
+
+    def test_export(self):
+        result = self.exportScript()
+        self.assertEqual(result["returncode"], 0)
+        self.compareTrees("export")
+
+    def test_import_local(self):
+        output, returncode, folder = self.importLocal()
+        self.assertEqual(returncode, 0)
+        self.compareTrees(os.path.join("import", "files"))
+        self.compareContents(
+            output, os.path.join(testDir, "expected", "import", "output.txt"), folder
+        )
+
+
+if __name__ == "__main__":
+    unittest.main()