TestDataSupport.java

package util;


import com.news_articles.domain.Article;
import com.news_articles.model.ArticleDto;
import com.news_articles.model.ArticlesDto;
import java.util.ArrayList;
import java.util.List;

public class TestDataSupport {

    public static final String SPORT = "sport";
    public static final String BUSINESS = "business";
    public static final String ART = "art";
    public static final String ENTERTAINMENT = "entertainment";

    /**
     * This test support class creates dummy test data for the final task.
     * Each of the Articles has a title, category and shareCount
     * The shareCount is the number of times a user has shared an article
     * A category with the most shareCount would be the user's favourite
     */

    public static ArticlesDto createRequestData() {
        var userID = TestSupportRandom.randomString();

        List<ArticleDto> sport = new ArrayList<>();
        int increment = 10;

        for (int i = 0; i < 5; i++) {
            sport.add(ArticleDto.builder()
                    .title(TestSupportRandom.randomString())
                    .category(SPORT)
                    .shareCount(i + increment)
                    .build());
        }

        List<ArticleDto> business = new ArrayList<>();
        increment = 20;
        for (int i = 0; i < 4; i++) {
            business.add(ArticleDto.builder()
                .title(TestSupportRandom.randomString())
                .category(BUSINESS)
                .shareCount(i + increment)
                .build());
        }

        List<ArticleDto> art = new ArrayList<>();
        increment = 30;
        for (int i = 0; i < 3; i++) {
            art.add(ArticleDto.builder()
                .title(TestSupportRandom.randomString())
                .category(ART)
                .shareCount(i + increment)
                .build());
        }

        List<ArticleDto> entertainment = new ArrayList<>();
        increment = 40;
        for (int i = 0; i < 2; i++) {
            entertainment.add(ArticleDto.builder()
                .title(TestSupportRandom.randomString())
                .category(ART)
                .shareCount(i + increment)
                .build());
        }

        List<ArticleDto> articleDtos = new ArrayList<>();
        articleDtos.addAll(sport);
        articleDtos.addAll(business);
        articleDtos.addAll(art);
        articleDtos.addAll(entertainment);

        return ArticlesDto.builder()
                .userID(userID)
                .articleDtos(articleDtos)
                .build();
    }

    public static List<Article> createArticles() {
        List<Article> sports = new ArrayList<>();
        for (int i = 0; i < 5; i++) {
            sports.add(Article.builder()
                    .title(TestSupportRandom.randomString())
                    .category(SPORT)
                    .shareCount(TestSupportRandom.randomInt())
                    .build());
        }

        List<Article> businesses = new ArrayList<>();
        for (int i = 0; i < 4; i++) {
            businesses.add(Article.builder()
                .title(TestSupportRandom.randomString())
                .category(BUSINESS)
                .shareCount(TestSupportRandom.randomInt())
                .build());
        }

        List<Article> arts = new ArrayList<>();
        for (int i = 0; i < 3; i++) {
            arts.add(Article.builder()
                .title(TestSupportRandom.randomString())
                .category(ART)
                .shareCount(TestSupportRandom.randomInt())
                .build());
        }

        List<Article> entertainments = new ArrayList<>();
        for (int i = 0; i < 2; i++) {
            entertainments.add(Article.builder()
                .title(TestSupportRandom.randomString())
                .category(ENTERTAINMENT)
                .shareCount(TestSupportRandom.randomInt())
                .build());
        }

        List<Article> categories = new ArrayList<>();
        categories.addAll(sports);
        categories.addAll(businesses);
        categories.addAll(arts);
        categories.addAll(entertainments);

        return categories;
    }

}