Fixing load using value type for sharded sessions

nberardi · Dec 11, 2011 · 1f9ecae · 1f9ecae
1 parent 53614e0
commit 1f9ecae
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 1 deletion.
diff --git a/Raven.Client.Lightweight/Shard/ShardedDocumentSession.cs b/Raven.Client.Lightweight/Shard/ShardedDocumentSession.cs
@@ -314,7 +314,8 @@ public T[] Load<T>(IEnumerable<string> ids)
 		/// </remarks>
 		public T Load<T>(ValueType id)
 		{
-			var documentKey = Conventions.FindFullDocumentKeyFromNonStringIdentifier(id, typeof(T), false);
+			// assuming all conventions are the same
+			var documentKey = shardSessions.First().Advanced.DocumentStore.Conventions.FindFullDocumentKeyFromNonStringIdentifier(id, typeof(T), false);
 			return Load<T>(documentKey);
 		}
 

diff --git a/Raven.Tests/Shard/When_Using_Sharded_Servers.cs b/Raven.Tests/Shard/When_Using_Sharded_Servers.cs
@@ -97,6 +97,21 @@ public void Can_override_the_shard_id_generation()
 			}
 		}
 
+		[Fact]
+		public void Can_query_using_int()
+		{
+			shardStrategy.Stub(x => x.ShardAccessStrategy).Return(new SequentialShardAccessStrategy());
+			using (var documentStore = new ShardedDocumentStore(shardStrategy, shards))
+			{
+				documentStore.Initialize();
+
+				using (var session = documentStore.OpenSession())
+				{
+					session.Load<Company>(1);
+				}
+			}
+		}
+
 		[Fact]
 		public void Can_insert_into_two_sharded_servers()
 		{