view test/test_status.py @ 40:c544903eeced

Add real enums for status information
author Bernhard Herzog <bh@intevation.de>
date Thu, 15 Mar 2007 21:03:44 +0100
parents 086c68ca51d2
children 3c5ab7a65384
line wrap: on
line source
# Copyright (C) 2007 by Intevation GmbH
# Authors:
# Bernhard Herzog <bh@intevation.de>
#
# This program is free software under the GPL (>=v2)
# Read the file COPYING coming with the software for details.

"""Tests for the Status classes"""

import os
import unittest
from datetime import datetime

from treepkg.status import RevisionStatus, Status, EnumFieldDesc
from treepkg.util import ensure_directory, writefile



class TestStatus(unittest.TestCase):

    def tempfilename(self):
        tempdir = os.path.join(os.path.dirname(__file__), "temp")
        ensure_directory(tempdir)
        return os.path.join(tempdir, self.id())

    def setUp(self):
        self.filename = self.tempfilename()
        if os.path.exists(self.filename):
            os.remove(self.filename)

    def test_status(self):
        status = RevisionStatus(self.filename)
        status.status = "testing"

        otherstatus = RevisionStatus(self.filename)
        self.assertEquals(otherstatus.status, "testing")

    def test_getting_unknown_fields(self):
        status = RevisionStatus(self.filename)
        self.assertRaises(AttributeError, getattr, status, "unknown_field")

    def test_setting_unknown_fields(self):
        status = RevisionStatus(self.filename)
        self.assertRaises(AttributeError,
                          setattr, status, "unknown_field", "some value")

    def test_default_values(self):
        status = RevisionStatus(self.filename)
        self.assertEquals(status.status, "unknown")
        self.assertEquals(status.start, None)
        self.assertEquals(status.stop, None)

    def test_date(self):
        timestamp = datetime(2007, 3, 9, 17, 32, 55)
        status = RevisionStatus(self.filename)
        status.start = timestamp

        otherstatus = RevisionStatus(self.filename)
        self.assertEquals(otherstatus.start, timestamp)

    def test_magic(self):
        writefile(self.filename,
                  "Some other magic\nstart: 2007-03-09 17:32:55\n")
        self.assertRaises(ValueError, RevisionStatus, self.filename)

    def test_status_enum(self):

        class TestStatus(Status):
            status = EnumFieldDesc()
            status.add("not_running", "Process is not running",
                       default=True)
            status.add("running", "Process is running")
            status.add("error", "An error occurred")

        status = TestStatus(self.filename)
        status.running()

        otherstatus = TestStatus(self.filename)
        self.assertEquals(otherstatus.status.name, "running")
This site is hosted by Intevation GmbH (Datenschutzerklärung und Impressum | Privacy Policy and Imprint)