package com.pprokurat.api.service;

import com.pprokurat.api.exception.FolderNotFoundException;
import com.pprokurat.api.model.Folder;
import com.pprokurat.api.repository.FolderRepository;
import org.junit.After;
import org.junit.Assert;
import org.junit.Before;
import org.junit.Test;

import java.util.ArrayList;
import java.util.List;

/**
 * Created by IntelliJ IDEA.
 *
 * @author Michał Gogolewski
 */
public class FoldersServiceImplTest {

    private FolderRepositoryStub repository = new FolderRepositoryStub();
    private FolderServiceImpl service = new FolderServiceImpl(repository);

    @Before
    public void setUp() throws Exception {
        repository.setup();
    }

    @After
    public void tearDown() throws Exception {
        repository.clear();
    }

    @Test
    public void getFoldersList_shouldThrowException_repositoryReturnsEmptyList() {
        //given
        //when
        try {
            service.getFoldersList();
            //then
            Assert.fail("Expected an FolderNotFoundException to be thrown");
        } catch (FolderNotFoundException folderNotFoundException) {
            //success
        }
    }

    @Test
    public void getFoldersList_shouldThrowException_repositoryReturnsNull() {
        //given
        repository.clear();
        //when
        try {
            service.getFoldersList();
            //then
            Assert.fail("Expected an FolderNotFoundException to be thrown");
        } catch (FolderNotFoundException folderNotFoundException) {
            //success
        }
    }

    @Test
    public void getFoldersList_shouldReturnListWithCorrectData_repositoryReturnsNotEmptyList() {
        //given
        Folder folder1 = new Folder();
        Folder folder2 = new Folder();
        repository.add(folder1);
        repository.add(folder2);

        //when
        List<Folder> result = service.getFoldersList();
        //then
        Assert.assertEquals(2, result.size());
        Assert.assertEquals(folder1, result.get(0));
        Assert.assertEquals(folder2, result.get(1));
    }

    @Test
    public void getFolderById_shouldThrowException_repositoryReturnsNull() {
        //given
        //when
        try {
            service.getFolderById(1);
            //then
            Assert.fail("Expected an FolderNotFoundException to be thrown");
        } catch (FolderNotFoundException folderNotFoundException) {
            //success
        }
    }

    @Test
    public void getFolderById_shouldReturnObject_repositoryReturnsObject() {
        //given
        Folder folder1 = new Folder();
        Folder folder2 = new Folder();
        folder1.setId(1);
        folder2.setId(2);
        repository.add(folder1);
        repository.add(folder2);

        //when
        Folder result = service.getFolderById(1);
        //then
        Assert.assertEquals(folder1, result);
    }

    private static class FolderRepositoryStub implements FolderRepository {
        private List<Folder> folders;

        public void setup() {
            folders = new ArrayList<Folder>();
        }

        public void clear() {
            folders = null;
        }

        public void add(Folder folder) {
            folders.add(folder);
        }

        public List<Folder> getAll() {
            return folders;
        }

        public Folder getOneById(int id) {
            for (Folder folder : folders) {
                if (folder.getId() == id) {
                    return folder;
                }
            }
            return null;
        }
    }
}