Apenas atente-se ao texto para não cometer equivocos, o uso do whatsapp não justifica uma justa causa, apenas se estiver interferindo em seu rendimento, o que pode ser punitivo com advertências ou no caso de algum acidente até uma suspensão porem algo alem disso acho que não seria bem visto nos tribunais.
Hoje não tem como proibir o uso, até porque o custo da internet movél esta "barato" e qualquer um pode te-la, o que deve ser observado é se a queda do rendimento e se isso esta realmente ligado ao aplicativo. O que deve haver é uma conversa e uma imposição aos costumes da empresa.
Exemplo real sobre os 5 minutos de tolerancia para horas extras, meu chefe assim que dava o horário falava "Fabio já desligou o computador, já esta indo embora" e isso enchia o saco mais se eu não quisesse mais ouvir eu tinha que começar a sair no horário.
Ele fez isso com toda a turma de DP durante 1 semana, na semana seguinte ele não precisou mais fazer pois criamos o habito de sair no horário.
Enfim, é apenas uma opinião tudo que disse acima.