Make SetTransactionRetryCount actually work.

impl/memory/datastore.go

Index: impl/memory/datastore.go
diff --git a/impl/memory/datastore.go b/impl/memory/datastore.go
index d622e8199a0bf32f41937eb4f81a391e9037746a..d02cfd61a27ae62f41ff9a8a8cff82aea27167cb 100644
--- a/impl/memory/datastore.go
+++ b/impl/memory/datastore.go
@@ -24,7 +24,7 @@ func useRDS(c context.Context) context.Context {
 
 		ns := curGID(ic).namespace
 		if x, ok := dsd.(*dataStoreData); ok {
-			return &dsImpl{x, ns, 0, ic}
+			return &dsImpl{x, ns, ic}
 		}
 		return &txnDsImpl{dsd.(*txnDataStoreData), ns}
 	})
@@ -34,10 +34,9 @@ func useRDS(c context.Context) context.Context {
 
 // dsImpl exists solely to bind the current c to the datastore data.
 type dsImpl struct {
-	data         *dataStoreData
-	ns           string
-	txnFakeRetry int
-	c            context.Context
+	data *dataStoreData
+	ns   string
+	c    context.Context
 }
 
 var _ ds.RawInterface = (*dsImpl)(nil)
@@ -109,7 +108,7 @@ func (d *dsImpl) CatchupIndexes() {
 }
 
 func (d *dsImpl) SetTransactionRetryCount(count int) {
-	d.txnFakeRetry = count
+	d.data.txnFakeRetry = count
 }
 
 func (d *dsImpl) Testable() ds.Testable {