Add a button test in context