PBI-4: Menambahkan kolom satuan pada model Produk

api/tests.py

@@ -1181,6 +1181,7 @@ class ProductTest(rest_framework_test.APITestCase):
         self.assertEqual(response.status_code, status.HTTP_201_CREATED)
         self.assertEqual(models.Product.objects.count(), 1)
         self.assertEqual(models.Product.objects.get(id=response.data['id']).name, data['name'])
+        self.assertEqual(models.Product.objects.get(id=response.data['id']).unit, data['unit'])
 
     def test_create_product_fail(self):
         data = dict(seeds.PRODUCT_DATA, subcategory=self.subcategory.id)
@@ -1215,6 +1216,7 @@ class ProductTest(rest_framework_test.APITestCase):
         self.assertEqual(response.status_code, status.HTTP_200_OK)
         self.assertEqual(models.Product.objects.get(id=product.id).profit,2000)
         self.assertEqual(models.Product.objects.get(id=product.id).name, data['name'])
+        self.assertEqual(models.Product.objects.get(id=product.id).unit, data['unit'])
         data = dict(seeds.PRODUCT_DATA, subcategory=self.subcategory.id)
         response = request(
             'PUT',

api/views.py

@@ -639,6 +639,7 @@ class ProductList(generics.ListCreateAPIView):
             modal=serializer.validated_data['modal'],
             subcategory=models.Subcategory.objects.get(name=serializer.validated_data['subcategory']),
             total_profit=0,
+            unit=serializer.validated_data['unit'],
             image=validated_image
         )
         product.profit= (product.price - product.modal)