Shift entity unit tests

Signed-off-by: Max Fierke <max@maxfierke.com>