def setUp(self): self.user = get_user_model().objects.create(username='******', is_superuser=True) self.sample = Sample.objects.create(contributor=self.user, name="Test sample") sample_viewset = SampleViewSet() self.sample_queryset = sample_viewset.get_queryset() presample_viewset = PresampleViewSet() self.presample_queryset = presample_viewset.get_queryset() detail_url_mapping = { 'get': 'retrieve', 'put': 'update', 'patch': 'partial_update', 'delete': 'destroy', } list_url_mapping = { 'get': 'list', 'post': 'create', } self.detail_sample_view = SampleViewSet.as_view(detail_url_mapping) self.detail_presample_view = PresampleViewSet.as_view( detail_url_mapping) self.list_sample_view = SampleViewSet.as_view(list_url_mapping) self.list_presample_view = PresampleViewSet.as_view(list_url_mapping) self.factory = APIRequestFactory()
def setUp(self): user_model = get_user_model() self.user = user_model.objects.create_user('test_user') self.collection = Collection.objects.create(name="Test Collection", contributor=self.user) self.sample = Sample.objects.create(name="Test sample", contributor=self.user) process = Process.objects.create(name="Test process", contributor=self.user) self.data = Data.objects.create(name="Test data", contributor=self.user, process=process) self.data_2 = Data.objects.create(name="Test data 2", contributor=self.user, process=process) # another Data object to make sure that other objects are not processed Data.objects.create(name="Dummy data", contributor=self.user, process=process) self.sample.data.add(self.data) assign_perm('add_collection', self.user, self.collection) assign_perm('add_sample', self.user, self.sample) self.sampleviewset = SampleViewSet()