added mockito tests but not working need consulting