feat: back up and restore Unix domain sockets

2 files changed, 9 insertions, 0 deletions

diff --git a/subplot/data.py b/subplot/data.py
index f7fb903..3c369d1 100644
--- a/subplot/data.py
+++ b/subplot/data.py
@@ -2,6 +2,7 @@ import json
 import logging
 import os
 import random
+import socket
 import yaml
 
 
@@ -18,6 +19,11 @@ def create_file_with_random_data(ctx, filename=None):
     create_file_with_given_data(ctx, filename=filename, data=data)
 
 
+def create_unix_socket(ctx, filename=None):
+    fd = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
+    fd.bind(filename)
+
+
 def create_nonutf8_filename(ctx, dirname=None):
     filename = "\x88"
     os.mkdir(dirname)
diff --git a/subplot/data.yaml b/subplot/data.yaml
index 1636e77..97e6eed 100644
--- a/subplot/data.yaml
+++ b/subplot/data.yaml
@@ -4,6 +4,9 @@
 - given: "a file {filename} containing some random data"
   function: create_file_with_random_data
 
+- given: "a Unix socket {filename}"
+  function: create_unix_socket
+
 - given: "a file in {dirname} with a non-UTF8 filename"
   function: create_nonutf8_filename