Add mock dependency
More tests