Unit test setup