ProfiledClassWithMethodTest.java

package de.twenty11.unitprofile;

import static org.hamcrest.Matchers.greaterThanOrEqualTo;
import static org.hamcrest.Matchers.is;
import static org.hamcrest.Matchers.not;
import static org.hamcrest.Matchers.nullValue;
import static org.junit.Assert.assertThat;

import java.util.List;

import org.junit.Ignore;
import org.junit.Test;

import de.twenty11.unitprofile.agent.Agent;
import de.twenty11.unitprofile.domain.MethodDescriptor;
import de.twenty11.unitprofile.domain.MethodInvocation;
import de.twenty11.unitprofile.helper.ExtendedTestClass;
import de.twenty11.unitprofiler.annotations.Profile;

public class ProfiledClassWithMethodTest {

    //@Profile
    @Test
    @Ignore
    public void testProfiler () {
        sleep20();
        sleep20();
        sleep40();
        
        List<MethodDescriptor> instrumentations = Agent.getInstrumentations();
        assertThat(instrumentations, is(not(nullValue())));
        
        MethodInvocation rootInvocation = Agent.getRootInvocation();
        assertThat(rootInvocation, is(not(nullValue())));
        assertThat(rootInvocation.getChildren().size(), is(2));
        assertThat(rootInvocation.getChildren().get(0).getChildren().size(), is(0));
        assertThat(rootInvocation.getTime(),is(greaterThanOrEqualTo(rootInvocation.getChildren().get(0).getTime())));
    }
    
    @Test
    @Profile
    public void testCorrectClassAndMethodName() {
        ExtendedTestClass extendedTestClass = new ExtendedTestClass(10);
        extendedTestClass.sleep(200);
        
        MethodInvocation rootInvocation = Agent.getRootInvocation();
        assertThat(rootInvocation, is(not(nullValue())));
    }
    
    private void sleep20() {
        try {
            Thread.sleep(20);
        } catch (InterruptedException e) {}
    }

    private void sleep40() {
        try {
            Thread.sleep(40);
        } catch (InterruptedException e) {}
    }

}