def get_form_brand_dtos():
brand_dtos = [
GetFormBrandDtoWithItemId(
brand_id=1,
name="HoneyBee",
min_quantity=2,
max_quantity=10,
price_per_item=100,
item_id=1),
GetFormBrandDtoWithItemId(
brand_id=2,
name="KnockOut",
min_quantity=2,
max_quantity=11,
price_per_item=200,
item_id=1),
GetFormBrandDtoWithItemId(
brand_id=3,
name="KingFisher",
min_quantity=2,
max_quantity=11,
price_per_item=200,
item_id=2)
]
return brand_dtos
def get_form_expected_form_details_dto():
brand1= GetFormBrandDtoWithItemId(
brand_id=1,
name="HoneyBee",
min_quantity=2,
max_quantity=10,
price_per_item=100,
item_id=1)
brand2 = GetFormBrandDtoWithItemId(
brand_id=2,
name="KnockOut",
min_quantity=2,
max_quantity=11,
price_per_item=200,
item_id=1)
brand3 = GetFormBrandDtoWithItemId(
brand_id=3,
name="KingFisher",
min_quantity=2,
max_quantity=11,
price_per_item=200,
item_id=2)
order1 = GetFormItemOrderDto(order_id=1,
brand_id=1,
ordered_count=10,
out_of_stock=0)
order2 = GetFormItemOrderDto(order_id=2,
brand_id=3,
ordered_count=10,
out_of_stock=0)
item1 = GetFormItemDto(item_id=1,
name="choco",
description="choco",
brands=[brand1, brand2],
order=order1)
item2 = GetFormItemDto(item_id=2,
name="lace",
description="lace",
brands=[brand3],
order=order2)
section1 = GetFormSectionDto(section_id=1,
name="SECTION1",
description="DESCRIP SECTION1",
items=[item1, item2])
section2 = GetFormSectionDto(section_id=2,
name="SECTION2",
description="DESCRIP SECTION2",
items=[])
get_form_dto = GetFormDto(form_id=1,
name="FORM1",
close_date=datetime.datetime(2020, 10, 10, 0, 0),
sections=[section1, section2])
return get_form_dto
def test_get_brand_dtos_with_valid_details_return_dtos(populate_data):
# Arrange
item_dtos = [
GetFormItemDtoWithSectionId(
item_id=1,
name="item1",
description="item1",
section_id=1),
GetFormItemDtoWithSectionId(
item_id=2,
name="item2",
description="item2",
section_id=1)]
expected_brand_dtos = [
GetFormBrandDtoWithItemId(
brand_id=1,
name="brand1",
min_quantity=1,
max_quantity=5,
price_per_item=100,
item_id=1),
GetFormBrandDtoWithItemId(
brand_id=2,
name="brand2",
min_quantity=2,
max_quantity=8,
price_per_item=200,
item_id=1),
GetFormBrandDtoWithItemId(
brand_id=3,
name="brand3",
min_quantity=3,
max_quantity=9,
price_per_item=300,
item_id=2)
]
storage = FormStorageImplementation()
# Act
actual_brand_dtos = storage.get_brand_dtos(item_dtos=item_dtos)
# Assert
assert actual_brand_dtos == expected_brand_dtos
def _convert_brands_to_dtos(self, item_id, brand_objs):
brand_dtos = []
for brand in brand_objs:
dto = GetFormBrandDtoWithItemId(
brand_id=brand.id,
name=brand.name,
min_quantity=brand.min_quantity,
max_quantity=brand.max_quantity,
price_per_item=brand.price_per_item,
item_id=item_id)
brand_dtos.append(dto)
return brand_dtos