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

var vows = require("vows"),
    assert = require("assert");

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

suite.addBatch({
  "select(body)": {
    topic: function() {
      return d3.select("body").html("");
    },
    "adds a missing class as true": function(body) {
      body.attr("class", null);
      body.classed("foo", true);
      assert.equal(document.body.className, "foo");
      body.classed("bar", true);
      assert.equal(document.body.className, "foo bar");
    },
    "adds a missing class as a function": function(body) {
      body.attr("class", null);
      body.classed("foo", function() { return true; });
      assert.equal(document.body.className, "foo");
      body.classed("bar", function() { return true; });
      assert.equal(document.body.className, "foo bar");
    },
    "removes an existing class as false": function(body) {
      body.attr("class", "bar foo");
      body.classed("foo", false);
      assert.equal(document.body.className, "bar");
      body.classed("bar", false);
      assert.equal(document.body.className, "");
    },
    "removes an existing class as a function": function(body) {
      body.attr("class", "bar foo");
      body.classed("foo", function() { return false; });
      assert.equal(document.body.className, "bar");
      body.classed("bar", function() { return false; });
      assert.equal(document.body.className, "");
    },
    "preserves an existing class as true": function(body) {
      body.attr("class", "bar foo");
      body.classed("foo", true);
      assert.equal(document.body.className, "bar foo");
      body.classed("bar", true);
      assert.equal(document.body.className, "bar foo");
    },
    "preserves an existing class as a function": function(body) {
      body.attr("class", "bar foo");
      body.classed("foo", function() { return true; });
      assert.equal(document.body.className, "bar foo");
      body.classed("bar", function() { return true; });
      assert.equal(document.body.className, "bar foo");
    },
    "preserves a missing class as false": function(body) {
      body.attr("class", "baz");
      body.classed("foo", false);
      assert.equal(document.body.className, "baz");
      body.attr("class", null);
      body.classed("bar", false);
      assert.equal(document.body.className, "");
    },
    "preserves a missing class as a function": function(body) {
      body.attr("class", "baz");
      body.classed("foo", function() { return false; });
      assert.equal(document.body.className, "baz");
      body.attr("class", null);
      body.classed("bar", function() { return false; });
      assert.equal(document.body.className, "");
    },
    "gets an existing class": function(body) {
      body.attr("class", " foo\tbar  baz");
      assert.isTrue(body.classed("foo"));
      assert.isTrue(body.classed("bar"));
      assert.isTrue(body.classed("baz"));
    },
    "does not get a missing class": function(body) {
      body.attr("class", " foo\tbar  baz");
      assert.isFalse(body.classed("foob"));
      assert.isFalse(body.classed("bare"));
      assert.isFalse(body.classed("rbaz"));
    },
    "returns the current selection": function(body) {
      assert.isTrue(body.classed("foo", true) === body);
    }
  }
});

suite.addBatch({
  "selectAll(div)": {
    topic: function() {
      return d3.select("body").html("").selectAll("div").data(d3.range(2)).enter().append("div");
    },
    "adds a missing class as true": function(div) {
      div.attr("class", null);
      div.classed("foo", true);
      assert.equal(div[0][0].className, "foo");
      assert.equal(div[0][1].className, "foo");
      div.classed("bar", true);
      assert.equal(div[0][0].className, "foo bar");
      assert.equal(div[0][1].className, "foo bar");
    },
    "adds a missing class as a function": function(div) {
      div.data([0, 1]).attr("class", null);
      div.classed("foo", function(d, i) { return d === 0; });
      assert.equal(div[0][0].className, "foo");
      assert.equal(div[0][1].className, "");
      div.classed("bar", function(d, i) { return i === 1; });
      assert.equal(div[0][0].className, "foo");
      assert.equal(div[0][1].className, "bar");
    },
    "removes an existing class as false": function(div) {
      div.attr("class", "bar foo");
      div.classed("foo", false);
      assert.equal(div[0][0].className, "bar");
      assert.equal(div[0][1].className, "bar");
      div.classed("bar", false);
      assert.equal(div[0][0].className, "");
      assert.equal(div[0][1].className, "");
    },
    "removes an existing class as a function": function(div) {
      div.data([0, 1]).attr("class", "bar foo");
      div.classed("foo", function(d, i) { return d === 0; });
      assert.equal(div[0][0].className, "bar foo");
      assert.equal(div[0][1].className, "bar");
      div.classed("bar", function(d, i) { return i === 1; });
      assert.equal(div[0][0].className, "foo");
      assert.equal(div[0][1].className, "bar");
      div.classed("foo", function() { return false; });
      assert.equal(div[0][0].className, "");
      assert.equal(div[0][1].className, "bar");
      div.classed("bar", function() { return false; });
      assert.equal(div[0][0].className, "");
      assert.equal(div[0][1].className, "");
    },
    "preserves an existing class as true": function(div) {
      div.attr("class", "bar foo");
      div.classed("foo", true);
      assert.equal(div[0][0].className, "bar foo");
      assert.equal(div[0][1].className, "bar foo");
      div.classed("bar", true);
      assert.equal(div[0][0].className, "bar foo");
      assert.equal(div[0][1].className, "bar foo");
    },
    "preserves an existing class as a function": function(div) {
      div.attr("class", "bar foo");
      div.classed("foo", function() { return true; });
      assert.equal(div[0][0].className, "bar foo");
      assert.equal(div[0][1].className, "bar foo");
      div.classed("bar", function() { return true; });
      assert.equal(div[0][0].className, "bar foo");
      assert.equal(div[0][1].className, "bar foo");
    },
    "preserves a missing class as false": function(div) {
      div.attr("class", "baz");
      div.classed("foo", false);
      assert.equal(div[0][0].className, "baz");
      assert.equal(div[0][1].className, "baz");
      div.attr("class", null);
      div.classed("bar", false);
      assert.equal(div[0][0].className, "");
      assert.equal(div[0][1].className, "");
    },
    "preserves a missing class as a function": function(div) {
      div.attr("class", "baz");
      div.classed("foo", function() { return false; });
      assert.equal(div[0][0].className, "baz");
      assert.equal(div[0][1].className, "baz");
      div.attr("class", null);
      div.classed("bar", function() { return false; });
      assert.equal(div[0][0].className, "");
      assert.equal(div[0][1].className, "");
    },
    "gets an existing class": function(div) {
      div[0][0].className = " foo\tbar  baz";
      assert.isTrue(div.classed("foo"));
      assert.isTrue(div.classed("bar"));
      assert.isTrue(div.classed("baz"));
    },
    "does not get a missing class": function(div) {
      div[0][0].className = " foo\tbar  baz";
      assert.isFalse(div.classed("foob"));
      assert.isFalse(div.classed("bare"));
      assert.isFalse(div.classed("rbaz"));
    },
    "ignores null nodes": function(div) {
      var some = d3.selectAll("div");
      some[0][1] = null;
      some.attr("class", null).classed("foo", true);
      assert.equal(div[0][0].className, "foo");
      assert.equal(div[0][1].className, "");
    },
    "returns the current selection": function(div) {
      assert.isTrue(div.classed("foo", true) === div);
    }
  }
});

suite.export(module);