Move entity declarations to files and add unit test

Moves a bit of code from one place to another (using new mixin-style) and ensures it works with... 🥁 🥁 🥁

The very first unit test for this extension! 🎉 🎉