Home > AI > Backend > SpringBoot > mysql-connector-java >

Test mysql repository layer

Example

Tutorial.java

@Data
@AllArgsConstructor
@NoArgsConstructor
@Entity
@Table(name = "tutorials")
public class Tutorial {
    @Id
    @Column(name = "id")
    @GeneratedValue(strategy = GenerationType.AUTO)
    private long id;

    @Column(name = "title")
    private String title;

    @Column(name = "description")
    private String description;

    @Column(name = "published")
    private boolean published;

    public Tutorial(String title, String description, boolean published) {
        this.title = title;
        this.description = description;
        this.published = published;
    }

}

TutorialRepository.java

public interface TutorialRepository extends JpaRepository<Tutorial, Long> {
    List<Tutorial> findByPublished(boolean published);
    List<Tutorial> findByTitleContaining(String title);
}

TutorialRepositoryTest.java

@RunWith(SpringRunner.class)
@SpringBootTest
public class TutorialRepositoryTest {

    @Autowired
    TutorialRepository repository;


    @Before
    public void setup() {
        repository.deleteAll();
    }

    @Test
    public void should_find_no_tutorials_if_repository_is_empty() {

        Iterable<Tutorial> tutorials = repository.findAll();

        assertThat(tutorials).isEmpty();
    }

    @Test
    public void should_store_a_tutorial() {
        Tutorial tutorial = repository.save(new Tutorial("Tut title", "Tut desc", true));

        assertThat(tutorial).hasFieldOrPropertyWithValue("title", "Tut title");
        assertThat(tutorial).hasFieldOrPropertyWithValue("description", "Tut desc");
        assertThat(tutorial).hasFieldOrPropertyWithValue("published", true);
    }

    @Test
    public void should_find_all_tutorials() {
        Tutorial tut1 = new Tutorial("Tut#1", "Desc#1", true);
        repository.save(tut1);

        Tutorial tut2 = new Tutorial("Tut#2", "Desc#2", false);
        repository.save(tut2);

        Tutorial tut3 = new Tutorial("Tut#3", "Desc#3", true);
        repository.save(tut3);

        Iterable<Tutorial> tutorials = repository.findAll();

        assertThat(tutorials).hasSize(3).contains(tut1, tut2, tut3);
    }

    @Test
    public void should_find_tutorial_by_id() {
        Tutorial tut1 = new Tutorial("Tut#1", "Desc#1", true);
        repository.save(tut1);

        Tutorial tut2 = new Tutorial("Tut#2", "Desc#2", false);
        repository.save(tut2);

        Tutorial foundTutorial = repository.findById(tut2.getId()).get();

        assertThat(foundTutorial).isEqualTo(tut2);
    }

    @Test
    public void should_find_published_tutorials() {
        Tutorial tut1 = new Tutorial("Tut#1", "Desc#1", true);
        repository.save(tut1);

        Tutorial tut2 = new Tutorial("Tut#2", "Desc#2", false);
        repository.save(tut2);

        Tutorial tut3 = new Tutorial("Tut#3", "Desc#3", true);
        repository.save(tut3);

        Iterable<Tutorial> tutorials = repository.findByPublished(true);

        assertThat(tutorials).hasSize(2).contains(tut1, tut3);
    }

    @Test
    public void should_find_tutorials_by_title_containing_string() {
        Tutorial tut1 = new Tutorial("Spring Boot Tut#1", "Desc#1", true);
        repository.save(tut1);

        Tutorial tut2 = new Tutorial("Java Tut#2", "Desc#2", false);
        repository.save(tut2);

        Tutorial tut3 = new Tutorial("Spring Data JPA Tut#3", "Desc#3", true);
        repository.save(tut3);

        Iterable<Tutorial> tutorials = repository.findByTitleContaining("ring");

        assertThat(tutorials).hasSize(2).contains(tut1, tut3);
    }

    @Test
    public void should_update_tutorial_by_id() {
        Tutorial tut1 = new Tutorial("Tut#1", "Desc#1", true);
        repository.save(tut1);

        Tutorial tut2 = new Tutorial("Tut#2", "Desc#2", false);
        repository.save(tut2);

        Tutorial updatedTut = new Tutorial("updated Tut#2", "updated Desc#2", true);

        Tutorial tut = repository.findById(tut2.getId()).get();
        tut.setTitle(updatedTut.getTitle());
        tut.setDescription(updatedTut.getDescription());
        tut.setPublished(updatedTut.isPublished());
        repository.save(tut);

        Tutorial checkTut = repository.findById(tut2.getId()).get();

        assertThat(checkTut.getId()).isEqualTo(tut2.getId());
        assertThat(checkTut.getTitle()).isEqualTo(updatedTut.getTitle());
        assertThat(checkTut.getDescription()).isEqualTo(updatedTut.getDescription());
        assertThat(checkTut.isPublished()).isEqualTo(updatedTut.isPublished());
    }

    @Test
    public void should_delete_tutorial_by_id() {
        Tutorial tut1 = new Tutorial("Tut#1", "Desc#1", true);
        repository.save(tut1);

        Tutorial tut2 = new Tutorial("Tut#2", "Desc#2", false);
        repository.save(tut2);

        Tutorial tut3 = new Tutorial("Tut#3", "Desc#3", true);
        repository.save(tut3);

        repository.deleteById(tut2.getId());

        Iterable<Tutorial> tutorials = repository.findAll();

        assertThat(tutorials).hasSize(2).contains(tut1, tut3);
    }

    @Test
    public void should_delete_all_tutorials() {
        repository.save(new Tutorial("Tut#1", "Desc#1", true));
        repository.save(new Tutorial("Tut#2", "Desc#2", false));

        repository.deleteAll();

        assertThat(repository.findAll()).isEmpty();
    }
}

Leave a Reply