Testing the Visitor
To test the example we can add some code to the Main method of the console application. The following code creates five employees in a three-level hierarchy. Once the structure is created, a payroll visitor is applied to the organisational structure to output the salary details for each employee. Next, a pay rise visitor increases the monthly salary for each employee by 5%. The payroll visitor is then used again to display the new salaries. Finally, the total increase in salary is displayed.
Manager bob = new Manager();
bob.Name = "Bob";
bob.MonthlySalary = 5000;
Manager sue = new Manager();
sue.Name = "Sue";
sue.MonthlySalary = 4000;
Worker jim = new Worker();
jim.Name = "Jim";
jim.MonthlySalary = 2000;
Worker tom = new Worker();
tom.Name = "Tom";
tom.MonthlySalary = 1800;
Worker mel = new Worker();
mel.Name = "Mel";
mel.MonthlySalary = 1900;
bob.Subordinates.Add(sue);
bob.Subordinates.Add(jim);
sue.Subordinates.Add(tom);
sue.Subordinates.Add(mel);
OrganisationalStructure org = new OrganisationalStructure(bob);
PayrollVisitor payroll = new PayrollVisitor();
PayriseVisitor payrise = new PayriseVisitor(0.05);
org.Accept(payroll);
org.Accept(payrise);
org.Accept(payroll);
Console.WriteLine("Total pay increase = {0}.", payrise.TotalIncrease);
/* OUTPUT
Bob paid 5000 + Bonus.
Sue paid 4000 + Bonus.
Tom paid 1800.
Mel paid 1900.
Jim paid 2000.
Bob salary increased by 250.
Sue salary increased by 200.
Tom salary increased by 90.
Mel salary increased by 95.
Jim salary increased by 100.
Bob paid 5250 + Bonus.
Sue paid 4200 + Bonus.
Tom paid 1890.
Mel paid 1995.
Jim paid 2100.
Total pay increase = 735.
*/21 August 2009