From f2bd5571863ef967bd730e8713efd69e3293ebdc Mon Sep 17 00:00:00 2001
From: Jim Blandy <jimb@red-bean.com>
Date: Sun, 17 Sep 2023 11:08:42 -0700
Subject: [PATCH] Tests for wgpu#4139. (#4148)

---
 tests/tests/query_set.rs | 26 ++++++++++++++++++++++++++
 tests/tests/root.rs      |  1 +
 2 files changed, 27 insertions(+)
 create mode 100644 tests/tests/query_set.rs

diff --git a/tests/tests/query_set.rs b/tests/tests/query_set.rs
new file mode 100644
index 000000000..16e509408
--- /dev/null
+++ b/tests/tests/query_set.rs
@@ -0,0 +1,26 @@
+use wgpu_test::{initialize_test, FailureCase, TestParameters};
+
+#[test]
+fn drop_failed_timestamp_query_set() {
+    let parameters = TestParameters::default()
+        // https://github.com/gfx-rs/wgpu/issues/4139
+        .expect_fail(FailureCase::always());
+    initialize_test(parameters, |ctx| {
+        // Enter an error scope, so the validation catch-all doesn't
+        // report the error too early.
+        ctx.device.push_error_scope(wgpu::ErrorFilter::Validation);
+
+        // Creating this query set should fail, since we didn't include
+        // TIMESTAMP_QUERY in our required features.
+        let bad_query_set = ctx.device.create_query_set(&wgpu::QuerySetDescriptor {
+            label: Some("doomed query set"),
+            ty: wgpu::QueryType::Timestamp,
+            count: 1,
+        });
+
+        // Dropping this should not panic.
+        drop(bad_query_set);
+
+        assert!(pollster::block_on(ctx.device.pop_error_scope()).is_some());
+    });
+}
diff --git a/tests/tests/root.rs b/tests/tests/root.rs
index 85901ae49..b2695fd82 100644
--- a/tests/tests/root.rs
+++ b/tests/tests/root.rs
@@ -20,6 +20,7 @@ mod instance;
 mod occlusion_query;
 mod partially_bounded_arrays;
 mod poll;
+mod query_set;
 mod queue_transfer;
 mod resource_descriptor_accessor;
 mod resource_error;