move join inference cleverness into a role

[dbsrgits/DBIx-Class.git] / t / dq / join.t

diff --git a/t/dq/join.t b/t/dq/join.t
index 099e672..125a0a0 100644 (file)
--- a/t/dq/join.t
+++ b/t/dq/join.t
@@ -32,13 +32,16 @@ is($mccrae->cds2->count, 3, 'CDs returned from expr join');
 
 is($mccrae->cds2_pre2k->count, 2, 'CDs returned from expr w/cond');
 
+$schema->source($_)->resultset_class('DBIx::Class::ResultSet::WithDQMethods')
+  for qw(CD Tag);
+
 my $cds = $schema->resultset('CD')
-                 ->search(expr { $_->artist->name eq 'Caterwauler McCrae' });
+                 ->where(expr { $_->artist->name eq 'Caterwauler McCrae' });
 
 is($cds->count, 3, 'CDs via join injection');
 
 my $tags = $schema->resultset('Tag')
-                  ->search(expr { $_->cd->artist->name eq 'Caterwauler McCrae' });
+                  ->where(expr { $_->cd->artist->name eq 'Caterwauler McCrae' });
 
 is($tags->count, 5, 'Tags via two step join injection');