/**
 * CopyRight by Chinamobile
 */
package com.chinamobile.bcbsp.bspstaff;

import org.junit.After;
import org.junit.Before;
import org.junit.Test;

import com.chinamobile.bcbsp.api.Vertex;
import com.chinamobile.bcbsp.examples.AggregateValueVertexNum;
import com.chinamobile.bcbsp.examples.PREdge;
import com.chinamobile.bcbsp.examples.PRVertex;

import junit.framework.TestCase;

public class TestAggregationContext extends TestCase {
    
    private AggregationContext context;
    @SuppressWarnings("unchecked")
    private Vertex vertex;
    private int currentSuperStepCounter = 5;
    private AggregateValueVertexNum aggValue;
    
    @SuppressWarnings("unchecked")
    @Before
    public void setUp() throws Exception {
        vertex = new PRVertex();
        vertex.setVertexID(100);
        vertex.setVertexValue(10.0f);
        PREdge edge = new PREdge();
        edge.setVertexID(200);
        vertex.addEdge(edge);
        context = new AggregationContext(null, vertex, currentSuperStepCounter);
        
        aggValue = new AggregateValueVertexNum();
        aggValue.setValue(1L);
        context.addAggregateValues("SUM", aggValue);
    }
    
    @After
    public void tearDown() throws Exception {
        // do nothing.
    }
    
    @Test
    public void testGetCurrentSuperStepCounter() {
        assertEquals(context.getCurrentSuperStepCounter(), currentSuperStepCounter);
    }
    
    @Test
    public void testGetOutgoingEdgesNum() {
        assertEquals(context.getOutgoingEdgesNum(), 1);
    }
    
    @Test
    public void testGetVertexID() {
        assertEquals(context.getVertexID(), vertex.getVertexID().toString());
    }
    
    @Test
    public void testGetVertexValue() {
        assertEquals(context.getVertexValue(), vertex.getVertexValue().toString());
    }
    
    @Test
    public void testGetAggregateValue() {
        assertEquals(context.getAggregateValue("SUM").getValue(), aggValue.getValue());
    }
}