From c4bc177658d2435454d83edfb35f60000e23fd3a Mon Sep 17 00:00:00 2001
From: Michal Sojka <michal.sojka@cvut.cz>
Date: Mon, 16 Sep 2024 08:19:34 +0200
Subject: [PATCH] Add --do-check

---
 ros2nix/nix_expression.py | 5 +++++
 ros2nix/ros2nix.py        | 8 ++++++++
 2 files changed, 13 insertions(+)

diff --git a/ros2nix/nix_expression.py b/ros2nix/nix_expression.py
index a365c39..7ca6719 100644
--- a/ros2nix/nix_expression.py
+++ b/ros2nix/nix_expression.py
@@ -94,12 +94,14 @@ class NixExpression:
                  propagated_native_build_inputs: Set[str] = set(),
                  src_param: Optional[str] = None,
                  source_root: Optional[str] = None,
+                 do_check: Optional[bool] = None,
                  ) -> None:
         self.name = name
         self.version = version
         self.src_param = src_param
         self.src_expr = src_expr
         self.source_root = source_root
+        self.do_check = do_check
 
         self.description = description
         self.licenses = licenses
@@ -170,6 +172,9 @@ class NixExpression:
         if self.source_root:
             ret += f'  sourceRoot = "{self.source_root}";\n'
 
+        if self.do_check is not None:
+            ret += f'  doCheck = {"true" if self.do_check else "false"};\n'
+
         if self.build_inputs:
             ret += "  buildInputs = {};\n" \
                 .format(self._to_nix_list(sorted(self.build_inputs)))
diff --git a/ros2nix/ros2nix.py b/ros2nix/ros2nix.py
index 1f7f985..018d8af 100755
--- a/ros2nix/ros2nix.py
+++ b/ros2nix/ros2nix.py
@@ -226,6 +226,11 @@ def ros2nix(args):
         help="Set sourceRoot attribute value in the generated Nix expression. "
         "Substring '{package_name}' gets replaced with the package name.",
     )
+    parser.add_argument(
+        "--do-check",
+        action="store_true",
+        help="Set doCheck attribute to true",
+    )
 
     parser.add_argument(
         "--extra-build-inputs", type=comma_separated, metavar="DEP1,DEP2,...", default=[],
@@ -369,6 +374,9 @@ def ros2nix(args):
             if args.source_root:
                 kwargs["source_root"] = args.source_root.replace('{package_name}', pkg.name)
 
+            if args.do_check:
+                kwargs["do_check"] = True
+
             derivation = NixExpression(
                 name=NixPackage.normalize_name(pkg.name),
                 version=pkg.version,