Add tests for the new container xblock API