/*******************************************************************************
 * Copyright (c) 2012-2017 Codenvy, S.A.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors:
 *   Codenvy, S.A. - initial API and implementation
 *******************************************************************************/
package org.eclipse.che.api.vfs.search;

import com.tngtech.java.junit.dataprovider.DataProvider;
import com.tngtech.java.junit.dataprovider.DataProviderRunner;
import com.tngtech.java.junit.dataprovider.UseDataProvider;

import org.eclipse.che.api.vfs.VirtualFile;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;

import java.io.ByteArrayInputStream;

import static org.junit.Assert.assertEquals;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.when;

/**
 * @author Valeriy Svydenko
 */
@RunWith(DataProviderRunner.class)
public class MediaTypeFilterTest {


    @DataProvider
    public static Object[][] testData() throws Exception {
        return new Object[][]{
       {virtualFileWithContent("to be or not to be".getBytes()), false},
       {virtualFileWithContent("<html><head></head></html>".getBytes()), false},
       {virtualFileWithContent("<a><b/></a>".getBytes()), false},
       {virtualFileWithContent("public class SomeClass {}".getBytes()), false},
       {virtualFileWithContent(new byte[10]), true}
        };
    }

    private static VirtualFile virtualFileWithContent(byte[] content) throws Exception {
        VirtualFile virtualFile = mock(VirtualFile.class);
        when(virtualFile.getContent()).thenReturn(new ByteArrayInputStream(content));
        return virtualFile;
    }

    private MediaTypeFilter mediaTypeFilter;

    @Before
    public void setUp() throws Exception {
        mediaTypeFilter = new MediaTypeFilter();
    }

    @UseDataProvider("testData")
    @Test
    public void testFilesShouldAccepted(VirtualFile virtualFile, boolean expectedResult) throws Exception {
        assertEquals(expectedResult, mediaTypeFilter.accept(virtualFile));
    }
}