Update tests to assume we're using an (almost) representative set
of test data. Combined tests into one class to reduce index load time.