adding mockito tests