Rework template testing so that we can assert against the context