#include "wagon.hpp"
#include "passager.hpp"
#include "train.hpp"
#include <iostream>
#include <string>


using namespace std;

void testPassager(void) {
    Passager lePassager("Nom", "Prenom");
    cout << lePassager << endl;
}

void testWagon(void) {
    Wagon w1(1);
    Wagon w2(2);
    Wagon w3(3);
    Passager lePassager("Maurice", "Mitchel");
    Passager lePassager2("Pierre", "Roberts");
    Passager lePassager3("Jean", "Dupont");
    w1.ajouter(lePassager);
    w1.ajouter(lePassager2);
    w3.ajouter(lePassager3);
    w1.enlever(lePassager2);
    cout << lePassager3 << endl;
    cout << w1 << endl;
    cout << w2 << endl;
    cout << w3 << endl;
}

void testTrain(void) {
    Train leTrain(3);
    Passager lePassager("Maurice", "Mitchel");
    Passager lePassager2("Pierre", "Roberts");
    Passager lePassager3("Jean", "Dupont");
    Passager lePassager4("Paul", "Dupond");
    Passager lePassager5("Jacques", "Dupont");
    leTrain.monterDansLeTrain(1,lePassager);
    leTrain.monterDansLeTrain(1,lePassager2);
    leTrain.monterDansLeTrain(3,lePassager3);
    leTrain.deplacerAuWagonSuivant(lePassager);
    leTrain.descendreDuTrain(lePassager2);
    leTrain.deplacerAuWagonPrecedent(lePassager3);
    leTrain.monterDansLeTrainAPartirDe(2,lePassager4);
    cout << leTrain << endl;
}

int main() {
    // testPassager();
    // testWagon();
    testTrain();
    return 0;
}