Add unit testing of mock.