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()