51 lines
998 B
Python
Executable File
51 lines
998 B
Python
Executable File
import unittest
|
|
|
|
import job_list
|
|
|
|
class Mock_Job() :
|
|
def __init__(self, id, p) :
|
|
self.id = id
|
|
|
|
def kill(self) :
|
|
return
|
|
|
|
class Test_Job_List(unittest.TestCase):
|
|
def setUp(self) :
|
|
self.was = job_list.Job
|
|
job_list.Job = Mock_Job
|
|
|
|
def tearDown(self) :
|
|
job_list.Job = self.was
|
|
|
|
def test_add(self) :
|
|
jl = job_list.Job_List()
|
|
self.assertEqual(len(jl), 0)
|
|
|
|
try :
|
|
jl.add(None)
|
|
self.fail("can add nil to job_list")
|
|
except Exception :
|
|
pass
|
|
|
|
jl.add(Mock_Job("a", "a"))
|
|
self.assertEqual(len(jl), 1)
|
|
|
|
jl.add(Mock_Job("a", "a"))
|
|
self.assertEqual(len(jl), 1)
|
|
|
|
jl.add(Mock_Job("b", "b"))
|
|
self.assertEqual(len(jl), 2)
|
|
|
|
other = job_list.Job_List()
|
|
other.add(Mock_Job("b", "b"))
|
|
other.add(Mock_Job("c", "c"))
|
|
|
|
jl.add(other)
|
|
self.assertEqual(len(jl), 3)
|
|
|
|
jl.kill("a")
|
|
self.assertEqual(len(jl), 2)
|
|
|
|
if __name__ == "__main__" :
|
|
unittest.main()
|