Serviceクラスのテスト

Question

Spring boot のserviceクラスのテストで、メソッド内でインスタンス化してRepositoryにsaveした時の
verifyの仕方がわからないです。

@Entity
@Data
public class FileEntity {
	@Id
	@GeneratedValue(strategy=GenerationType.AUTO)
	private Integer id;
	
	@Transient
	private MultipartFile file;
	
	private String base64_str;
	
	@ManyToOne(cascade=CascadeType.ALL)
	private Blog blog;
}
@Service
public class UploadFileService {
	@Autowired
	private FileRepository fileRepository;
	
	public void saveFile(List<MultipartFile> mfiles,Blog blog) {
		for(MultipartFile file:mfiles) {
			if(!file.isEmpty()) {
				FileEntity fileE=new FileEntity();
				fileE.setBlog(blog);
				try {
					StringBuffer data = new StringBuffer();
					String base64 = new String(Base64.encodeBase64(file.getBytes()),"ASCII");
					data.append("data:image/jpeg;base64,");
					data.append(base64);
					fileE.setBase64_str(data.toString());
					fileRepository.save(fileE);
				} catch (IOException e) {
					e.printStackTrace();
				}
			}else {
				FileEntity fileE=new FileEntity();
				fileE.setBlog(blog);
				fileRepository.save(fileE);
			}
		}
	}

public interface FileRepository extends JpaRepository<FileEntity, Integer>{}

UploadFileServiceクラスsaveFile()メソッド内でfileEをインスタンス化してリポジトリに保存しているのですが
テストの時にどう書いたらいいのかわからないです。
試しに書いたテストコードが下記になります。

@SpringJUnitConfig
public class UploadFileServiceTest {
	@TestConfiguration
	static class Config{
		@Bean
		public UploadFileService uploadFileService() {
			return new UploadFileService();
		}
	}
	@Autowired
	private UploadFileService uploadFileService;
	
	@MockBean
	private FileRepository fileRepository;

	@Test
	public void testSaveFile() {
		Blog blog=new Blog();
		FileEntity file=new FileEntity();
		List<MultipartFile> files=new ArrayList<>();
		files.add(new MockMultipartFile("file", "test.text","text/plain","Spring Framework".getBytes()));
		uploadFileService.saveFile(files,blog);
		verify(fileRepository).save(file);
	}
}

UploadFileServiceクラスのsaveFileメソッドを呼び出して引数にファイルとブログオブジェクトを渡したらfileRepositoryクラスのsaveメソッドが呼び出されることをテストしたいのですが、
上記テストコードだと「fileとsaveFileされたオブジェクトは違うものです」って出ます。
uploadFileService.saveFile(files,blog)でインスタンス化されたオブジェクトを取り出してsave()に入れたらいいと思うのですが、どう書けばいいのかわからないです。
ご教授宜しくお願い致します。

開発環境
spring boot 2.2.12
java 8
mariaDB 10.4.17

Comments

[kiino]

> UploadFileServiceクラスsaveFile()メソッド内でfileEをインスタンス化して

質問文が途中で途切れているような印象を受けました。
どんなテストを書きたいのか、どんか課題をもっているのか、わからなかったので記入途中であれば記入お願いします！

[chikara]

教えて頂きありがとうございます！
書いてる途中にネットワーク接続が切れてたみたいですorz
追記致しました。

[kiino]

修正ありがとうございます！

Answer 1

検査で失敗している原因は、verify する時の file が saveFile 内で変更される処理内容と合致していないためだと思われます。

具体的には以下の2点かと思われます。

• new Blog() で生成したインスタンスを file にセットしていない
• mfiles 中のファイルを Base64 エンコード化した文字列を base64_str に設定していない

lombok の @Data をつけているので、Hash 生成メソッドも自動生成されているので、設定するメンバ変数が設定されていれば、verify の比較でも合致できるようになると思います。

私の手元では以上の変更でテストが成功するのを確認しました。

Comments

[chikara]

返信遅くなり申し訳ありませんorz

@kiinoさんのおっしゃる通りにやってみたらできました！
ありがとうございます。！！
「mfiles 中のファイルを Base64 エンコード化した文字列を base64_str に設定していない」
の部分は、
StringBuffer data = new StringBuffer();
String base64 = new String(Base64.encodeBase64("Spring Framework".getBytes()),"ASCII");
data.append("data:image/jpeg;base64,");
data.append(base64);
file.setBase64_str(data.toString());
を追記するで合ってますでしょうか？

[kiino]

すいません。回答への質問にずいぶん経ってしまいました。

はい、それで問題ありません。
base64 化された値を期待値に入れれば問題ない認識です