def setUpTestData(cls): cls.factory = APIRequestFactory() cls.test_user = User.objects.create_user(username='******', password='******') cls.test_admin = User.objects.create_superuser( username='******', email='*****@*****.**', password='******') cls.report_name = "Q2 2019 Report" cls.test_supply1 = Supply(name="3D printer", state="Brand new", description="User manual is lost") cls.test_supply1.save() cls.test_supply2 = Supply(name="Screwdriver", state="Worn out", description="Handle is broken") cls.test_supply2.save() # Report has to be generated through view to make sure it's populated with supplies cls.test_report_name = 'Test report' request = cls.factory.post('/api/inventories/', {'name': cls.test_report_name}) force_authenticate(request, user=cls.test_admin) InventoryReportListCreateView.as_view()(request) cls.test_report = InventoryReport.objects.get( name=cls.test_report_name)
def setUpTestData(cls): cls.factory = APIRequestFactory() cls.test_user = User.objects.create_user(username='******', password='******') cls.test_admin = User.objects.create_superuser( username='******', email='*****@*****.**', password='******') cls.report_name = "Q2 2019 Report" cls.test_supply_name = "3D printer" cls.test_supply_state = "Brand new" cls.test_supply_description = "User manual is lost" cls.test_supply = Supply(name=cls.test_supply_name, state=cls.test_supply_state, description=cls.test_supply_description) cls.test_supply.save() cls.test_report = InventoryReport(name=cls.report_name) cls.test_report.save() cls.test_inventory_supply = InventorySupply( inventory_supply=cls.test_supply, inventory_report=cls.test_report) cls.test_inventory_supply.save()
def setUpTestData(cls): cls.factory = APIRequestFactory() cls.test_user = User.objects.create_user(username='******', password='******') cls.test_admin = User.objects.create_superuser( username='******', email='*****@*****.**', password='******') cls.report_name = "Q2 2019 Report" cls.test_supply1 = Supply(name="3D printer", state="Brand new", description="User manual is lost") cls.test_supply1.save() cls.test_supply2 = Supply(name="Screwdriver", state="Worn out", description="Handle is broken") cls.test_supply2.save()
def setUp(self): """ Sets up for tests """ super(ProjectResourceTestCase, self).setUp() self.create_user() #self.api_client.client.login(username='******', password='******') #self.customer = Customer.create(**base_customer) #self.project = Project.create(**base_project) self.project = Project(codename="Ladawan") self.project.save() self.supply = Supply(description='Grommet') self.supply.save() self.supply2 = Supply(description='Hinge') self.supply2.save() self.project_supply = ProjectSupply(supply=self.supply, project=self.project, quantity=2) self.project_supply.save()
def work(): data = [] review = [] supplier = Supplier.objects.get(name__icontains="blue international") with open("/Users/Charlie/Sites/employee/backend/blue-inter.csv", 'r') as f: reader = csv.reader(f) for row in reader: data.append({ 'reference': row[0], 'description': row[1], 'cost': row[2], 'units': row[3] }) for index, d in enumerate(data): for key in d.keys(): try: d[key] = d[key].encode('utf-8') except UnicodeDecodeError as e: pass if key == "cost": try: d[key] = Decimal(d[key].replace(',', '')) except InvalidOperation as e: review.append(data.pop(index)) if key == "units": if not d[key]: review.append(data.pop(index)) for index, d in enumerate(data): try: try: supply = Supply.objects.get(description=d['description']) except Supply.DoesNotExist: supply = Supply() supply.description = d['description'] supply.units = d['units'] supply.full_clean() supply.save() try: product = Product.objects.get(supply=supply) except Product.DoesNotExist: product = Product(supply=supply, supplier=supplier) product.supplier = supplier product.supply = supply product.reference = d['reference'] product.cost = d['cost'] product.purchasing_units = d['units'] product.full_clean() product.save() except ValidationError as e: print e review.append(data.pop(index)) assert Supply.objects.filter(description=d['description']).count() == 1 with open('blue-inter-review.csv', 'w') as f: fieldnames = ['reference', 'description', 'cost', 'units'] writer = csv.DictWriter(f) writer.write_header() for d in review: writer.writerow(d) assert supplier.supplies.all().count() == len(data)