@@ -226,10 +226,8 @@ describe('<TreeNode />', () => {
226226 < TreeNode { ...baseProps } icons = { { uncheck : < span className = "other-uncheck" /> } } /> ,
227227 ) ;
228228
229- assert . isTrue ( wrapper . contains (
230- < span className = "rct-checkbox" >
231- < span className = "other-uncheck" />
232- </ span > ,
229+ assert . isTrue ( wrapper . find ( '.rct-checkbox' ) . contains (
230+ < span className = "other-uncheck" /> ,
233231 ) ) ;
234232 } ) ;
235233 } ) ;
@@ -254,10 +252,8 @@ describe('<TreeNode />', () => {
254252 < TreeNode { ...baseProps } /> ,
255253 ) ;
256254
257- assert . isTrue ( wrapper . contains (
258- < span className = "rct-checkbox" >
259- < span className = "rct-icon rct-icon-uncheck" />
260- </ span > ,
255+ assert . isTrue ( wrapper . find ( '.rct-checkbox' ) . contains (
256+ < span className = "rct-icon rct-icon-uncheck" /> ,
261257 ) ) ;
262258 } ) ;
263259
@@ -392,6 +388,25 @@ describe('<TreeNode />', () => {
392388 assert . isTrue ( actual . checked ) ;
393389 } ) ;
394390
391+ it ( 'should trigger on key press' , ( ) => {
392+ let actual = { } ;
393+
394+ const wrapper = shallow (
395+ < TreeNode
396+ { ...baseProps }
397+ checked = { 2 }
398+ value = "jupiter"
399+ onCheck = { ( node ) => {
400+ actual = node ;
401+ } }
402+ /> ,
403+ ) ;
404+
405+ wrapper . find ( '.rct-checkbox' ) . simulate ( 'keyup' , { keyCode : 32 } ) ;
406+
407+ assert . isTrue ( actual . checked ) ;
408+ } ) ;
409+
395410 describe ( 'optimisticToggle' , ( ) => {
396411 it ( 'should toggle a partially-checked node to unchecked' , ( ) => {
397412 let actual = { } ;
0 commit comments