require("../env");

require("../../d3");

var vows = require("vows"),

    assert = require("assert");

var suite = vows.describe("selection.selectAll");

suite.addBatch({

  "select(body)": {

    topic: function() {

      var body = d3.select("body").html("");

      if ("__data__" in body[0][0]) delete body[0][0].__data__;

      body.append("div").attr("class", "first");

      body.append("div").attr("class", "second");

      return body;

    },

    "selects all matching elements": function(body) {

      var div = body.selectAll("div");

      assert.isTrue(div[0][0] === document.body.firstChild);

      assert.isTrue(div[0][1] === document.body.lastChild);

      assert.equal(div.length, 1);

      assert.equal(div[0].length, 2);

    },

    "propagates parent node to the selected elements": function(body) {

      var div = body.selectAll("div");

      assert.isNotNull(div[0].parentNode);

      assert.isTrue(div[0].parentNode === document.body);

    },

    "does not propagate data if data was specified": function(body) {

      var div = body.data([new Object()]).selectAll("div");

      assert.isUndefined(div[0][0].__data__);

      assert.isUndefined(div[0][1].__data__);

    },

    "does not propagate data if data was not specified": function(body) {

      var div = body.selectAll("div").data([1, 2]);

      div = body.data([new Object()]).selectAll("div");

      assert.equal(div[0][0].__data__, 1);

      assert.equal(div[0][1].__data__, 2);

    },

    "returns empty array if no match is found": function(body) {

      var span = body.selectAll("span");

      assert.equal(span.length, 1);

      assert.equal(span[0].length, 0);

    },

    "can select via function": function(body) {

      var d = {}, dd = [], ii = [], tt = [],

          s = body.data([d]).selectAll(function(d, i) { dd.push(d); ii.push(i); tt.push(this); return this.childNodes; });

      assert.deepEqual(dd, [d], "expected data, got {actual}");

      assert.deepEqual(ii, [0], "expected index, got {actual}");

      assert.domEqual(tt[0], document.body, "expected this, got {actual}");

      assert.domEqual(s[0][0], document.body.firstChild);

      assert.domEqual(s[0][1], document.body.lastChild);

      delete document.body.__data__;

    }

  }

});

suite.addBatch({

  "selectAll(div)": {

    topic: function() {

      var div = d3.select("body").html("").selectAll("div").data(d3.range(2)).enter().append("div");

      div.append("span").attr("class", "first");

      div.append("span").attr("class", "second");

      return div;

    },

    "selects all matching elements": function(div) {

      var span = div.selectAll("span");

      assert.equal(span.length, 2);

      assert.equal(span[0].length, 2);

      assert.isTrue(span[0][0].parentNode === div[0][0]);

      assert.isTrue(span[0][1].parentNode === div[0][0]);

      assert.isTrue(span[1][0].parentNode === div[0][1]);

      assert.isTrue(span[1][1].parentNode === div[0][1]);

    },

    "propagates parent node to the selected elements": function(div) {

      var span = div.selectAll("span");

      assert.isNotNull(span[0].parentNode);

      assert.isTrue(span[0].parentNode === div[0][0]);

      assert.isTrue(span[1].parentNode === div[0][1]);

    },

    "does not propagate data if data was specified": function(div) {

      var span = div.selectAll("span");

      delete span[0][0].__data__;

      delete span[0][1].__data__;

      span = div.data([new Object(), new Object()]).selectAll("span");

      assert.isUndefined(span[0][0].__data__);

      assert.isUndefined(span[0][1].__data__);

    },

    "does not propagate data if data was not specified": function(div) {

      var a = new Object(), b = new Object(), span = div.selectAll("span").data([a, b]);

      delete div[0][0].__data__;

      delete div[0][1].__data__;

      span = div.selectAll("span");

      assert.equal(span[0][0].__data__, a);

      assert.equal(span[0][1].__data__, b);

    },

    "returns empty array if no match is found": function(div) {

      var div = div.selectAll("div");

      assert.equal(div.length, 2);

      assert.equal(div[0].length, 0);

      assert.equal(div[1].length, 0);

    },

    "ignores null nodes": function(div) {

      var some = d3.selectAll("div");

      some[0][1] = null;

      var span = some.selectAll("span");

      assert.equal(span.length, 1);

      assert.equal(span[0].length, 2);

      assert.isTrue(span[0][0].parentNode === div[0][0]);

      assert.isTrue(span[0][1].parentNode === div[0][0]);

    },

    "can select via function": function(div) {

      var dd = [], ii = [], tt = [],

          s = div.data(["a", "b"]).selectAll(function(d, i) { dd.push(d); ii.push(i); tt.push(this); return this.childNodes; });

      assert.deepEqual(dd, ["a", "b"], "expected data, got {actual}");

      assert.deepEqual(ii, [0, 1], "expected index, got {actual}");

      assert.domEqual(tt[0], div[0][0], "expected this, got {actual}");

      assert.domEqual(tt[1], div[0][1], "expected this, got {actual}");

      assert.domEqual(s[0][0], div[0][0].firstChild);

      assert.domEqual(s[0][1], div[0][0].lastChild);

      assert.domEqual(s[1][0], div[0][1].firstChild);

      assert.domEqual(s[1][1], div[0][1].lastChild);

    }

  }

});

suite.export(module);