mirror of
https://github.com/b-ryan/powerline-shell.git
synced 2026-09-10 07:26:28 -04:00
22 lines
725 B
Python
22 lines
725 B
Python
import unittest
|
|
import mock
|
|
import powerline_shell.segments.hostname as hostname
|
|
from powerline_shell.themes.default import Color
|
|
from argparse import Namespace
|
|
|
|
|
|
class HostnameTest(unittest.TestCase):
|
|
def setUp(self):
|
|
self.powerline = mock.MagicMock()
|
|
self.powerline.theme = Color
|
|
self.segment = hostname.Segment(self.powerline, {})
|
|
|
|
def test_colorize(self):
|
|
self.powerline.segment_conf.return_value = True
|
|
self.segment.start()
|
|
self.segment.add_to_powerline()
|
|
args = self.powerline.append.call_args[0]
|
|
self.assertNotEqual(args[0], r" \h ")
|
|
self.assertNotEqual(args[1], Color.HOSTNAME_FG)
|
|
self.assertNotEqual(args[2], Color.HOSTNAME_BG)
|