aswi2021tri-musketyri-gitlab/Server/ServerAppFunctionalTests/Predictor/PredictionControllerTests.cs @ 3729dcae

using Microsoft.VisualStudio.TestTools.UnitTesting;
using ServerApp.Predictor;
//
// Author: Roman Kalivoda
//

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using ServerApp.WeatherPredictionParser;
using ServerApp.Parser.Parsers;
using ServerApp.Parser.InputData;
using ServerApp.Connection.XMLProtocolHandler;
using ServerApp.DataDownload;

namespace ServerApp.Predictor.Tests
{
    [TestClass()]
    public class PredictionControllerTests
    {
        [TestMethod()]
        public void PredictSingleTimeWeatherTest()
        {
            System.IO.Directory.CreateDirectory(@".\data");
            DataDownloader dd = new DataDownloader(@".\data\dd\", "http://openstore.zcu.cz/", "OD_ZCU_{type}_{month}_{year}_{format}.zip");
            PredictionController instance = new PredictionController(new JsonParser(dd, new CsvDataLoader()), new DataParser(dd));

            dd.DownloadData(DataType.JIS, DataFormat.CSV, new DataDownload.Date(1, 2019), new DataDownload.Date(12, 2020));
            dd.DownloadData(DataType.STROJE, DataFormat.CSV, new DataDownload.Date(1, 2017), new DataDownload.Date(12, 2020));
            dd.DownloadData(DataType.POCASI, DataFormat.CSV, new DataDownload.Date(1, 2017), new DataDownload.Date(12, 2020));
            instance.Train();

            Request request = new Request()
            {
                useEndDate = false,
                useWeather = false,
                start = new Connection.XMLProtocolHandler.Date()
                {
                    day = DateTime.Now.Day,
                    month = DateTime.Now.Month,
                    year = DateTime.Now.Year,
                    hour = 10
                }
            };
            Response actual = instance.Predict(request);
            CollectionAssert.AllItemsAreInstancesOfType(actual.predicitons, typeof(Prediction));
            Assert.AreEqual(1, actual.predicitons.Length);
        }
    }
}