234 lines
6.8 KiB
Python
234 lines
6.8 KiB
Python
"""Unit tests for Barcode endpoints."""
|
|
|
|
from django.urls import reverse
|
|
|
|
from rest_framework import status
|
|
|
|
from InvenTree.unit_test import InvenTreeAPITestCase
|
|
from stock.models import StockItem
|
|
|
|
|
|
class BarcodeAPITest(InvenTreeAPITestCase):
|
|
"""Tests for barcode api."""
|
|
|
|
fixtures = [
|
|
'category',
|
|
'part',
|
|
'location',
|
|
'stock'
|
|
]
|
|
|
|
def setUp(self):
|
|
"""Setup for all tests."""
|
|
super().setUp()
|
|
|
|
self.scan_url = reverse('api-barcode-scan')
|
|
self.assign_url = reverse('api-barcode-link')
|
|
|
|
def postBarcode(self, url, barcode):
|
|
"""Post barcode and return results."""
|
|
return self.client.post(url, format='json', data={'barcode': str(barcode)})
|
|
|
|
def test_invalid(self):
|
|
"""Test that invalid requests fail."""
|
|
# test scan url
|
|
response = self.client.post(self.scan_url, format='json', data={})
|
|
self.assertEqual(response.status_code, status.HTTP_400_BAD_REQUEST)
|
|
|
|
# test wrong assign urls
|
|
response = self.client.post(self.assign_url, format='json', data={})
|
|
self.assertEqual(response.status_code, status.HTTP_400_BAD_REQUEST)
|
|
|
|
response = self.client.post(self.assign_url, format='json', data={'barcode': '123'})
|
|
self.assertEqual(response.status_code, status.HTTP_400_BAD_REQUEST)
|
|
|
|
response = self.client.post(self.assign_url, format='json', data={'barcode': '123', 'stockitem': '123'})
|
|
self.assertEqual(response.status_code, status.HTTP_400_BAD_REQUEST)
|
|
|
|
def test_empty(self):
|
|
"""Test an empty barcode scan.
|
|
|
|
Ensure that all required data is in the respomse.
|
|
"""
|
|
response = self.postBarcode(self.scan_url, '')
|
|
|
|
self.assertEqual(response.status_code, 400)
|
|
|
|
data = response.data
|
|
self.assertIn('barcode', data)
|
|
self.assertIn('Missing barcode data', str(response.data['barcode']))
|
|
|
|
def test_find_part(self):
|
|
"""Test that we can lookup a part based on ID."""
|
|
response = self.client.post(
|
|
self.scan_url,
|
|
{
|
|
'barcode': {
|
|
'part': 1,
|
|
},
|
|
},
|
|
format='json',
|
|
)
|
|
|
|
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
|
self.assertIn('part', response.data)
|
|
self.assertIn('barcode_data', response.data)
|
|
self.assertEqual(response.data['part']['pk'], 1)
|
|
|
|
def test_invalid_part(self):
|
|
"""Test response for invalid part."""
|
|
response = self.client.post(
|
|
self.scan_url,
|
|
{
|
|
'barcode': {
|
|
'part': 999999999,
|
|
}
|
|
},
|
|
format='json'
|
|
)
|
|
|
|
self.assertEqual(response.status_code, 400)
|
|
self.assertIn('error', response.data)
|
|
|
|
def test_find_stock_item(self):
|
|
"""Test that we can lookup a stock item based on ID."""
|
|
response = self.client.post(
|
|
self.scan_url,
|
|
{
|
|
'barcode': {
|
|
'stockitem': 1,
|
|
}
|
|
},
|
|
format='json',
|
|
)
|
|
|
|
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
|
self.assertIn('stockitem', response.data)
|
|
self.assertIn('barcode_data', response.data)
|
|
self.assertEqual(response.data['stockitem']['pk'], 1)
|
|
|
|
def test_invalid_item(self):
|
|
"""Test response for invalid stock item."""
|
|
response = self.client.post(
|
|
self.scan_url,
|
|
{
|
|
'barcode': {
|
|
'stockitem': 999999999,
|
|
}
|
|
},
|
|
format='json'
|
|
)
|
|
|
|
self.assertEqual(response.status_code, 400)
|
|
self.assertIn('error', response.data)
|
|
|
|
def test_find_location(self):
|
|
"""Test that we can lookup a stock location based on ID."""
|
|
response = self.client.post(
|
|
self.scan_url,
|
|
{
|
|
'barcode': {
|
|
'stocklocation': 1,
|
|
},
|
|
},
|
|
format='json'
|
|
)
|
|
|
|
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
|
self.assertIn('stocklocation', response.data)
|
|
self.assertIn('barcode_data', response.data)
|
|
self.assertEqual(response.data['stocklocation']['pk'], 1)
|
|
|
|
def test_invalid_location(self):
|
|
"""Test response for an invalid location."""
|
|
response = self.client.post(
|
|
self.scan_url,
|
|
{
|
|
'barcode': {
|
|
'stocklocation': 999999999,
|
|
}
|
|
},
|
|
format='json'
|
|
)
|
|
|
|
self.assertEqual(response.status_code, 400)
|
|
self.assertIn('error', response.data)
|
|
|
|
def test_integer_barcode(self):
|
|
"""Test scan of an integer barcode."""
|
|
response = self.postBarcode(self.scan_url, '123456789')
|
|
|
|
self.assertEqual(response.status_code, 400)
|
|
|
|
data = response.data
|
|
self.assertIn('error', data)
|
|
|
|
def test_array_barcode(self):
|
|
"""Test scan of barcode with string encoded array."""
|
|
response = self.postBarcode(self.scan_url, "['foo', 'bar']")
|
|
|
|
self.assertEqual(response.status_code, 400)
|
|
|
|
data = response.data
|
|
self.assertIn('error', data)
|
|
|
|
def test_barcode_generation(self):
|
|
"""Test that a barcode is generated with a scan."""
|
|
item = StockItem.objects.get(pk=522)
|
|
|
|
response = self.postBarcode(self.scan_url, item.format_barcode())
|
|
data = response.data
|
|
|
|
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
|
|
|
self.assertIn('stockitem', data)
|
|
|
|
pk = data['stockitem']['pk']
|
|
|
|
self.assertEqual(pk, item.pk)
|
|
|
|
def test_association(self):
|
|
"""Test that a barcode can be associated with a StockItem."""
|
|
item = StockItem.objects.get(pk=522)
|
|
|
|
self.assignRole('stock.change')
|
|
|
|
self.assertEqual(len(item.barcode_hash), 0)
|
|
|
|
barcode_data = 'A-TEST-BARCODE-STRING'
|
|
|
|
response = self.client.post(
|
|
self.assign_url, format='json',
|
|
data={
|
|
'barcode': barcode_data,
|
|
'stockitem': item.pk
|
|
}
|
|
)
|
|
|
|
data = response.data
|
|
|
|
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
|
|
|
self.assertIn('success', data)
|
|
|
|
result_hash = data['barcode_hash']
|
|
|
|
# Read the item out from the database again
|
|
item = StockItem.objects.get(pk=522)
|
|
|
|
self.assertEqual(result_hash, item.barcode_hash)
|
|
|
|
# Ensure that the same barcode hash cannot be assigned to a different stock item!
|
|
response = self.client.post(
|
|
self.assign_url, format='json',
|
|
data={
|
|
'barcode': barcode_data,
|
|
'stockitem': 521
|
|
}
|
|
)
|
|
|
|
data = response.data
|
|
|
|
self.assertIn('error', data)
|
|
self.assertNotIn('success', data)
|