60 lines
1.8 KiB
Python
60 lines
1.8 KiB
Python
|
# -*- coding: utf-8 -*-
|
|||
|
|
|
|||
|
|
import mock
|
|||
|
|
import unittest
|
|||
|
|
|
|||
|
|
from bumblebee.config import Config
|
|||
|
|
import bumblebee.modules.hddtemp
|
|||
|
|
|
|||
|
|
|
|||
|
|
class TestHddtempModule(unittest.TestCase):
|
|||
|
|
def setUp(self):
|
|||
|
|
config = Config()
|
|||
|
|
self.module = bumblebee.modules.hddtemp.Module(
|
|||
|
|
engine=mock.Mock(), config={"config": config})
|
|||
|
|
self.data_line = "|/dev/sda|TOSHIBA DT01ACA100 <20>|35|C||/dev/sdb|TOSHIBA DT01ACA100 <20>|37|C|"
|
|||
|
|
self.expected_parts = [
|
|||
|
|
"/dev/sda",
|
|||
|
|
"TOSHIBA DT01ACA100 <20>",
|
|||
|
|
"35",
|
|||
|
|
"C",
|
|||
|
|
"",
|
|||
|
|
"/dev/sdb",
|
|||
|
|
"TOSHIBA DT01ACA100 <20>",
|
|||
|
|
"37",
|
|||
|
|
"C",
|
|||
|
|
""]
|
|||
|
|
self.expected_per_disk = [
|
|||
|
|
["/dev/sda",
|
|||
|
|
"TOSHIBA DT01ACA100 <20>",
|
|||
|
|
"35",
|
|||
|
|
"C",
|
|||
|
|
""],
|
|||
|
|
["/dev/sdb",
|
|||
|
|
"TOSHIBA DT01ACA100 <20>",
|
|||
|
|
"37",
|
|||
|
|
"C",
|
|||
|
|
""]]
|
|||
|
|
self.device_record = self.expected_per_disk[0]
|
|||
|
|
self.expected_name_and_temp = ("sda", "35")
|
|||
|
|
self.expected_hddtemp = "sda+35°C"
|
|||
|
|
|
|||
|
|
def test_get_parts(self):
|
|||
|
|
self.assertEqual(
|
|||
|
|
self.expected_parts, self.module._get_parts(self.data_line))
|
|||
|
|
|
|||
|
|
def test_partition_parts(self):
|
|||
|
|
self.assertEqual(
|
|||
|
|
self.expected_per_disk,
|
|||
|
|
self.module._partition_parts(self.expected_parts))
|
|||
|
|
|
|||
|
|
def test_get_name_and_temp(self):
|
|||
|
|
self.assertEqual(
|
|||
|
|
self.expected_name_and_temp,
|
|||
|
|
self.module._get_name_and_temp(self.device_record))
|
|||
|
|
|
|||
|
|
def test_get_hddtemp(self):
|
|||
|
|
self.assertEqual(
|
|||
|
|
self.expected_hddtemp,
|
|||
|
|
self.module._get_hddtemp(self.expected_name_and_temp))
|