Updated checkout button directive and tests