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()