Rewrite existing integration test as a unit test

[eyurtsev]

Re-write existing integration test as a unit test, replace with mock response