test.py

import datetime
import os
import subprocess
import time
import unittest

class TestCase(unittest.TestCase):

    @classmethod
    def setUpClass(cls):
        cls.year = 2021
        subprocess.run(['make'], capture_output=True)
        subprocess.run(['./ext2-create'], capture_output=True)
        p = subprocess.run(['dumpe2fs', 'base.img'], capture_output=True, text=True)
        cls.dump_lines = p.stdout.splitlines()
        p = subprocess.run(['fsck.ext2', '-f', '-n', 'base.img'], capture_output=True, text=True)
        cls.fsck_lines = p.stdout.splitlines()
        p = subprocess.run(['ls', '-f', '-n', 'base.img'], capture_output=True, text=True)
        cls.ls_lines = p.stdout.splitlines()
        subprocess.run(['mkdir', 'mnt'], capture_output=True)
        subprocess.run(['sudo', 'mount', '-o', 'loop', 'base.img', 'mnt'])
        cls.root_stat = os.lstat('mnt') if os.path.exists('mnt') else None
        cls.hello_world_stat = os.lstat('mnt/hello-world') if os.path.exists('mnt/hello-world') else None
        cls.hello_stat = os.lstat('mnt/hello') if os.path.exists('mnt/hello') else None

    @classmethod
    def tearDownClass(cls):
        subprocess.run(['sudo', 'umount', 'mnt'])
        subprocess.run(['rmdir', 'mnt'], capture_output=True)
        subprocess.run(['make', 'clean'], capture_output=True)

    def test_hello(self):
        self.assertEqual(os.readlink('mnt/hello'), 'hello-world')

    def test_hello_world(self):
        with open('mnt/hello-world') as f:
            self.assertEqual(f.read(), "Hello world\n")