Re-added the schema unit tests unintentionall removed.

Jira issue: WP-803