#include "pch.h"
#include "CppUnitTest.h"
#include "../Course_work/FordFulkerson.cpp"

using namespace Microsoft::VisualStudio::CppUnitTestFramework;

using namespace Microsoft::VisualStudio::CppUnitTestFramework;

namespace UnitTest1
{
    TEST_CLASS(UnitTest1)
    {
    public:

        TEST_METHOD(Overall_test_insert_1)
        {
            FordFulkerson g;

            g.insert_edge('0', '1', 16);
            g.insert_edge('0', '2', 13);
            g.insert_edge('1', '2', 10);
            g.insert_edge('2', '1', 4);
            g.insert_edge('1', '3', 12);
            g.insert_edge('2', '4', 14);
            g.insert_edge('3', '2', 9);
            g.insert_edge('3', '5', 20);
            g.insert_edge('4', '3', 7);
            g.insert_edge('4', '5', 4);

            Assert::IsTrue(g.fordFulkerson('0', '5') == 23);
        }

        TEST_METHOD(Overall_test_insert_2)
        {
            FordFulkerson g;

            g.insert_edge('1', '2', 20);
            g.insert_edge('1', '3', 10);
            g.insert_edge('2', '3', 30);
            g.insert_edge('2', '4', 10);
            g.insert_edge('3', '4', 20);

            Assert::IsTrue(g.fordFulkerson('1', '4') == 30);
        }

        TEST_METHOD(Overall_test_insert_no_path)
        {
            FordFulkerson g;

            g.insert_edge('0', '1', 16);
            g.insert_edge('0', '2', 13);
            g.insert_edge('3', '2', 9);
            g.insert_edge('3', '5', 20);
            g.insert_edge('4', '3', 7);
            g.insert_edge('4', '5', 4);

            Assert::IsTrue(g.fordFulkerson('0', '5') == 0);
        }

        TEST_METHOD(Overall_test_insert_wrong_stock)
        {
            try
            {
                FordFulkerson g;

                g.insert_edge('1', '2', 20);

                Assert::IsTrue(g.fordFulkerson('5', '5') == 30);
            }
            catch (std::exception e)
            {
                Assert::AreEqual(e.what(), "No vertex with this letter.");
            }
        }

        TEST_METHOD(Overall_test_read_all_ok)
        {
            FordFulkerson g;
            g.read("C:\\Users\\besth\\OneDrive\\Visual Studio 2019\\Projects\\AISTRD_2\\Course_work\\Read.txt");

            Assert::IsTrue(g.fordFulkerson('1', '4') == 30);
        }

        TEST_METHOD(Overall_test_read_empty)
        {
            try
            {
                FordFulkerson g;
                g.read("C:\\Users\\besth\\OneDrive\\Visual Studio 2019\\Projects\\AISTRD_2\\Course_work\\ReadEmpty.txt");
            }
            catch (std::exception e)
            {
                Assert::AreEqual(e.what(), "File is empty.");
            }
        }

        TEST_METHOD(Overall_test_read_wrong_data1)
        {
            try
            {
                FordFulkerson g;
                g.read("C:\\Users\\besth\\OneDrive\\Visual Studio 2019\\Projects\\AISTRD_2\\Course_work\\ReadWrongData1.txt");
            }
            catch (std::exception e)
            {
                Assert::AreEqual(e.what(), "Wrong input data in file.");
            }
        }

        TEST_METHOD(Overall_test_read_wrong_data2)
        {
            try
            {
                FordFulkerson g;
                g.read("C:\\Users\\besth\\OneDrive\\Visual Studio 2019\\Projects\\AISTRD_2\\Course_work\\ReadWrongData2.txt");
            }
            catch (std::exception e)
            {
                Assert::AreEqual(e.what(), "Wrong input data in file.");
            }
        }

        TEST_METHOD(Overall_test_read_wrong_data3)
        {
            try
            {
                FordFulkerson g;
                g.read("C:\\Users\\besth\\OneDrive\\Visual Studio 2019\\Projects\\AISTRD_2\\Course_work\\ReadWrongData3.txt");
            }
            catch (std::exception e)
            {
                Assert::AreEqual(e.what(), "Wrong input data in file.");
            }
        }

        TEST_METHOD(Overall_test_not_exist)
        {
            try
            {
                FordFulkerson g;
                g.read("NonExist.txt");
            }
            catch (std::exception e)
            {
                Assert::AreEqual(e.what(), "No file found.");
            }
        }
    };
}