From 59a848c5929cbfca45d95860eb60dfebd0786c94 Mon Sep 17 00:00:00 2001
From: Gonéri Le Bouder <goneri@lebouder.net>
Date: Mon, 14 Jun 2021 15:39:05 -0400
Subject: add DragonFlyBSD support (#904)

- Mostly based on FreeBSD, the main exception is that
  `find_devs_with_on_freebsd` does not work.
- Since we cannot get the CDROM or the partition labels,
  `find_devs_with_on_dragonflybsd()` has a more naive approach and
  returns all the block devices.
---
 tests/unittests/test_distros/test_dragonflybsd.py | 25 +++++++++++++++++++++++
 tests/unittests/test_util.py                      | 18 ++++++++++++++++
 2 files changed, 43 insertions(+)
 create mode 100644 tests/unittests/test_distros/test_dragonflybsd.py

(limited to 'tests/unittests')

diff --git a/tests/unittests/test_distros/test_dragonflybsd.py b/tests/unittests/test_distros/test_dragonflybsd.py
new file mode 100644
index 00000000..df2c00f4
--- /dev/null
+++ b/tests/unittests/test_distros/test_dragonflybsd.py
@@ -0,0 +1,25 @@
+#!/usr/bin/env python3
+
+
+import cloudinit.util
+from cloudinit.tests.helpers import mock
+
+
+def test_find_dragonflybsd_part():
+    assert cloudinit.util.find_dragonflybsd_part("/dev/vbd0s3") == "vbd0s3"
+
+
+@mock.patch("cloudinit.util.is_DragonFlyBSD")
+@mock.patch("cloudinit.subp.subp")
+def test_parse_mount(mock_subp, m_is_DragonFlyBSD):
+    mount_out = """
+vbd0s3 on / (hammer2, local)
+devfs on /dev (devfs, nosymfollow, local)
+/dev/vbd0s0a on /boot (ufs, local)
+procfs on /proc (procfs, local)
+tmpfs on /var/run/shm (tmpfs, local)
+"""
+
+    mock_subp.return_value = (mount_out, "")
+    m_is_DragonFlyBSD.return_value = True
+    assert cloudinit.util.parse_mount("/") == ("vbd0s3", "hammer2", "/")
diff --git a/tests/unittests/test_util.py b/tests/unittests/test_util.py
index e5292001..2290cab7 100644
--- a/tests/unittests/test_util.py
+++ b/tests/unittests/test_util.py
@@ -999,4 +999,22 @@ class TestFindDevs:
         devlist = util.find_devs_with_netbsd(criteria=criteria)
         assert devlist == expected_devlist
 
+    @pytest.mark.parametrize(
+        'criteria,expected_devlist', (
+            (None, ['/dev/vbd0', '/dev/cd0', '/dev/acd0']),
+            ('TYPE=iso9660', ['/dev/cd0', '/dev/acd0']),
+            ('TYPE=vfat', ['/dev/vbd0']),
+            ('LABEL_FATBOOT=A_LABEL',  # lp: #1841466
+             ['/dev/vbd0', '/dev/cd0', '/dev/acd0']),
+        )
+    )
+    @mock.patch("cloudinit.subp.subp")
+    def test_find_devs_with_dragonflybsd(self, m_subp, criteria,
+                                         expected_devlist):
+        m_subp.return_value = (
+            'md2 md1 cd0 vbd0 acd0 vn3 vn2 vn1 vn0 md0', ''
+        )
+        devlist = util.find_devs_with_dragonflybsd(criteria=criteria)
+        assert devlist == expected_devlist
+
 # vi: ts=4 expandtab
-- 
cgit v1.2.3