Reset mock environment in assign private ip tests