Can I use the Sonoff Basic R4 with my Water Heater?

No, the water heater usually has a lot higher start Amp, so you should connect it to a minimum of 20 AMP. And these Relays have no power protection.

Better use:

Sonoff TH Elite Temperature and Humidity Monitoring Switch 20A.