Move Di test to integration test as it requires DB connection