Use mock Count to check size
Merge pull request #3 from iporsut/master

Use mock Count to check size