[Table] Increase test coverage. (mui#8910)

the-noob · Nov 17, 2017 · 12bc480 · 12bc480
1 parent 2f96f6c
commit 12bc480
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 0 deletions.
diff --git a/src/Table/Table.spec.js b/src/Table/Table.spec.js
@@ -44,4 +44,9 @@ describe('<Table />', () => {
     const wrapper = shallow(<Table>{children}</Table>);
     assert.strictEqual(wrapper.childAt(0).equals(children), true);
   });
+
+  it('should define table in the child context', () => {
+    const wrapper = shallow(<Table />);
+    assert.deepStrictEqual(wrapper.instance().getChildContext().table, {});
+  });
 });
diff --git a/src/Table/TableBody.spec.js b/src/Table/TableBody.spec.js
@@ -35,4 +35,9 @@ describe('<TableBody />', () => {
     const wrapper = shallow(<TableBody>{children}</TableBody>);
     assert.strictEqual(wrapper.childAt(0).equals(children), true);
   });
+
+  it('should define table.body in the child context', () => {
+    const wrapper = shallow(<TableBody />);
+    assert.strictEqual(wrapper.instance().getChildContext().table.body, true);
+  });
 });
diff --git a/src/Table/TableHead.spec.js b/src/Table/TableHead.spec.js
@@ -35,4 +35,9 @@ describe('<TableHead />', () => {
     const wrapper = shallow(<TableHead>{children}</TableHead>);
     assert.strictEqual(wrapper.childAt(0).equals(children), true);
   });
+
+  it('should define table.head in the child context', () => {
+    const wrapper = shallow(<TableHead />);
+    assert.strictEqual(wrapper.instance().getChildContext().table.head, true);
+  });
 });