From 24ad2c84c82427d5309e33fab37c30d530565a31 Mon Sep 17 00:00:00 2001
From: Fabian Affolter <mail@fabian-affolter.ch>
Date: Sun, 3 Jan 2021 22:53:04 +0100
Subject: [PATCH] wfuzz: switch to pytestCheckHook

---
 pkgs/development/python-modules/wfuzz/default.nix | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/pkgs/development/python-modules/wfuzz/default.nix b/pkgs/development/python-modules/wfuzz/default.nix
index f6a86b7b5ed5..f42df0054952 100644
--- a/pkgs/development/python-modules/wfuzz/default.nix
+++ b/pkgs/development/python-modules/wfuzz/default.nix
@@ -12,6 +12,7 @@
 , pycurl
 , pyparsing
 , pytest
+, pytestCheckHook
 , setuptools
 , six
 , stdenv
@@ -44,15 +45,15 @@ buildPythonPackage rec {
   checkInputs = [
     netaddr
     pytest
+    pytestCheckHook
   ] ++ lib.optionals isPy27 [
     mock
   ];
 
+  preCheck = "export HOME=$(mktemp -d)";
   # The skipped tests are requiring a local web server
-  checkPhase = ''
-    HOME=$TMPDIR pytest \
-      tests/test_{moduleman,filterintro,reqresp,api,clparser}.py
-  '';
+  pytestFlagsArray = [ "tests/test_{moduleman,filterintro,reqresp,api,clparser}.py" ];
+  pythonImportsCheck = [ "wfuzz" ];
 
   meta = with lib; {
     description = "Web content fuzzer to facilitate web applications assessments";