Add repository tests

Tests for custom queries