def insert_client(self): if not Client.objects.filter(name='Productor').exists(): client_productor = Client() client_productor.name = 'Productor' client_productor.ruc = '12345678901' client_productor.save() client_productor.type_client.add( TypeClient.objects.get(name='Productora')) if not Client.objects.filter(name='Realizadora').exists(): client_director = Client() client_director.name = 'Realizadora' client_director.ruc = '12333678901' client_director.save() client_director.type_client.add( TypeClient.objects.get(name='Realizadora')) if not Client.objects.filter(name='Agencia').exists(): client_agency = Client() client_agency.name = 'Agencia' client_agency.ruc = '12555678901' client_agency.save() client_agency.type_client.add( TypeClient.objects.get(name='Agencia'))
def test_list_view_client(self): """ Tests data: List """ view = ClientListView.as_view() request = self.request_factory.get(reverse('client_list')) request.user = self.user response = view(request) self.assertEqual(response.status_code, 200) self.assertEqual(response.context_data['object_list'].count(), 3) client = Client() client.name = 'Client Test' client.ruc = '2345654321' client.address = 'Direccion' client.save() response = view(request) self.assertEqual(response.context_data['object_list'].count(), 4)
def test_data_list(self): """ Tests List """ self.insert_test_data() view = ClientDataListView.as_view() request = self.request_factory.get(reverse('client_data_list')) request.user = self.user response = view(request) self.assertEqual(response.status_code, 200) content = json.loads(response._container[0]) self.assertEqual(len(content.get('client')), 3) client = Client() client.name = 'Client Test' client.ruc = '2345654321' client.address = 'Direccion' client.save() response = view(request) content = json.loads(response._container[0]) self.assertEqual(len(content.get('client')), 4)