/*
 * Copyright 2000-2013 Enonic AS
 * http://www.enonic.com/license
 */
package com.enonic.cms.core.content.binary;

import java.util.Date;

import org.junit.Assert;
import org.junit.Before;
import org.junit.Test;

import com.enonic.cms.core.Path;
import com.enonic.cms.core.PathAndParams;
import com.enonic.cms.core.RequestParameters;
import com.enonic.cms.core.content.ContentEntity;
import com.enonic.cms.core.content.ContentKey;
import com.enonic.cms.core.content.ContentVersionEntity;
import com.enonic.cms.core.content.ContentVersionKey;
import com.enonic.cms.core.content.category.CategoryEntity;
import com.enonic.cms.core.content.contenttype.ContentHandlerEntity;
import com.enonic.cms.core.content.contenttype.ContentHandlerKey;
import com.enonic.cms.core.content.contenttype.ContentHandlerName;
import com.enonic.cms.core.content.contenttype.ContentTypeEntity;

/**
 * Created by IntelliJ IDEA.
 * User: rmh
 * Date: Dec 17, 2009
 * Time: 11:38:02 AM
 */
public class AttachmentRequestResolverTest

{

    ContentEntity content;

    ContentVersionEntity contentVersion;

    ContentTypeEntity contentType;

    CategoryEntity category;

    ContentHandlerEntity contentHandler;


    @Before
    public void setUp()
    {
        content = new ContentEntity();
        content.setKey( new ContentKey( 123 ) );

        contentHandler = createContentHandler( "File content", ContentHandlerName.FILE.getHandlerClassShortName() );

        contentType = new ContentTypeEntity();
        contentType.setContentHandler( contentHandler );

        category = new CategoryEntity();
        category.setContentType( contentType );

        content.setCategory( category );

        contentVersion = new ContentVersionEntity();
        contentVersion.setKey( new ContentVersionKey( "1" ) );

        content.setMainVersion( contentVersion );

    }

    @Test
    public void testNativeLinkWithNoLabel()
    {

        AttachmentRequestResolver attachmentRequestResolver = new AttachmentRequestResolver()
        {
            @Override
            protected BinaryDataKey getBinaryData( ContentEntity content, String label )
            {
                Assert.assertEquals( label, "source" );

                return new BinaryDataKey( "1" );
            }

            @Override
            protected ContentEntity getContent( ContentKey contentKey )
            {
                return content;
            }
        };

        attachmentRequestResolver.resolveBinaryDataKey( new PathAndParams( new Path( "_attachment/123" ), new RequestParameters() ) );
    }

    protected ContentHandlerEntity createContentHandler( String name, String handlerClassName )
    {
        ContentHandlerName contentHandlerName = ContentHandlerName.parse( handlerClassName );
        ContentHandlerEntity contentHandler = new ContentHandlerEntity();
        contentHandler.setKey( new ContentHandlerKey( 1 ) );
        contentHandler.setName( name );
        contentHandler.setClassName( contentHandlerName.getHandlerClassShortName() );
        contentHandler.setTimestamp( new Date() );
        return contentHandler;
    }

}