1 files changed, 31 insertions, 0 deletions
diff --git a/go/game/ai_test.go b/go/game/ai_test.go
index 08cef2cb..632c6991 100644
--- a/go/game/ai_test.go
+++ b/go/game/ai_test.go
@@ -168,3 +168,34 @@ symbols:
 		t.Fatal("Target is not a Tile")
 	}
 }
+
+func TestSuggestUnitAi(t *testing.T) {
+	mapDef := `map: |1-
+ HST
+ HSF
+ TST
+symbols:
+  T: tower
+  H: house
+  F: farm
+  S: street
+`
+	r := strings.NewReader(mapDef)
+	m, _ := readMap(r)
+	a := NewUnit(NewCard("base/archer"), m.TileAt(Position{0, 0}), nil)
+	if SuggestUnitAI(a) != "aggressive" {
+		t.Fatal("expected aggressive")
+	}
+
+	f := NewUnit(NewCard("misc/farmer"), m.TileAt(Position{2, 2}), nil)
+	exp := "target-oriented farm tile"
+	is := SuggestUnitAI(f)
+	if exp != is {
+		t.Fatal(exp, " != ", is)
+	}
+
+	tc := NewUnit(NewCard("base/tax_collector"), m.TileAt(Position{3, 3}), nil)
+	if SuggestUnitAI(tc) != "shy" {
+		t.Fatal("expected shy")
+	}
+}