@@ -264,7 +264,7 @@ describe('Directive: pfToolbar', function () {
264264 } ) ;
265265
266266 it ( 'should show the correct view selection buttons' , function ( ) {
267- var selectors = element . find ( '.view-selector' ) ;
267+ var selectors = element . find ( '.toolbar-pf- view-selector .btn-link ' ) ;
268268 expect ( selectors . length ) . toBe ( 5 ) ;
269269
270270 expect ( element . find ( '.fa-dashboard' ) . length ) . toBe ( 1 ) ;
@@ -291,22 +291,21 @@ describe('Directive: pfToolbar', function () {
291291 it ( 'should update the currently selected view when a view selector clicked' , function ( ) {
292292 var viewSelector = element . find ( '.toolbar-pf-view-selector' ) ;
293293 var active = element . find ( '.active' ) ;
294- var listSelector = element . find ( '.fa-th-list ' ) ;
294+ var listSelector = element . find ( '.toolbar-pf-view-selector .btn-link ' ) ;
295295
296296 expect ( viewSelector . length ) . toBe ( 1 ) ;
297297 expect ( active . length ) . toBe ( 0 ) ;
298- expect ( listSelector . length ) . toBe ( 1 ) ;
298+ expect ( listSelector . length ) . toBe ( 5 ) ;
299299
300300 eventFire ( listSelector [ 0 ] , 'click' ) ;
301301 $scope . $apply ( ) ;
302302
303- listSelector = element . find ( '.fa-th-list' ) ;
304303 active = element . find ( '.active' ) ;
305304 expect ( active . length ) . toBe ( 1 ) ;
306305 } ) ;
307306
308307 it ( 'should call the callback function when a view selector clicked' , function ( ) {
309- var listSelector = element . find ( '.fa-th-list ' ) ;
308+ var listSelector = element . find ( '.toolbar-pf-view-selector .btn-link ' ) ;
310309 var functionCalled = false ;
311310
312311 var onViewSelect = function ( ) {
@@ -315,7 +314,7 @@ describe('Directive: pfToolbar', function () {
315314
316315 $scope . config . viewsConfig . onViewSelect = onViewSelect ;
317316 expect ( functionCalled ) . toBeFalsy ( ) ;
318- expect ( listSelector . length ) . toBe ( 1 ) ;
317+ expect ( listSelector . length ) . toBe ( 5 ) ;
319318
320319 eventFire ( listSelector [ 0 ] , 'click' ) ;
321320 $scope . $apply ( ) ;
0 commit comments