def post_to_add_batch(ref, sku, qty, eta):
url = config.get_api_url()
r = requests.post(
f'{url}/add_batch',
json={'ref': ref, 'sku': sku, 'qty': qty, 'eta': eta}
)
assert r.status_code == 201
def test_400_message_for_invalid_sku():
unknown_sku, orderid = random_sku(), random_orderid()
data = {'orderid': orderid, 'sku': unknown_sku, 'qty': 20}
url = config.get_api_url()
r = requests.post(f'{url}/allocate', json=data)
assert r.status_code == 400
assert r.json()['message'] == f'Invalid sku {unknown_sku}'
def test_deallocate(add_stock):
sku, order1, order2 = random_sku(), random_orderid(), random_orderid()
batch = random_batchref()
add_stock([
(batch, sku, 100, '2011-01-02')
])
data = {'orderid': order1, 'sku': sku, 'qty': 100}
url = config.get_api_url()
# fully allocate
r = requests.post(f'{url}/allocate', json={
'orderid': order1, 'sku': sku, 'qty': 100
})
assert r.json()['batchid'] == batch
# cannot allocate second order
r = requests.post(f'{url}/allocate', json={
'orderid': order2, 'sku': sku, 'qty': 100
})
assert r.status_code == 400
# deallocate
r = requests.post(f'{url}/deallocate', json={
'orderid': order1, 'sku': sku
})
assert r.status_code == 201
# allocate second order
r = requests.post(f'{url}/allocate', json={
'orderid': order2, 'sku': sku, 'qty': 100
})
assert r.status_code == 201
assert r.json()['batchid'] == batch
def wait_for_webapp_to_come_up():
deadline = time.time() + 10
url = config.get_api_url()
while time.time() < deadline:
try:
return requests.get(url)
except ConnectionError:
time.sleep(0.5)
pytest.fail('API never came up')
def test_400_message_for_out_of_stock(add_stock):
sku, small_batch, large_order = random_sku(), random_batchref(), random_orderid()
add_stock([
(small_batch, sku, 10, '2011-01-01')
])
data = {'orderid': large_order, 'sku': sku, 'qty': 20}
url = config.get_api_url()
r = requests.post(f'{url}/allocate', json=data)
assert r.status_code == 400
assert r.json()['message'] == f'Out of stock for sku {sku}'
def post_to_allocate(orderid, sku, qty, expect_success=True):
url = config.get_api_url()
r = requests.post(f'{url}/allocate',
json={
'orderid': orderid,
'sku': sku,
'qty': qty,
})
if expect_success:
assert r.status_code == 202
return r
def test_happy_path_returns_201_and_allocated_batch():
sku, othersku = random_sku("test"), random_sku('other')
earlybatch = random_batchref("1")
laterbatch = random_batchref("2")
otherbatch = random_batchref("3")
post_to_add_batch(laterbatch, sku, 100, '2011-01-02')
post_to_add_batch(earlybatch, sku, 100, '2011-01-01')
post_to_add_batch(otherbatch, othersku, 100, None)
data = {'orderid': random_orderid('lala'), 'sku': sku, 'qty': 3}
url = config.get_api_url()
r = requests.post(f'{url}/allocate', json=data)
assert r.status_code == 201
assert r.json()['batchref'] == earlybatch
def test_api_returns_201_andgithu_allocation(add_stock):
sku, othersku = random_sku(), random_sku('other')
earlybatch = random_batchref(1)
laterbatch = random_batchref(2)
otherbatch = random_batchref(3)
post_to_add_batch(laterbatch, sku, 100, '2011-01-02')
post_to_add_batch(earlybatch, sku, 100, '2011-01-01')
post_to_add_batch(otherbatch, othersku, 100, None)
data = {'orderid': random_orderid(), 'sku': sku, 'qty': 3}
url = config.get_api_url()
r = requests.post(f'{url}/allocate', json=data)
assert r.status_code == 201
assert r.json()['batchref'] == earlybatch
def get_allocation(orderid):
url = config.get_api_url()
return requests.get(f'{url}/allocations/{orderid}')
def wait_for_webapp_to_come_up():
return requests.get(config.get_api_url())