From 9b445d35967b84d9fa106ee3f38fc4019f424599 Mon Sep 17 00:00:00 2001
From: Sizhe Zhao <prc.zhao@outlook.com>
Date: Sat, 7 Jun 2025 20:36:19 +0800
Subject: [PATCH] nixosTests.cntr: handleTest -> runTest

---
 nixos/tests/all-tests.nix |  8 +++++-
 nixos/tests/cntr.nix      | 60 ++++++++++++++++-----------------------
 2 files changed, 31 insertions(+), 37 deletions(-)

diff --git a/nixos/tests/all-tests.nix b/nixos/tests/all-tests.nix
index 8535c8d10062..9ef6c07f82fb 100644
--- a/nixos/tests/all-tests.nix
+++ b/nixos/tests/all-tests.nix
@@ -321,7 +321,13 @@ in
   cloud-init = runTest ./cloud-init.nix;
   cloud-init-hostname = runTest ./cloud-init-hostname.nix;
   cloudlog = runTest ./cloudlog.nix;
-  cntr = handleTestOn [ "aarch64-linux" "x86_64-linux" ] ./cntr.nix { };
+  cntr = import ./cntr.nix {
+    inherit (pkgs) lib;
+    runTest = runTestOn [
+      "aarch64-linux"
+      "x86_64-linux"
+    ];
+  };
   cockpit = runTest ./cockpit.nix;
   cockroachdb = handleTestOn [ "x86_64-linux" ] ./cockroachdb.nix { };
   code-server = runTest ./code-server.nix;
diff --git a/nixos/tests/cntr.nix b/nixos/tests/cntr.nix
index 33c21dd2252d..87abf97ed776 100644
--- a/nixos/tests/cntr.nix
+++ b/nixos/tests/cntr.nix
@@ -1,41 +1,31 @@
 # Test for cntr tool
-{
-  system ? builtins.currentSystem,
-  config ? { },
-  pkgs ? import ../.. { inherit system config; },
-  lib ? pkgs.lib,
-}:
+
+{ runTest, lib }:
 
 let
-  inherit (import ../lib/testing-python.nix { inherit system pkgs; }) makeTest;
-
   mkOCITest =
     backend:
-    makeTest {
+    runTest {
       name = "cntr-${backend}";
 
-      meta = {
-        maintainers = with lib.maintainers; [
-          sorki
-          mic92
-        ];
-      };
+      meta.maintainers = with lib.maintainers; [
+        sorki
+        mic92
+      ];
 
-      nodes = {
-        ${backend} =
-          { pkgs, ... }:
-          {
-            environment.systemPackages = [ pkgs.cntr ];
-            virtualisation.oci-containers = {
-              inherit backend;
-              containers.nginx = {
-                image = "nginx-container";
-                imageStream = pkgs.dockerTools.examples.nginxStream;
-                ports = [ "8181:80" ];
-              };
+      nodes.${backend} =
+        { pkgs, ... }:
+        {
+          environment.systemPackages = [ pkgs.cntr ];
+          virtualisation.oci-containers = {
+            inherit backend;
+            containers.nginx = {
+              image = "nginx-container";
+              imageStream = pkgs.dockerTools.examples.nginxStream;
+              ports = [ "8181:80" ];
             };
           };
-      };
+        };
 
       testScript = ''
         start_all()
@@ -53,18 +43,16 @@ let
       '';
     };
 
-  mkContainersTest = makeTest {
+  mkContainersTest = runTest {
     name = "cntr-containers";
 
-    meta = with pkgs.lib.maintainers; {
-      maintainers = [
-        sorki
-        mic92
-      ];
-    };
+    meta.maintainers = with lib.maintainers; [
+      sorki
+      mic92
+    ];
 
     nodes.machine =
-      { lib, ... }:
+      { pkgs, ... }:
       {
         environment.systemPackages = [ pkgs.cntr ];
         containers.test = {