Mocking out enum as well