org.powertac.producer.windfarm.WindFarmTest.java Source code

Introduction

Here is the source code for org.powertac.producer.windfarm.WindFarmTest.java

Source

/*******************************************************************************
 * Copyright 2014 Spyros Papageorgiou
 * 
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * 
 *   http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 ******************************************************************************/
package org.powertac.producer.windfarm;

import static org.junit.Assert.*;

import java.io.File;
import java.io.FileWriter;
import java.io.IOException;

import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.powertac.common.Competition;
import org.powertac.common.TimeService;
import org.powertac.common.interfaces.Accounting;
import org.powertac.common.interfaces.ServerConfiguration;
import org.powertac.common.interfaces.TariffMarket;
import org.powertac.common.repo.BrokerRepo;
import org.powertac.common.repo.CustomerRepo;
import org.powertac.common.repo.RandomSeedRepo;
import org.powertac.common.repo.TariffRepo;
import org.powertac.common.repo.TariffSubscriptionRepo;
import org.powertac.common.repo.TimeslotRepo;
import org.powertac.common.repo.WeatherReportRepo;
import org.powertac.producer.utils.Curve;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.test.annotation.DirtiesContext;
import org.springframework.test.context.ContextConfiguration;
import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;

import com.thoughtworks.xstream.XStream;

@RunWith(SpringJUnit4ClassRunner.class)
@ContextConfiguration(locations = { "classpath:test-config.xml" })
@DirtiesContext
public class WindFarmTest {
    @Autowired
    private TimeService timeService;

    @Autowired
    private Accounting mockAccounting;

    @Autowired
    private TariffMarket mockTariffMarket;

    @Autowired
    private ServerConfiguration mockServerProperties;

    @Autowired
    private TariffRepo tariffRepo;

    @Autowired
    private CustomerRepo customerRepo;

    @Autowired
    private TariffSubscriptionRepo tariffSubscriptionRepo;

    @Autowired
    private TimeslotRepo timeslotRepo;

    @Autowired
    private WeatherReportRepo weatherReportRepo;

    @Autowired
    private BrokerRepo brokerRepo;

    @Autowired
    private RandomSeedRepo randomSeedRepo;

    @Before
    public void setup() {

        tariffRepo.recycle();

        tariffSubscriptionRepo.recycle();

        timeslotRepo.recycle();

        weatherReportRepo.recycle();

        randomSeedRepo.recycle();

        brokerRepo.recycle();

        customerRepo.recycle();

        Competition.newInstance("Wind Farm Test");
    }

    @Test
    public void testSerialize() {
        double[] x = { 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25 };
        double[] y = { 0, -66, -166, -288, -473, -709, -1000, -1316, -1651, -1860, -1968, -2000, -2000, -2000,
                -2000, -2000, -2000, -2000, -2000, -2000, -2000, -2000, -2000 };
        Curve c = new Curve(x, y);
        c.setCustomLastValue(0);
        WindFarm wf = new WindFarm();
        WindTurbine wt = new WindTurbine(22, 0.01, -2000, 80, c);
        wf.addWindTurbine(wt);
        wf.addWindTurbine(wt);
        wf.addWindTurbine(wt);
        wf.addWindTurbine(wt);
        wf.addWindTurbine(wt);
        wf.addWindTurbine(wt);
        wf.addWindTurbine(wt);
        wf.addWindTurbine(wt);
        wf.addWindTurbine(wt);
        wf.addWindTurbine(wt);
        wf.addWindTurbine(wt);
        wf.addWindTurbine(wt);
        wf.addWindTurbine(wt);

        XStream xstream = new XStream();
        xstream.autodetectAnnotations(true);
        String out = xstream.toXML(wf);
        wf = (WindFarm) xstream.fromXML(out);

        assertTrue(wf.getTurbineList().get(3).getPowerCurve() != null);
        assertTrue(wf.getTurbineList().get(3).getPowerCurve().getCustomLastValue() == 0);
    }

    @Test
    public void dataGenerateXML() throws IOException {
        double[] x = { 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25 };
        double[] y = { 0, -66, -166, -288, -473, -709, -1000, -1316, -1651, -1860, -1968, -2000, -2000, -2000,
                -2000, -2000, -2000, -2000, -2000, -2000, -2000, -2000, -2000 };
        Curve c = new Curve(x, y);
        c.setCustomLastValue(0);
        WindFarm wf = new WindFarm();
        WindTurbine wt = new WindTurbine(22, 0.01, -2000, 80, c);
        wf.addWindTurbine(wt);
        wf.addWindTurbine(wt);
        wf.addWindTurbine(wt);
        wf.addWindTurbine(wt);
        wf.addWindTurbine(wt);
        wf.addWindTurbine(wt);
        wf.addWindTurbine(wt);
        wf.addWindTurbine(wt);
        wf.addWindTurbine(wt);
        wf.addWindTurbine(wt);
        wf.addWindTurbine(wt);
        wf.addWindTurbine(wt);
        wf.addWindTurbine(wt);
        new File("data/").mkdir();
        FileWriter fw = new FileWriter("data/wind-farm.xml");
        XStream xstream = new XStream();
        xstream.autodetectAnnotations(true);
        xstream.toXML(wf, fw);

        fw.close();
    }

}