summaryrefslogtreecommitdiffstats
path: root/src/test/pybind/test_ceph_daemon.py
blob: df8d4c0b09b91548069e7f0efdfcf53d5889a4c1 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
#!/usr/bin/env python3
# -*- mode:python; tab-width:4; indent-tabs-mode:t -*-
# vim: ts=4 sw=4 smarttab expandtab
#
"""
Copyright (C) 2015 Red Hat

This is free software; you can redistribute it and/or
modify it under the terms of the GNU General Public
License version 2, as published by the Free Software
Foundation.  See file COPYING.
"""

import unittest

from ceph_daemon import DaemonWatcher

try:
    from StringIO import StringIO
except ImportError:
    from io import StringIO


class TestDaemonWatcher(unittest.TestCase):
    def test_format(self):
        dw = DaemonWatcher(None)

        self.assertEqual(dw.format_dimless(1, 4), "  1 ")
        self.assertEqual(dw.format_dimless(1000, 4), "1.0k")
        self.assertEqual(dw.format_dimless(3.14159, 4), "  3 ")
        self.assertEqual(dw.format_dimless(1400000, 4), "1.4M")

    def test_col_width(self):
        dw = DaemonWatcher(None)

        self.assertEqual(dw.col_width("foo"), 4)
        self.assertEqual(dw.col_width("foobar"), 6)

    def test_supports_color(self):
        dw = DaemonWatcher(None)
        # Can't count on having a tty available during tests, so only test the false case
        self.assertFalse(dw.supports_color(StringIO()))


if __name__ == '__main__':
    unittest.main()


# Local Variables:
# compile-command: "cd ../../..;
#  PYTHONPATH=src/pybind python3 src/test/pybind/test_ceph_daemon.py"
# End: