require("../env");

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

var vows = require("vows"),

    assert = require("assert");

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

suite.addBatch({

  "select(body)": {

    topic: function() {

      return d3.select("body").html("");

    },

    "updates the data according to the map function": function(body) {

      body.data([42]).map(function(d, i) { return d + i; });

      assert.equal(document.body.__data__, 42);

    },

    "returns the current selection": function(body) {

      assert.isTrue(body.map(function() { return 1; }) === body);

    }

  }

});

suite.addBatch({

  "selectAll(div)": {

    topic: function() {

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

    },

    "updates the data according to the map function": function(div) {

      div.data([42, 43]).map(function(d, i) { return d + i; });

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

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

    },

    "returns the current selection": function(div) {

      assert.isTrue(div.map(function() { return 1; }) === div);

    },

    "ignores null nodes": function(div) {

      var some = d3.selectAll("div").data([42, 43]);

      some[0][1] = null;

      some.map(function() { return 1; });

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

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

    }

  }

});

suite.export(module);