From f04a95f79f087869a44ea3d63d55ce7b27b8e922 Mon Sep 17 00:00:00 2001
From: Julien Olivain <ju.o@free.fr>
Date: Sat, 20 Jul 2024 13:45:57 +0200
Subject: [PATCH] support/testing: add mawk runtime test

Signed-off-by: Julien Olivain <ju.o@free.fr>
Signed-off-by: Thomas Petazzoni <thomas.petazzoni@bootlin.com>
---
 DEVELOPERS                                 |   1 +
 support/testing/tests/package/test_mawk.py | 114 +++++++++++++++++++++
 2 files changed, 115 insertions(+)
 create mode 100644 support/testing/tests/package/test_mawk.py

diff --git a/DEVELOPERS b/DEVELOPERS
index c4f1565edd..06eef3740f 100644
--- a/DEVELOPERS
+++ b/DEVELOPERS
@@ -1901,6 +1901,7 @@ F:	support/testing/tests/package/test_lzip.py
 F:	support/testing/tests/package/test_lsof.py
 F:	support/testing/tests/package/test_lz4.py
 F:	support/testing/tests/package/test_lzop.py
+F:	support/testing/tests/package/test_mawk.py
 F:	support/testing/tests/package/test_mdadm.py
 F:	support/testing/tests/package/test_mdadm/
 F:	support/testing/tests/package/test_micropython.py
diff --git a/support/testing/tests/package/test_mawk.py b/support/testing/tests/package/test_mawk.py
new file mode 100644
index 0000000000..5caf77c116
--- /dev/null
+++ b/support/testing/tests/package/test_mawk.py
@@ -0,0 +1,114 @@
+import os
+
+import infra.basetest
+
+
+class TestMawk(infra.basetest.BRTest):
+    config = infra.basetest.BASIC_TOOLCHAIN_CONFIG + \
+        """
+        BR2_PACKAGE_MAWK=y
+        BR2_TARGET_ROOTFS_CPIO=y
+        # BR2_TARGET_ROOTFS_TAR is not set
+        """
+
+    def basic_mawk_tests(self):
+        # Check the program can execute
+        self.assertRunOk("mawk --version")
+
+        # Check "mawk" can return a specific exit code
+        code = 123
+        cmd = "mawk 'BEGIN { exit(" + str(code) + "); }'"
+        _, exit_code = self.emulator.run(cmd)
+        self.assertEqual(exit_code, code)
+
+        # Run a basic print program
+        test_string = "Hello Buildroot"
+        cmd = "mawk 'BEGIN {print \"" + test_string + "\"; }'"
+        output, exit_code = self.emulator.run(cmd)
+        self.assertEqual(exit_code, 0)
+        self.assertEqual(output[0], test_string)
+
+    def create_test_data(self):
+        # Create some test data
+        entries = ["one", "two", "three", "four"]
+        for entry in entries:
+            self.assertRunOk(f"echo {entry} >> data1.txt")
+
+    def add_line_numbers(self):
+        # Add line numbers with mawk
+        cmd = "mawk '{ print NR \"\\t\" $1; }' data1.txt > data2.txt"
+        self.assertRunOk(cmd)
+
+    def sum_column(self):
+        # Check the sum of the first column is 1+2+3+4 == 10
+        awk_prg = "BEGIN { SUM = 0; } { SUM = SUM + $1; } END { print SUM; }"
+        cmd = f"mawk '{awk_prg}' data2.txt"
+        output, exit_code = self.emulator.run(cmd)
+        self.assertEqual(exit_code, 0)
+        self.assertEqual(int(output[0]), 10)
+
+    def uppercase_column(self):
+        # Extract only column 2 and convert it to upper case
+        cmd = "mawk '{ print toupper($2); }' data2.txt > data3.txt"
+        self.assertRunOk(cmd)
+
+        # Prepare the same output using "data1.txt" and the "tr" command,
+        # for verification
+        cmd = "tr a-z A-Z < data1.txt > data3-tr.txt"
+        self.assertRunOk(cmd)
+
+        # "mawk" and "tr" output are expected to be the same
+        self.assertRunOk("cmp data3.txt data3-tr.txt")
+
+    def mawk_head(self):
+        # Show the first 2 lines of a file
+        cmd = "mawk 'NR <= 2 { print $0; }' data2.txt > data4.txt"
+        self.assertRunOk(cmd)
+
+        # Prepare the same output using the "head" command
+        cmd = "head -2 data2.txt > data4-head.txt"
+        self.assertRunOk(cmd)
+
+        # "mawk" and "tr" output are expected to be the same
+        self.assertRunOk("cmp data4.txt data4-head.txt")
+
+    def mawk_specific(self):
+        # Use the "-W dump" mawk specific option.
+        # See: https://invisible-island.net/mawk/manpage/mawk.html
+        # We create an arbitrary awk program with an integer and
+        # string constant. We then check those constants are in the
+        # mawk "assembler" output.
+        awk_int = 12345
+        awk_str = "Buildroot"
+        awk_expr = f"print ($1 + {awk_int}) \"{awk_str}\";"
+        awk_prg = "BEGIN { " + awk_expr + " }"
+        cmd = f"mawk -W dump '{awk_prg}'"
+        output, exit_code = self.emulator.run(cmd)
+        self.assertEqual(exit_code, 0)
+        out_str = "\n".join(output)
+        self.assertIn(str(awk_int), out_str)
+        self.assertIn(awk_str, out_str)
+
+    def mawk_numeric(self):
+        value = 1234
+        squared_value = value * value
+        cmd = "mawk 'BEGIN { print sqrt(" + str(squared_value) + "); }'"
+        output, exit_code = self.emulator.run(cmd)
+        self.assertEqual(exit_code, 0)
+        self.assertEqual(int(output[0]), value)
+
+    def test_run(self):
+        cpio_file = os.path.join(self.builddir, "images", "rootfs.cpio")
+        self.emulator.boot(arch="armv5",
+                           kernel="builtin",
+                           options=["-initrd", cpio_file])
+        self.emulator.login()
+
+        self.basic_mawk_tests()
+        self.create_test_data()
+        self.add_line_numbers()
+        self.sum_column()
+        self.uppercase_column()
+        self.mawk_head()
+        self.mawk_specific()
+        self.mawk_numeric()