class ModelDALTestCase(TestCase): def setUp(self): super(ModelDALTestCase, self).setUp() self.useFixture(DBConnectionFixture()) self.dbfixture = self.useFixture(DBFixture()) self.session = self.dbfixture.session self.dal = ModelDal() def tearDown(self): super(ModelDALTestCase, self).tearDown() self.session.execute('delete from mp_model') def test_get_model_all(self): ret = self.dal.get_model_all() self.assertEqual(len(ret), 3) def test_get_model_by_prov_id(self): # 通过id返回的还是一个集合,会有1对多吗 model = self.dal.get_model_by_prov_id(11) self.assertEqual(model[0].name, 'model1') def test_add_model(self): self.dal.add_model(3, 'model3') ret = self.dal.get_model_all() self.assertEqual(len(ret), 4) def test_del_model_by_id(self): # 这个测试跑不过去 self.dal.del_model_by_id(1) ret = self.dal.get_model_all() self.assertEqual(2, len(ret)) def test_get_model_by_pd_ch(self): model = self.dal.get_model_by_pd_ch(11, 'model1') self.assertIsNotNone(model) self.assertEqual(model.name, 'model1') self.assertEqual(model.provider_id, 11) model = self.dal.get_model_by_pd_ch(1, 'model2') self.assertIsNone(model) def test_get_model_name_by_id(self): model = self.dal.get_model_name_by_id(1) self.assertIsNotNone(model) self.assertEqual(model.name, 'model1') def test_get_model_list_by_prov_id(self): models = self.dal.get_model_list_by_prov_id(22) self.assertIsNotNone(models) self.assertEqual(len(models), 2)
def get_model_name_by_num(self, model_id): md = ModelDal() model = md.get_model_name_by_id(model_id) return self.actual_return_name_val(model)
def get_model_by_num(self, model_id): md = ModelDal() model = md.get_model_name_by_id(model_id) return model