"""
Unit test for localdisk_service - Treadmill local disk service
"""

import collections
import os
import shutil
import subprocess
import tempfile
import unittest

import mock

import treadmill
from treadmill.services import localdisk_service


class LocalDiskServiceTest(unittest.TestCase):
    """Unit tests for the local disk service implementation.
    """
    def setUp(self):
        self.root = tempfile.mkdtemp()

    def tearDown(self):
        if self.root and os.path.isdir(self.root):
            shutil.rmtree(self.root)