def test_post_exists(self): Department.objects.all().delete() departmentName = "CSE" department = Department(departmentName=departmentName) department.save() data = {"departmentName": "CSE"} response = self.client.post('/department/', data) self.assertEqual( json.loads(response.content)['Status'], status.HTTP_409_CONFLICT)
def test_delete(self): Department.objects.all().delete() department = Department(departmentName='CSE') department.save() saved_data = Department.objects.get(departmentName='CSE') response = self.client.delete('/department/%s/' % saved_data.departmentId) self.assertEqual( json.loads(response.content)['Status'], status.HTTP_202_ACCEPTED)
def test_delete_not_found(self): Department.objects.all().delete() department = Department(departmentName='CSE') department.save() saved_data = Department.objects.get(departmentName='CSE') id = int(saved_data.departmentId) + 1 response = self.client.delete('/department/%s/' % id) self.assertEqual( json.loads(response.content)['Status'], status.HTTP_404_NOT_FOUND)
def test_put_not_found(self): Professor.objects.all().delete() self.setup() id=int(self.professorId)+1 departmentName="EXTC" department=Department(departmentName=departmentName) department.save() departmentId=Department.objects.get(departmentName='EXTC').departmentId data={"fname":"rkc","lname":"rkc","email":"*****@*****.**","mobile":"123456789","city":"nanded","state":"maha","department":departmentId} response=self.client.put('/professor/%s/' % id,data) self.assertEqual(json.loads(response.content)['Status'],status.HTTP_404_NOT_FOUND)
def test_put(self): Department.objects.all().delete() departmentName = "CSE" department = Department(departmentName=departmentName) department.save() saved_data = Department.objects.get(departmentName='CSE') data = {"departmentName": "EXTC"} response = self.client.put('/department/%s/' % saved_data.departmentId, data) self.assertEqual( json.loads(response.content)['Status'], status.HTTP_201_CREATED)
def test_put(self): HOD.objects.all().delete() self.setup() department = Department(departmentName='EXTC') department.save() departmentId = Department.objects.get( departmentName='EXTC').departmentId data = {"departmentId": departmentId} response = self.client.put('/hod/%s/' % self.hodId, data) self.assertEqual( json.loads(response.content)['Status'], status.HTTP_201_CREATED)
def test_put_not_found(self): Department.objects.all().delete() departmentName = 'CSE' department = Department(departmentName=departmentName) department.save() data = Department.objects.get(departmentName='CSE') id = int(data.departmentId) + 1 data = {"departmentName": "EXTC"} response = self.client.put('/department/%s/' % id) self.assertEqual( json.loads(response.content)['Status'], status.HTTP_404_NOT_FOUND)
def setup(self): departmentName='CSE' professerName='RKC' department=Department(departmentName=departmentName) department.save() department=Department.objects.get(departmentName=departmentName) self.department=department self.departmentId=department.departmentId professor=Professor(fname=professerName,lname=professerName,email='*****@*****.**',mobile='123456789',city='nanded',state='maha',department=self.department) professor.save() professor=Professor.objects.get(fname=professerName) self.professor=professor self.professorId=professor.professorId
def test_put_not_found(self): HOD.objects.all().delete() self.setup() id = int(self.hodId) + 1 departmentName = "EXTC" department = Department(departmentName=departmentName) department.save() departmentId = Department.objects.get( departmentName='EXTC').departmentId data = {"departmentId": departmentId} response = self.client.put('/hod/%s/' % id, data) self.assertEqual( json.loads(response.content)['Status'], status.HTTP_404_NOT_FOUND)
class TestScrapyPlan(APITestCase): """ 测试爬取天窗修计划的情形 """ def setUp(self): self.test_department = Department(name='测试部门', username='******', password='******') self.test_department.save() user = User.objects.create_user( username='******', password='******', email='*****@*****.**' ) self.test_user = UserDetailInfo(username='******', department=self.test_department, user=user) def test_scrapy_is_ok(self): """ 测试爬取是否正常,仅测试在内网的情况 :return: """ if is_in_rail_net: start_date = datetime.date.today() - datetime.timedelta(days=8) end_date = datetime.date.today() - datetime.timedelta(days=1) url = reverse('查询天窗修计划历史') url += '?' url += '&'.join([ 'start_date=' + start_date.strftime('%Y-%m-%d'), 'end_date=' + end_date.strftime('%Y-%m-%d'), 'force_update=' + 'true' ]) self.client.force_login(self.test_user.user) response = self.client.get(url) self.assertEqual(response.status_code, 200) self.assertEqual(response.data['length'], len(response.data['data'])) for i in response.data['data']: self.assertGreaterEqual(len(i['content']), 1) else: return def test_get_data_from_cache(self): """从缓存获取计划""" pass def test_get_date_by_generator(self): """ 测试生成函数是否工作 :return: """ pass
def test_put_coflict(self): HOD.objects.all().delete() self.setup() departmentName = "EXTC" department = Department(departmentName=departmentName) department.save() hod = HOD(departmentId=department, hod=self.professor) hod.save() departmentId = Department.objects.get( departmentName='EXTC').departmentId data = {"departmentId": departmentId} response = self.client.put('/hod/%s/' % self.hodId, data, fomat='json') self.assertEqual( json.loads(response.content)['Status'], status.HTTP_409_CONFLICT)
def test_put_coflict(self): Department.objects.all().delete() departmentName = "CSE" department = Department(departmentName=departmentName) department.save() departmentName = "EXTC" department = Department(departmentName=departmentName) department.save() saved_data = Department.objects.get(departmentName='EXTC') data = {"departmentName": "CSE"} response = self.client.put('/department/%s/' % saved_data.departmentId, data, fomat='json') self.assertEqual( json.loads(response.content)['Status'], status.HTTP_409_CONFLICT)
def add_department(title): dept = Department() dept.title = title deptUser = User() deptUser.username = title deptUser.set_password(title) deptUser.is_staff=True deptUser.is_active=True dept_group = Group.objects.get(name='Department') deptUser.save() dept.user = deptUser dept.save() dept_group.user_set.add(deptUser)
def init_department(): ''' 增加顶层部门 ''' from department.models import Department if not Department.objects.all().exists(): top = Department(name='总公司', parent=None) top.save() d_1 = Department(name='开发部', parent=top) d_11 = Department(name='前端组', parent=d_1) d_12 = Department(name='后端组', parent=d_1) d_2 = Department(name='市场部', parent=top) d_3 = Department(name='运维部', parent=top) d_4 = Department(name='人力资源部', parent=top) d_5 = Department(name='财务部', parent=top) for department in [d_1, d_11, d_12, d_2, d_3, d_4, d_5]: department.save()
def departmentFactory(): deps = [] new_department = Department() new_department.name = dep_levels.pop(0) new_department.parent = None new_department.save() deps.append(new_department) try: for parent in deps: new_department = Department() if len(dep_levels) > 0: new_department.name = dep_levels.pop(0) new_department.parent = parent new_department.save() deps.append(new_department) else: print('List is empty') except: print('finish')
def seed_department_table(): dept = Department(name="Testing Department", code="TEST") dept.save() dept = Department(name="Computer Science/Engineering", code="CSCE") dept.save()