it: test wlr-data-control

src/it/test_ifs/test_data_control_offer.rs

@@ -0,0 +1,64 @@
+use {
+    crate::{
+        it::{
+            test_error::{TestError, TestResult},
+            test_object::TestObject,
+            test_transport::TestTransport,
+            testrun::ParseFull,
+        },
+        utils::buffd::MsgParser,
+        wire::{zwlr_data_control_offer_v1::*, ZwlrDataControlOfferV1Id},
+    },
+    ahash::AHashSet,
+    std::{
+        cell::{Cell, RefCell},
+        rc::Rc,
+    },
+    uapi::{c, OwnedFd},
+};
+
+pub struct TestDataControlOffer {
+    pub id: ZwlrDataControlOfferV1Id,
+    pub tran: Rc<TestTransport>,
+    pub destroyed: Cell<bool>,
+    pub offers: RefCell<AHashSet<String>>,
+}
+
+impl TestDataControlOffer {
+    pub fn destroy(&self) -> TestResult {
+        if !self.destroyed.replace(true) {
+            self.tran.send(Destroy { self_id: self.id })?;
+        }
+        Ok(())
+    }
+
+    pub fn receive(&self, mime_type: &str) -> TestResult<Rc<OwnedFd>> {
+        let (read, write) = uapi::pipe2(c::O_CLOEXEC)?;
+        self.tran.send(Receive {
+            self_id: self.id,
+            mime_type,
+            fd: Rc::new(write),
+        })?;
+        Ok(Rc::new(read))
+    }
+
+    fn handle_offer(&self, parser: MsgParser<'_, '_>) -> Result<(), TestError> {
+        let ev = Offer::parse_full(parser)?;
+        self.offers.borrow_mut().insert(ev.mime_type.to_string());
+        Ok(())
+    }
+}
+
+impl Drop for TestDataControlOffer {
+    fn drop(&mut self) {
+        let _ = self.destroy();
+    }
+}
+
+test_object! {
+    TestDataControlOffer, ZwlrDataControlOfferV1;
+
+    OFFER => handle_offer,
+}
+
+impl TestObject for TestDataControlOffer {}