Added tests for LoadSingleById

Author: jklemmack

src/ServiceStack.OrmLite/OrmLiteReadCommandExtensions.cs

@@ -884,7 +884,7 @@ internal static T LoadSingleById<T>(this IDbCommand dbCmd, object value, string[
             if (row == null)
                 return default(T);
 
-            dbCmd.LoadReferences(row);
+            dbCmd.LoadReferences(row, include);
 
             return row;
         }
@@ -905,7 +905,7 @@ public static void LoadReferences<T>(this IDbCommand dbCmd, T instance, string[]
                 fieldDefs = fieldDefs.Where(fd => include.Contains(fd.FieldName)).ToList();
             }
 
-            foreach (var fieldDef in loadRef.FieldDefs)
+            foreach (var fieldDef in fieldDefs)
             {
                 dbCmd.Parameters.Clear();
                 var listInterface = fieldDef.FieldType.GetTypeWithGenericInterfaceOf(typeof(IList<>));

tests/ServiceStack.OrmLite.Tests/LoadReferencesTests.cs

@@ -656,6 +656,13 @@ public void Can_load_only_included_references()
             Assert.That(dbCustomers[0].Orders, Is.Null);
             Assert.That(dbCustomers[0].PrimaryAddress, Is.Not.Null);
 
+            // Test LoadSingleById
+            var dbCustomer = db.LoadSingleById<Customer>(customer.Id, include: new[] { "PrimaryAddress" });
+            Assert.That(dbCustomer.Name, Is.EqualTo("Customer 1"));
+
+            Assert.That(dbCustomer.Orders, Is.Null);
+            Assert.That(dbCustomer.PrimaryAddress, Is.Not.Null);
+
 
             // Invalid field name
             try
@@ -670,6 +677,20 @@ public void Can_load_only_included_references()
             {
                 Assert.Fail();
             }
+
+
+            try
+            {
+                dbCustomer = db.LoadSingleById<Customer>(customer.Id, include: new[] { "InvalidOption1", "InvalidOption2" });
+                Assert.Fail();
+            }
+            catch (System.ArgumentException ex)
+            {
+            }
+            catch (System.Exception ex)
+            {
+                Assert.Fail();
+            }
         }
 
         [Test]