2014-10-29 54 views
2

我看到这个错误,只是一个简单的集成测试。该django应用程序被称为storageadmin与称为Disk模型与自定义模型方法称为pool_name。我正在使用南和Django休息框架,并有SOUTH_TESTS_MIGRATE = Falsesettings.pyDjango fixtures和AttributeError:'QuerySet'对象没有属性<custom_model_method>

夹具是用此命令创建的。在dumpdata之前,默认数据库由应用程序填充一些模型,只是足够用于测试。

./bin/django dumpdata storageadmin oauth2_provider auth.User --indent=2 > src/rockstor /storageadmin/fixtures/storageadmin.json

错误涉及下列模型

class Disk(models.Model): 
    pool = models.ForeignKey(Pool, null=True, on_delete=models.SET_NULL) 
    name = models.CharField(max_length=10, unique=True)  

def pool_name(self, *args, **kwargs): 
    try: 
     return self.pool.name 
    except: 
     return None 

class Meta: 
    app_label = 'storageadmin' 

有其在视图用于发送响应

class DiskInfoSerializer(serializers.ModelSerializer): 
    pool_name = serializers.CharField(source='pool_name') 

    class Meta: 
     model = Disk 

观串行器返回像这样的响应

ds = DiskInfoSerializer(Disk.objects.all()) 
return Response(ds.data) 

编辑:如正确答案中所示,问题出现在上述视图中。我使用的是很旧的DRF(2.1.15),但目前正在测试版本(2.4.3)要求ds = DiskInfoSerializer(Disk.objects.all(), many=True)

简单的测试案例引发的错误是:

class DiskTests(APITestCase): 
    fixtures = ['storageadmin.json'] 
    BASE_URL = '/api/disks/scan' 

    def test_disk_scan(self): 
     self.client.login(username='admin', password='admin') 
     response = self.client.post(self.BASE_URL, data=None, format='json') 
     self.assertEqual(response.status_code, status.HTTP_200_OK) 

最后,错误是

File "/opt/rock-dep/src/rockstor/storageadmin/views/disk.py", line 88, in _scan 
    return Response(ds.data) 
File "/opt/rock-dep/eggs/djangorestframework-2.4.3-py2.7.egg/rest_framework /serializers.py", line 572, in data 
    self._data = self.to_native(obj) 
File "/opt/rock-dep/eggs/djangorestframework-2.4.3-py2.7.egg/rest_framework/serializers.py", line 351, in to_native 
    value = field.field_to_native(obj, field_name) 
File "/opt/rock-dep/eggs/djangorestframework-2.4.3-py2.7.egg/rest_framework/fields.py", line 336, in field_to_native 
    return super(WritableField, self).field_to_native(obj, field_name) 
File "/opt/rock-dep/eggs/djangorestframework-2.4.3-py2.7.egg/rest_framework/fields.py", line 207, in field_to_native 
    value = get_component(value, component) 
File "/opt/rock-dep/eggs/djangorestframework-2.4.3-py2.7.egg/rest_framework/fields.py", line 58, in get_component 
    val = getattr(obj, attr_name) 
AttributeError: 'QuerySet' object has no attribute 'pool_name' 

该代码工作正常,问题只是在测试它时。

回答

8

就像错误说的那样,你传递了一个查询集,但是你没有告诉DRF这就是你在做什么。您需要通过many=True

ds = DiskInfoSerializer(Disk.objects.all(), many=True) 
相关问题