TestDataLoader.java

package roomescape;

import org.springframework.boot.CommandLineRunner;
import org.springframework.context.annotation.Profile;
import org.springframework.stereotype.Component;
import roomescape.member.Member;
import roomescape.member.MemberRepository;
import roomescape.reservation.Reservation;
import roomescape.reservation.ReservationRepository;
import roomescape.theme.Theme;
import roomescape.theme.ThemeRepository;
import roomescape.time.Time;
import roomescape.time.TimeRepository;

@Profile("test")
@Component
public class TestDataLoader implements CommandLineRunner {

    private TimeRepository timeRepository;
    private ThemeRepository themeRepository;
    private ReservationRepository reservationRepository;
    private MemberRepository memberRepository;

    public TestDataLoader(TimeRepository timeRepository,
                          ThemeRepository themeRepository,
                          ReservationRepository reservationRepository,
                          MemberRepository memberRepository) {
        this.timeRepository = timeRepository;
        this.themeRepository = themeRepository;
        this.reservationRepository = reservationRepository;
        this.memberRepository = memberRepository;
    }

    @Override
    public void run(String... args) throws Exception {
        Member admin = memberRepository.save(new Member("어드민", "admin@email.com", "password", "ADMIN"));
        Member testUser1 = memberRepository.save(new Member("브라운", "brown@email.com", "password", "USER"));

        final Theme theme1 = themeRepository.save(new Theme("테마1", "테마1입니다."));
        final Theme theme2 = themeRepository.save(new Theme("테마2", "테마2입니다."));
        final Theme theme3 = themeRepository.save(new Theme("테마3", "테마3입니다."));

        final Time time1 = timeRepository.save(new Time("10:00"));
        final Time time2 = timeRepository.save(new Time("12:00"));
        final Time time3 = timeRepository.save(new Time("14:00"));
        final Time time4 = timeRepository.save(new Time("16:00"));
        final Time time5 = timeRepository.save(new Time("18:00"));
        final Time time6 = timeRepository.save(new Time("20:00"));

        reservationRepository.save(new Reservation("어드민", "2024-03-01", time1, theme1, admin));
        reservationRepository.save(new Reservation("어드민", "2024-03-01", time2, theme2, admin));
        reservationRepository.save(new Reservation("어드민", "2024-03-01", time3, theme3, admin));
    }
}