#include "BuddyList.h"
#include <iostream>
using namespace std;

int main(int argc, char** argv)
{
  BuddyList buddies;

  buddies.addBuddy("Harry Potter", "Ron Weasley");
  buddies.addBuddy("Harry Potter", "Hermione Granger");
  buddies.addBuddy("Harry Potter", "Hagrid");
  buddies.addBuddy("Harry Potter", "Draco Malfoy");
  // That's not right! Remove Draco
  buddies.removeBuddy("Harry Potter", "Draco Malfoy");
  
  buddies.addBuddy("Hagrid", "Harry Potter");
  buddies.addBuddy("Hagrid", "Ron Weasley");
  buddies.addBuddy("Hagrid", "Hermione Granger");

  list<string> harryBuds = buddies.getBuddies("Harry Potter");

  cout << "Harry's friends: \n";
  for (list<string>::const_iterator it = harryBuds.begin();
       it != harryBuds.end(); ++it) {
    cout << "\t" << *it << endl;
  }

  return (0);
}