def test_upload_holes(app_and_db, add_investment):
investment = Investment.query.first()
Wall.upload_walls(investment.id, "test/walls.csv")
Wall.upload_holes(investment.id, "test/holes.csv")
messages = Wall.upload_holes(investment.id, "test/holes.csv")
assert len(Hole.query.all()) == 5
assert len(messages) == 3
assert messages[0] == "Uploaded 5 items."
assert (
messages[1] ==
"Items: [5, 1, 1, 1] not added because they has the wrong format.")
assert (
messages[2] ==
"Items: [6, 7, 12, 13, 14, 15, 16, 17] not added because wall with specified id does not exist. Add wall first."
)
def test_upload_processing(app_and_db, add_investment):
investment = Investment.query.first()
Wall.upload_walls(investment.id, "test/walls.csv")
Wall.upload_processing(investment.id, "test/processing.csv")
messages = Wall.upload_processing(investment.id, "test/processing.csv")
assert len(Processing.query.all()) == 6
assert len(messages) == 4
assert messages[0] == "Uploaded 6 items."
assert (
messages[1] ==
"Items: [1, 1, 1, 2, 2] not added because they has the wrong format."
)
assert (
messages[2] ==
"Items: [7, 8, 9] not added because wall with specified id does not exist. Add wall first."
)
assert (messages[3] ==
"Items: [1, 3] not added because value of left_to_sale is 0.")
def test_upload_walls(app_and_db, add_investment):
investment = Investment.query.first()
messages = Wall.upload_walls(investment.id, "test/walls.csv")
assert len(Wall.query.all()) == 5
assert len(messages) == 2
assert messages[0] == "Uploaded 6 items."
assert (
messages[1] ==
"Items: [8, 9, 10, 11, 12, 13, 14, 15] not added because they has the wrong format."
)
assert Wall.query.filter_by(id=1).first().wall_width == 18
def test_upload_walls_when_wrong_file(app_and_db, add_investment):
investment = Investment.query.first()
messages = Wall.upload_walls(investment.id, "test/holes.csv")
assert len(Wall.query.all()) == 0
assert len(messages) == 1
assert messages[0] == "Uploaded 0 items."
