i would go with yours it feels more unit test-y