add Farm test, maximize coverage

presbrey · Aug 25, 2024 · 92203d2 · 92203d2
1 parent c624f79
commit 92203d2
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/farm_test.go b/farm_test.go
@@ -195,4 +195,11 @@ func TestFarmMethods(t *testing.T) {
 			t.Errorf("Expected 2 llamas, got %d", len(llamas))
 		}
 	})
+
+	t.Run("TestOllamaFarm", func(t *testing.T) {
+		llama := farm.First(&ollamafarm.Where{Group: "group1"})
+		if llama.Farm() != farm {
+			t.Error("Expected farm, got nil")
+		}
+	})
 }