diff --git a/test/main.spec.js b/test/main.spec.js index 60db2880..2f25d570 100755 --- a/test/main.spec.js +++ b/test/main.spec.js @@ -240,14 +240,16 @@ describe('CodeBreaker :', () => { document.getElementById('user-guess').value = '1234'; window.guess(); assert(document.getElementById('guessing-div').style.display == "none", '`showReplay` was not run when game win condition was met.'); - assert(document.getElementById('message').innerHTML == 'You Win! :)','`showAnswer` was not run when game win condition was met.'); - //lose outcame + assert(document.getElementById('message').innerHTML == 'You Win! :)','`setMessage` was not run when game win condition was met.'); + assert(document.getElementById('code').className.includes(' success'),'`showAnswer` was not run when game win condition was met.'); + //lose outcome document.getElementById('answer').value = '1234'; document.getElementById('attempt').value = '10'; document.getElementById('user-guess').value = '4321'; window.guess(); assert(document.getElementById('guessing-div').style.display == "none", '`showReplay` was not run when game lose condition was met.'); - assert(document.getElementById('message').innerHTML == 'You Lose! :(','`showAnswer` was not run when game lose condition was met.'); + assert(document.getElementById('message').innerHTML == 'You Lose! :(','`setMessage` was not run when game lose condition was met.'); + assert(document.getElementById('code').className.includes(' failure'),'`showAnswer` was not run when game lose condition was met.'); }); }); });