#!/usr/bin/env python
# ClusterShell (local) test suite
# Written by S. Thiell 2009-02-09


"""Unit test for ClusterShell Task/Worker timeout support"""

import copy
import sys
import unittest

sys.path.insert(0, '../lib')

import ClusterShell

from ClusterShell.NodeSet import NodeSet
from ClusterShell.Task import *

import socket

import thread


class TaskTimeoutTest(unittest.TestCase):
            
    def testWorkersTimeoutBuffers(self):
        """test worker buffers with timeout"""
        task = task_self()
        self.assert_(task != None)

        worker = task.shell("python test_command.py --timeout=10", timeout=4)
        self.assert_(worker != None)

        task.resume()
        self.assertEqual(worker.read(), """some buffer
here...""")
        test = 1
        for buf, keys in task.iter_buffers():
            test = 0
            self.assertEqual(buf, """some buffer
here...""")
        self.assertEqual(test, 0, "task.iter_buffers() did not work")

    
    

if __name__ == '__main__':
    suite = unittest.TestLoader().loadTestsFromTestCase(TaskTimeoutTest)
    unittest.TextTestRunner(verbosity=2).run(suite)