Using vanilla mob animation on new mob

I'm working on a custom fish mob that I want to animate like the vanilla salmon, but I just can't figure out how to do it. My mob is modeled extremely similarly to the vanilla salmon (same number of cubes, same layout, mostly just different sized cubes). How would I go about doing this?