class DiscTest(unittest.TestCase):
def setUp(self):
self.disc = Disc()
program1 = Program('program1')
self.disc.programs.update({'program1':program1})
def testGetProgram(self):
program = self.disc.getProgram('program1')
self.assertIsNotNone(program)
def testGetProgramThatNotExists(self):
with self.assertRaises(ProgramNotInDiscException):
self.disc.getProgram('program3')
def setUp(self):
self.disc = Disc()
program1 = Program('program1')
self.disc.programs.update({'program1':program1})
