class TestCaseAsignareService(unittest.TestCase):
    def setUp(self):
        self.asign=AsignService(StudentRepository,ProbRepository,AsignRepository(),AsignValidator())
        self.asign.create_asign("1", "2", "3")
    def test_create(self):
        self.assertTrue(len(self.asign.get_tot())==1)
        self.assertRaises(AsignValidatorException,self.asign.create_asign, "", "", "3")
        self.assertRaises(AsignRepositoryException,self.asign.create_asign,"1","2","3")
    def test_sterge(self):
        self.assertRaises(AsignRepositoryException,self.asign.sterge_prob,"2")
        self.assertTrue(self.asign.get_asig_nr()==1)
        asg=self.asign.get("1")
        self.assertTrue(asg.getStatus()=="asignare")
        self.asign.return_prob("1")
        self.assertTrue(asg.getStatus()=="returnat")
        self.asign.sterge_prob("1")
        self.assertTrue(self.asign.get_asig_nr()==0)
        self.assertEquals(asg.getAsignID(),"1")
        self.assertTrue(asg.getStudentID()=="2")
        self.assertTrue(asg.getNrLab()=="3")
def suite6():
    s=unittest.TestSuite()
    t=unittest.TestLoader().loadTestsFromTestCase(TestCaseAsignareService)
    s.addTests(t)
    return s

class TestCaseAsignareRepository(unittest.TestCase):
    def setUp(self):
        self.rep=AsignRepository()
        self.rep.store(Asignare("1","2","3"))

    def test_create(self):
        self.assertTrue(self.rep.size()==1)
        self.assertRaises(AsignRepositoryException,self.rep.store,"2")
        self.assertTrue(self.rep.size()==1)
        self.rep.sterge("1")
        self.assertTrue(self.rep.size()==0)
def suite7():
    s=unittest.TestSuite()
    t=unittest.TestLoader().loadTestsFromTestCase(TestCaseAsignareRepository)
    s.addTests(t)
    return s

class TestCaseAsignareValidator(unittest.TestCase):
    def setUp(self):
        self.val=AsignValidator()
    def test_validate(self):
        self.assertRaises(AsignValidatorException, self.val.validate,Asignare("","2","3"))
        self.assertRaises(AsignValidatorException, self.val.validate,Asignare("1","","3"))
        self.assertRaises(AsignValidatorException, self.val.validate,Asignare("1","2",""))
        self.val.validate(Asignare("1","2","3"))

def suite8():
    s=unittest.TestSuite()
    t=unittest.TestLoader().loadTestsFromTestCase(TestCaseAsignareValidator)
    s.addTests(t)
    return s