Example #1
0
class PencilTest(TestCase):
    def setUp(self):
        self.pencil = Pencil()

    def test_begin(self):
        self.pencil.begin("-2hours")
        self.assertEqual(self.pencil._from, "-2hours")

    def test_y_min(self):
        self.pencil.y_min(32)
        self.assertEqual(self.pencil._yMin, 32)

    def test_y_max(self):
        self.pencil.y_max(42)
        self.assertEqual(self.pencil._yMax, 42)

    def test_line_width(self):
        self.pencil.line_width(2)
        self.assertEqual(self.pencil._lineWidth, 2)

    def test_set_title(self):
        self.pencil.set_title("Pencil")
        self.assertEqual(self.pencil._title, "Pencil")

    def test_set_vtitle(self):
        self.pencil.set_vtitle("Awesomeness")
        self.assertEqual(self.pencil._vtitle, "Awesomeness")

    def test_fgcolor(self):
        self.pencil.set_fgcolor("blue")
        self.assertEqual(self.pencil._fgcolor, "blue")

    def test_hide_legend(self):
        self.pencil.hide_legend(True)
        self.assertEqual(self.pencil._hideLegend, "true")
        self.pencil.hide_legend(False)
        self.assertEqual(self.pencil._hideLegend, "false")

    def test_hide_axes(self):
        self.pencil.hide_axes(True)
        self.assertEqual(self.pencil._hideAxes, "true")
        self.pencil.hide_axes(False)
        self.assertEqual(self.pencil._hideAxes, "false")

    def test_set_template(self):
        self.pencil.set_template("plain")
        self.assertEqual(self.pencil._template, "plain")

    def test_set_font(self):
        self.pencil.set_font("Verdana")
        self.assertEqual(self.pencil._fontName, "Verdana")

    def test_area_mode(self):
        self.pencil.area_mode("stacked")
        self.assertEqual(self.pencil._areaMode, "stacked")

    def test_line_mode(self):
        self.pencil.line_mode("staircase")
        self.assertEqual(self.pencil._lineMode, "staircase")

    def test_set_bgcolor(self):
        self.pencil.set_bgcolor("red")
        self.assertEqual(self.pencil._bgcolor, "red")

    def test_add_metric(self):
        self.pencil.add_metric("pencil.urls.count")
        self.assertEqual(self.pencil._target, ["pencil.urls.count"])

    def test_add_metric_with_color(self):
        self.pencil.add_metric("pencil.urls.count", "000000")
        self.assertEqual(self.pencil._target, ["pencil.urls.count"])
        self.assertEqual(self.pencil._colorList, "000000")

    def test_add_metric_with_alias(self):
        self.pencil.add_metric("pencil.urls.count", alias="urls")
        self.assertEqual(self.pencil._target, ['alias(pencil.urls.count, "urls")'])

    def test_graph_type(self):
        self.pencil.graph_type("pie")
        self.assertEqual(self.pencil._graphType, "pie")

    def test_add_deploy(self):
        self.pencil.add_deploy("pencil")
        self.assertEqual(self.pencil._target, ["drawAsInfinite(pencil)"])

    def test_add_deploy_with_alias(self):
        self.pencil.add_deploy("pencil", alias="deploy")
        self.assertEqual(self.pencil._target, ['alias(drawAsInfinite(pencil), "deploy")'])