From e1091ffc95cd1769c31c6ecf748613dd3cd5ce3b Mon Sep 17 00:00:00 2001
From: Anton Hvornum <anton@hvornum.se>
Date: Sun, 1 May 2022 18:39:02 +0200
Subject: [PATCH] Added BlockDevice() tests

---
 tests/disk-related/test_stat_blockdev.py | 117 +++++++++++++++++++++++
 1 file changed, 117 insertions(+)
 create mode 100644 tests/disk-related/test_stat_blockdev.py

diff --git a/tests/disk-related/test_stat_blockdev.py b/tests/disk-related/test_stat_blockdev.py
new file mode 100644
index 00000000..0128d8ed
--- /dev/null
+++ b/tests/disk-related/test_stat_blockdev.py
@@ -0,0 +1,117 @@
+import pytest
+import subprocess
+import string
+import random
+import pathlib
+
+def simple_exec(cmd):
+	proc = subprocess.Popen(
+		cmd,
+		shell=True,
+		stdout=subprocess.PIPE,
+		stderr=subprocess.STDOUT
+	)
+
+	while proc.poll() is None:
+		pass
+
+	result = proc.stdout.read()
+	proc.stdout.close()
+
+	return {'exit_code' : proc.poll(), 'data' : result.decode().strip()}
+
+def random_filename():
+	return ''.join([random.choice(string.ascii_letters) for x in range(20)]) + '.img'
+
+def truncate_file(filename):
+	result = simple_exec(f"truncate -s 20G {filename}")
+
+	if not result['exit_code'] == 0:
+		raise AssertionError(f"Could not generate a testimage with truncate: {result['data']}")
+
+	return filename
+
+def get_loopdev(filename):
+	result = simple_exec(f"""losetup -a | grep "{filename}" | awk -F ":" '{{print $1}}'""")
+	return result['data']
+
+def detach_loopdev(path):
+	result = simple_exec(f"losetup -d {path}")
+	return result['exit_code'] == 0
+
+def create_loopdev(path):
+	result = simple_exec(f"losetup -fP {path}")
+	return result['exit_code'] == 0
+
+def test_stat_blockdev():
+	import archinstall
+
+	filename = pathlib.Path(random_filename()).resolve()
+	if loopdev := get_loopdev(filename):
+		if not detach_loopdev(loopdev):
+			raise AssertionError(f"Could not detach {loopdev} before performing test with {filename}.")
+
+	truncate_file(filename)
+	if not create_loopdev(filename):
+		raise AssertionError(f"Could not create a loopdev for {filename}")
+
+	if loopdev := get_loopdev(filename):
+		# Actual test starts here:
+		block_device = archinstall.BlockDevice(loopdev)
+
+		# Make sure the backfile reported by BlockDevice() is the same we mounted
+		if block_device.device_or_backfile != str(filename):
+			raise AssertionError(f"archinstall.BlockDevice().device_or_backfile differs from loopdev path: {block_device.device_or_backfile} vs {filename}")
+
+		# Make sure the device path equals to the device we setup (/dev/loop0)
+		if block_device.device != loopdev:
+			raise AssertionError(f"archinstall.BlockDevice().device difers from {loopdev}")
+
+		# Check that the BlockDevice is clear of partitions
+		if block_device.partitions:
+			raise AssertionError(f"BlockDevice().partitions reported partitions, despire being a new trunkfile")
+
+		if block_device.has_partitions():
+			raise AssertionError(f"BlockDevice().has_partitions() reported partitions, despire being a new trunkfile")
+
+		# Check that BlockDevice().size returns a float of the size in GB
+		if block_device.size != 20.0:
+			raise AssertionError(f"The size reported by BlockDevice().size is not 20.0 as expected")
+
+		if block_device.bus_type != None:
+			raise AssertionError(f"The .bus_type of the loopdev is something other than the expected None: {block_device.bus_type}")
+
+		if block_device.spinning != False:
+			raise AssertionError(f"The expected BlockDevice().spinnig was False, but got True")
+
+		# if list(block_device.free_space) != [[0, 20, 20]]:
+		# 	raise AssertionError(f"The reported free space of the loopdev was not [0, 20, 20]")
+
+		# print(block_device.largest_free_space)
+		if block_device.first_free_sector != '512MB':
+			raise AssertionError(f"First free sector of BlockDevice() was not 512MB")
+
+		if block_device.first_end_sector != '20.0GB':
+			raise AssertionError(f"Last sector of BlockDevice() was not 20.0GB")
+
+		if not block_device.partprobe():
+			raise AssertionError(f"Could not partprobe BlockDevice() of loopdev")
+
+		if block_device.has_mount_point('/'):
+			raise AssertionError(f"BlockDevice() reported a mountpoint despite never being mounted")
+
+		try:
+			if block_device.get_partition('FAKE-UUID-TEST'):
+				raise AssertionError(f"BlockDevice() reported a partition despite never having any")
+		except archinstall.DiskError:
+			pass # We're supposed to not find any
+
+		# Test ended, cleanup commences
+		if not detach_loopdev(loopdev):
+			raise AssertionError(f"Could not detach {loopdev} after performing tests on {filename}.")
+	else:
+		raise AssertionError(f"Could not retrieve a loopdev for testing on {filename}")
+
+	pathlib.Path(filename).resolve().unlink()
+
+test_stat_blockdev()
\ No newline at end of file