CHEF-4413 Add unit tests