<?xml version="1.0" encoding="UTF-8"?>
<commit>
  <added type="array"/>
  <modified type="array">
    <modified>
      <diff>@@ -10,7 +10,7 @@ class AccountFigmentsTest &lt; ActionController::IntegrationTest
     assert_equal 0, conn0.count_for('figments')
     assert_equal 0, conn1.count_for('figments')
 
-    new_session do |user|
+    new_session(0) do |user|
       user.goes_home
       mike = user.creates_account('mike', '0')
       user.selects_account(mike)
@@ -25,7 +25,7 @@ class AccountFigmentsTest &lt; ActionController::IntegrationTest
     assert_equal 1, conn0.count_for('figments')
     assert_equal 0, conn1.count_for('figments')
 
-    new_session do |user|
+    new_session(1) do |user|
       user.goes_home
       bob = user.creates_account('bob', '1')
       user.selects_account(bob)
@@ -54,10 +54,12 @@ class AccountFigmentsTest &lt; ActionController::IntegrationTest
     conn
   end
 
-  def new_session
+  def new_session(shard)
     open_session do |sess|
-      sess.extend(Operations)
-      yield sess if block_given?
+      DataFabric.activate_shard :shard =&gt; shard do
+        sess.extend(Operations)
+        yield sess if block_given?
+      end
     end
   end
 </diff>
      <filename>example/test/integration/account_figments_test.rb</filename>
    </modified>
  </modified>
  <removed type="array">
    <removed>
      <filename>TODO</filename>
    </removed>
  </removed>
  <parents type="array">
    <parent>
      <id>1d1957dbe624ac1add1cb3862e859af7ed4de12e</id>
    </parent>
  </parents>
  <author>
    <name>Mike Perham</name>
    <email>mperham@gmail.com</email>
  </author>
  <url>http://github.com/fiveruns/data_fabric/commit/b6e2e9a721bf30e3351711ff0d4a1e5c138b563a</url>
  <id>b6e2e9a721bf30e3351711ff0d4a1e5c138b563a</id>
  <committed-date>2008-09-02T04:50:36-07:00</committed-date>
  <authored-date>2008-09-02T04:50:36-07:00</authored-date>
  <message>Remove TODO
Looks like the integration tests broke under Rails 2.1.  Fixed.</message>
  <tree>2b6fa6fced16aece996a5e39aa541f6c3c2206a2</tree>
  <committer>
    <name>Mike Perham</name>
    <email>mperham@gmail.com</email>
  </committer>
</commit>
