Add tests for SchemaBuilder