Chatto/chatto/src/test/java/org/ros/chatto/ChattoApplicationTests.java

package org.ros.chatto;

import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.Mock;
import org.mockito.Mockito;
import org.ros.chatto.model.ChatMessage;
import org.ros.chatto.model.ChatUser;
import org.ros.chatto.repository.ChatMessageRepository;
import org.ros.chatto.repository.UserRepository;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.test.context.junit4.SpringRunner;
import static org.mockito.ArgumentMatchers.any;
import static org.mockito.Mockito.times;
import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.when;

@RunWith(SpringRunner.class)
@SpringBootTest
public class ChattoApplicationTests {
	
	@Autowired
	ChatMessageRepository chatMessageRepository;
	
	@Mock
	ChatMessageRepository mockChatMessageRepository;
	
	@Autowired
	UserRepository userRepository;

	@Test
	public void contextLoads() {
	}

	@Test
	public void testMessageRepo() {
		chatMessageRepository.findAll().toString();
	}
	
//	@Test
//	public void testSave() {
//		ChatUser fromUser = new ChatUser();
//		fromUser = userRepository.findByUserName("hmm");
//		ChatUser toUser = new ChatUser();
//		toUser = userRepository.findByUserName("user2");
//		ChatMessage chatMessage = new ChatMessage();
//		chatMessage.setMessage("Hello!");
//		chatMessage.setFromUser(fromUser);
//		chatMessage.setToUser(toUser);
//		
//		chatMessageRepository.save(chatMessage);
//	}
	
	@Test
	public void testSave() {
		ChatUser fromUser = new ChatUser();
		fromUser = userRepository.findByUserName("hmm");
		ChatUser toUser = new ChatUser();
		toUser = userRepository.findByUserName("user2");
		ChatMessage chatMessage = new ChatMessage();
		chatMessage.setMessage("Hello!");
		chatMessage.setFromUser(fromUser);
		chatMessage.setToUser(toUser);
		
//		chatMessageRepository.save(chatMessage);
		when(mockChatMessageRepository.save(any(ChatMessage.class))).thenReturn(chatMessage);
		verify(mockChatMessageRepository, times(1)).save(Mockito.any(ChatMessage.class));
	}
}