def test_list_delete(self): ''' test for handling_list and waiting_list issue_delete ''' asset = Asset.objects.get(id=1) Issue.objects.create(initiator=User.admin(), handler=User.admin(), assignee=User.admin(), asset=asset, type_name='MAINTAIN') category: AssetCategory = AssetCategory.root() require_issue = RequireIssue(initiator=User.admin(), handler=User.admin(), asset_category=category, reason='') require_issue.save() require_issue.asset.add(asset) response = self.client.get('/api/issue/handling').json() self.assertEqual(response['code'], 200) self.assertEqual(len(response['data']), 2) response = self.client.get('/api/issue/waiting').json() self.assertEqual(response['code'], 200) self.assertEqual(len(response['data']), 2) # delete response = self.client.post('/api/issue/delete', json.dumps({ "nid": 1, "type_name": 'RETURN' }), content_type='json') self.assertEqual(response.json()['code'], 200) self.assertFalse(Issue.objects.filter().exists()) # 删除 RequireIssue response = self.client.post('/api/issue/delete', json.dumps({ "nid": 1, "type_name": 'REQUIRE' }), content_type='json') self.assertEqual(response.json()['code'], 200) self.assertFalse(RequireIssue.objects.filter().exists())
def setUp(self) -> None: ''' 创建一个admin名下的资产 ''' init_test(self) init_category() Asset.objects.create(name='资产', value=1, category=AssetCategory.root(), status='IDLE', service_life=10, owner=User.admin()) self.category = AssetCategory.root().name
def add_old_asset(): ''' 增加一个旧资产,以展示资产折旧 ''' from asset.models import Asset, AssetCategory from user.models import User from datetime import timedelta if not Asset.objects.all().exists(): asset = Asset(name='旧资产', value=10000, category=AssetCategory.root(), status='IDLE', service_life=10, owner=User.admin()) asset.save() asset.start_time -= timedelta(366) asset.save()