Revise tests for base puppet class