view test/test_status.py @ 57:d0d08c7e7d37

allow optionally set the file permissions in writefile
author Bernhard Herzog <bh@intevation.de>
date Tue, 10 Apr 2007 17:42:07 +0200
parents 3c5ab7a65384
children 39b2deea8481
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.error()

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

    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.name, "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)