Refactor getPalette for DI
Refactor test for DI