/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/*
 * Test suite for nsAbManager functions relating to listeners.
 */

var nsIAbListener = Components.interfaces.nsIAbListener;
var numListenerOptions = 4;

var gAblAll;
var gAblSingle = new Array(numListenerOptions);

function abL() {}

abL.prototype = {
 mReceived: 0,
 mAutoRemoveItem: false,

  onItemAdded: function (parentItem, item) {
    this.mReceived |= nsIAbListener.itemAdded;
    if (this.mAutoRemoveItem)
      MailServices.ab.removeAddressBookListener(this);
  },
  onItemRemoved: function (parentItem, item) {
    this.mReceived |=
      (item == MailServices.ab ? nsIAbListener.directoryRemoved :
                                 nsIAbListener.directoryItemRemoved);
    if (this.mAutoRemoveItem)
      MailServices.ab.removeAddressBookListener(this);
  },
  onItemPropertyChanged: function (item, property, oldValue, newValue) {
    this.mReceived |= nsIAbListener.itemChanged;
    if (this.mAutoRemoveItem)
      MailServices.ab.removeAddressBookListener(this);
  }
};

function NotifyAbManager() {
  MailServices.ab.notifyItemPropertyChanged(null, null, null, null);
  MailServices.ab.notifyDirectoryItemAdded(null, null);
  MailServices.ab.notifyDirectoryItemDeleted(null, null);
  // MailServices.ab just happens to be nsISupports derived and makes it easy for
  // us to distinguish between xxxItemDeleted and xxxDeleted.
  MailServices.ab.notifyDirectoryDeleted(null, MailServices.ab);
}

function run_test() {
  var i;

  // Test - Add a listener

  gAblAll = new abL;

  MailServices.ab.addAddressBookListener(gAblAll, nsIAbListener.all);

  for (i = 0; i < numListenerOptions; ++i) {
    gAblSingle[i] = new abL;
    MailServices.ab.addAddressBookListener(gAblSingle[i], 1 << i);
  }

  // Test - Notify listener on all available items

  NotifyAbManager();

  Assert.equal(gAblAll.mReceived, (1 << numListenerOptions) - 1);
  gAblAll.mReceived = 0;

  for (i = 0; i < numListenerOptions; ++i) {
    Assert.equal(gAblSingle[i].mReceived, 1 << i);
    gAblSingle[i].mReceived = 0;

    // And prepare for test 3.
    gAblSingle[i].mAutoRemoveItem = true;
  }

  // Test - Remove Single Listeners as we go through the functions

  // Check the for loop above for changes to the single listeners.

  NotifyAbManager();

  Assert.equal(gAblAll.mReceived, (1 << numListenerOptions) - 1);
  gAblAll.mReceived = 0;

  for (i = 0; i < numListenerOptions; ++i) {
    Assert.equal(gAblSingle[i].mReceived, 1 << i);
    gAblSingle[i].mReceived = 0;
  }

  // Test - Ensure the single listeners have been removed.

  NotifyAbManager();

  Assert.equal(gAblAll.mReceived, (1 <<  numListenerOptions) - 1);
  gAblAll.mReceived = 0;

  for (i = 0; i < numListenerOptions; ++i) {
    Assert.equal(gAblSingle[i].mReceived, 0);
  }

  // Test - Remove main listener

  MailServices.ab.removeAddressBookListener(gAblAll);
};
