diff --git a/spec/controllers/api/v0/shipments_controller_spec.rb b/spec/controllers/api/v0/shipments_controller_spec.rb index aff739a8d2..cf04568a78 100644 --- a/spec/controllers/api/v0/shipments_controller_spec.rb +++ b/spec/controllers/api/v0/shipments_controller_spec.rb @@ -135,12 +135,24 @@ describe Api::V0::ShipmentsController, type: :controller do }.to change { inventory_units_for(new_variant).size }.by(2) end + it 'adjusts stock when adding a variant' do + expect { + api_put :add, params.merge(variant_id: new_variant.to_param) + }.to change { new_variant.reload.on_hand }.by(-2) + end + it 'removes a variant from a shipment' do expect { api_put :remove, params.merge(variant_id: existing_variant.to_param) expect(response.status).to eq(200) }.to change { inventory_units_for(existing_variant).size }.by(-2) end + + it 'adjusts stock when removing a variant' do + expect { + api_put :remove, params.merge(variant_id: existing_variant.to_param) + }.to change { existing_variant.reload.on_hand }.by(2) + end end end