From 37d00bbcf112de941872ba24233b71fbc89aa9a5 Mon Sep 17 00:00:00 2001
From: Robert Scott <code@humanleg.org.uk>
Date: Tue, 2 Nov 2021 00:26:33 +0000
Subject: [PATCH] python3Packages.streamz: 0.6.2 -> 0.6.3

convert to pytestCheckHook, add patch to fix tests against
distributed-2021.10.0
---
 .../python-modules/streamz/default.nix        | 36 ++++++++++++-------
 1 file changed, 24 insertions(+), 12 deletions(-)

diff --git a/pkgs/development/python-modules/streamz/default.nix b/pkgs/development/python-modules/streamz/default.nix
index ee96efe5e01f..bb3e2b68cb14 100644
--- a/pkgs/development/python-modules/streamz/default.nix
+++ b/pkgs/development/python-modules/streamz/default.nix
@@ -14,6 +14,7 @@
 , toolz
 , tornado
 , zict
+, fetchpatch
 }:
 
 buildPythonPackage rec {
@@ -28,6 +29,16 @@ buildPythonPackage rec {
     sha256 = "sha256-0wZ1ldLFRAIL9R+gLfwsFbL+gvdORAkYWNjnDmeafm8=";
   };
 
+  patches = [
+    # remove with next bump
+    (fetchpatch {
+      name = "fix-tests-against-distributed-2021.10.0.patch";
+      url = "https://github.com/python-streamz/streamz/commit/5bd3bc4d305ff40c740bc2550c8491be9162778a.patch";
+      sha256 = "1xzxcbf7yninkyizrwm3ahqk6ij2fmh0454iqjx2n7mmzx3sazx7";
+      includes = ["streamz/tests/test_dask.py"];
+    })
+  ];
+
   propagatedBuildInputs = [
     networkx
     six
@@ -46,22 +57,23 @@ buildPythonPackage rec {
     requests
   ];
 
-  disabledTests = [
-    # Disable test_tcp_async because fails on sandbox build
-    "test_partition_timeout"
-    "test_tcp_async"
-    "test_tcp"
-  ];
-
-  disabledTestPaths = [
-    # Disable kafka tests
-    "streamz/tests/test_kafka.py"
-  ];
-
   pythonImportsCheck = [
     "streamz"
   ];
 
+  disabledTests = [
+    # test_tcp_async fails on sandbox build
+    "test_tcp_async"
+    "test_tcp"
+    "test_partition_timeout"
+    # flaky
+    "test_from_iterable_backpressure"
+  ];
+  disabledTestPaths = [
+    # disable kafka tests
+    "streamz/tests/test_kafka.py"
+  ];
+
   meta = with lib; {
     description = "Pipelines to manage continuous streams of data";
     homepage = "https://github.com/python-streamz/streamz";