building proper unit tests