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"