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)
