def test_update_files(self, create_app, mock_blog_model):
m = mock_blog_model(
name="hello",
my_video__file_name="my_video.mp4",
my_video__mime_type="video/mpeg",
my_video__file_type="mp4",
)
db.session.add(m)
db.session.commit()
new_file = FileStorage(
stream=open(self.my_video_update, "rb"),
filename="my_video_updated.mp4",
content_type="video/mpeg",
)
blog = m.get_blog()
file_upload.update_files(
blog,
db,
files={"my_video": new_file},
)
# Test files / dirs
assert "my_video_updated.mp4" in os.listdir("tests/test_path/blogs/1")
assert "my_video.mp4" not in os.listdir("tests/test_path/blogs/1")
def test_update_files_2(self, mock_blog_model):
db.init_app(app)
db.create_all()
file_upload = FileUpload()
file_upload.init_app(app, db)
new_file = FileStorage(
stream=open(self.my_video_update, "rb"),
filename="my_video_updated.mp4",
content_type="video/mpeg",
)
model = mock_blog_model(**self.attrs)
assert model.my_video__file_name == "my_video.mp4"
assert model.my_video__mime_type == "video/mpeg"
assert model.my_video__file_type == "mp4"
result = file_upload.update_files(
model,
files={"my_video": new_file},
)
assert result.my_video__file_name == "my_video_updated.mp4"
assert result.my_video__mime_type == "mp4"
assert result.my_video__file_type == "video/mpeg"