My boss locked a Coworker into the toilet for not cleaning it! Should I go?

Yesterday my Boss locked my coworker into the toilet for not cleaning it.

We are both in charge for cleaning them but because I am brand new to the Job I didn’t know that I have to do that. My Coworker was on a construction site all afternoon and I was in the house of the company just cleaning up stuff. When my Coworker got back he wanted to go Home and enjoy his Weekend but my boss had other plans for him. He asked him if he had cleaned the toilet yet. Of course my Coworker said No because he was on the Construction site all day and had no time to do it. Then my boss screamed at him to now clean the toilet. My Coworker then went on to clean the toilett and I helped him and cleaned the floor around the toilet after some time we were done and wanted to go home. Our boss then went to check if the toilet really was cleaned but acording to his standards it wasn’t. He then proceeded to scream at ONLY my coworker, not me, and put a piece of wood under the doorhandle of the toilet were my coworker still was inside. He then told him he would check on him in two hours and he should get the toilet cleaned by then. I then went home because I was so scared I was next. Should I go?

Sorry for the bad grammar!