1 require("../env");

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

     3 

     4 var vows = require("vows"),

     5     assert = require("assert");

     6 

     7 var suite = vows.describe("selection.select");

     8 

     9 suite.addBatch({

    10   "select(body)": {

    11     topic: function() {

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

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

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

    15       return body;

    16     },

    17     "selects the first matching element": function(body) {

    18       var div = body.select("div");

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

    20       assert.equal(div.length, 1);

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

    22       assert.equal(div.attr("class"), "first");

    23     },

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

    25       var div = body.select("div");

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

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

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

    29     },

    30     "propagates data to the selected elements": function(body) {

    31       var data = new Object(), div = body.data([data]).select("div");

    32       assert.strictEqual(div[0][0].__data__, data);

    33     },

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

    35       delete document.body.__data__;

    36       var data = new Object(), div = body.select("div").data([data]);

    37       div = body.select("div");

    38       assert.strictEqual(div[0][0].__data__, data);

    39       assert.isUndefined(document.body.__data__);

    40     },

    41     "returns null if no match is found": function(body) {

    42       var span = body.select("span");

    43       assert.equal(span[0][0], null);

    44       assert.equal(span.length, 1);

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

    46     },

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

    48       body.append("foo");

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

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

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

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

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

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

    55       delete document.body.__data__;

    56     }

    57   }

    58 });

    59 

    60 suite.addBatch({

    61   "selectAll(div)": {

    62     topic: function() {

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

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

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

    66       return div;

    67     },

    68     "selects the first matching element": function(div) {

    69       var span = div.select("span");

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

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

    72       assert.equal(span.length, 1);

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

    74       assert.equal(span.attr("class"), "first");

    75     },

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

    77       var span = div.select("span");

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

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

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

    81     },

    82     "propagates data to the selected elements": function(div) {

    83       var data = new Object(), span = div.data([data]).select("span");

    84       assert.strictEqual(span[0][0].__data__, data);

    85     },

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

    87       delete div[0][0].__data__;

    88       delete div[0][1].__data__;

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

    90       span = div.select("span");

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

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

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

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

    95     },

    96     "returns null if no match is found": function(div) {

    97       var div = div.select("div");

    98       assert.equal(div[0][0], null);

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

   100       assert.equal(div.length, 1);

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

   102     },

   103     "ignores null nodes": function(div) {

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

   105       some[0][1] = null;

   106       var span = some.select("span");

   107       assert.equal(span.length, 1);

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

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

   110       assert.isNull(span[0][1]);

   111     },

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

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

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

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

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

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

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

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

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

   121     }

   122   }

   123 });

   124 

   125 suite.export(module);