When phone is in AOD and I get a notification the screen wakes up and goes to the lock screen instead of staying in AOD. I tried doing a screen recording but it stops recording when I hit the power button.
Okay AOD is working fine for me. Whatsapp notifications doesn't wake my phone up at all, just get the notification. Regular messages wakes my phone up to lock screen for a second but goes back to AOD.
I actually found out that an app that I was using for edge lighting was causing the issue. I uninstalled, and the issue went away. But now edge lighting will only work with messages. It doesn't work with outlook, email or Gmail anymore.