Merge branch 'phase/42-talk-circles'
Co-authored-by: Cursor <cursoragent@cursor.com> # Conflicts: # docs/phases/README.md # src/HSchool.Simulation/PersonLog.cs
This commit is contained in:
Leonid Pershin
@@ -0,0 +1,276 @@
|
||||
using Arch.Core;
|
||||
using HSchool.Ai;
|
||||
using HSchool.Content;
|
||||
using HSchool.People;
|
||||
using HSchool.Simulation;
|
||||
|
||||
namespace HSchool.Simulation.Tests;
|
||||
|
||||
public class TalkCircleTests
|
||||
{
|
||||
private static readonly DateTime TuesdayMorning = new(2012, 4, 3, 6, 0, 0, DateTimeKind.Utc);
|
||||
|
||||
[Fact]
|
||||
public void FriendsOnBreakInCorridor_FormTalkCircle()
|
||||
{
|
||||
var (school, first, second) = TwoPupilsOnBreak();
|
||||
using (school)
|
||||
{
|
||||
OpinionStore.Set(first, second.Id, 55);
|
||||
OpinionStore.Set(second, first.Id, 55);
|
||||
PlaceAt(school, first.Id, "corridor-1");
|
||||
PlaceAt(school, second.Id, "corridor-1");
|
||||
|
||||
Assert.True(school.TryStartAction(first.Id, TalkActions.Chat));
|
||||
|
||||
var rows = school.CapturePresence();
|
||||
Assert.Equal(TalkActions.Chat, rows.Single(row => row.PersonId == first.Id).ActionId);
|
||||
Assert.Equal(TalkActions.Chat, rows.Single(row => row.PersonId == second.Id).ActionId);
|
||||
}
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public void EnemiesOnBreak_DoNotFormCircle()
|
||||
{
|
||||
var (school, first, second) = TwoPupilsOnBreak();
|
||||
using (school)
|
||||
{
|
||||
OpinionStore.Set(first, second.Id, -55);
|
||||
OpinionStore.Set(second, first.Id, -55);
|
||||
PlaceAt(school, first.Id, "corridor-1");
|
||||
PlaceAt(school, second.Id, "corridor-1");
|
||||
|
||||
Assert.False(school.TryStartAction(first.Id, TalkActions.Chat));
|
||||
Assert.Null(school.CapturePresence().Single(row => row.PersonId == first.Id).ActionId);
|
||||
}
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public void FifthPerson_DoesNotJoinFullCircle()
|
||||
{
|
||||
var (school, pupils) = FivePupilsOnBreak();
|
||||
using (school)
|
||||
{
|
||||
foreach (var pupil in pupils.Take(4))
|
||||
{
|
||||
PlaceAt(school, pupil.Id, "corridor-1");
|
||||
}
|
||||
|
||||
PlaceAt(school, pupils[0].Id, "corridor-1");
|
||||
Assert.True(school.TryStartAction(pupils[0].Id, TalkActions.Chat));
|
||||
PlaceAt(school, pupils[4].Id, "corridor-1");
|
||||
Assert.False(school.TryStartAction(pupils[4].Id, TalkActions.Chat));
|
||||
Assert.Null(school.CapturePresence().Single(row => row.PersonId == pupils[4].Id).ActionId);
|
||||
}
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public void StaffInTeachersRoom_CanStaffChat()
|
||||
{
|
||||
var (school, staff) = TwoStaffOnBreak();
|
||||
using (school)
|
||||
{
|
||||
PlaceAt(school, staff[0].Id, "teachers-room");
|
||||
PlaceAt(school, staff[1].Id, "teachers-room");
|
||||
|
||||
Assert.True(school.TryStartAction(staff[0].Id, TalkActions.StaffChat));
|
||||
Assert.Equal(TalkActions.StaffChat, ActivityOf(school, staff[0].Id));
|
||||
Assert.Equal(TalkActions.StaffChat, ActivityOf(school, staff[1].Id));
|
||||
}
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public void Pupil_CannotStaffChat()
|
||||
{
|
||||
var (school, first, _) = TwoPupilsOnBreak();
|
||||
using (school)
|
||||
{
|
||||
PlaceAt(school, first.Id, "teachers-room");
|
||||
Assert.False(school.TryStartAction(first.Id, TalkActions.StaffChat));
|
||||
}
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public void TwoWithPhoneOnBreak_CanPhoneChat()
|
||||
{
|
||||
var (school, first, second) = TwoPupilsWithPhonesOnBreak();
|
||||
using (school)
|
||||
{
|
||||
PlaceAt(school, first.Id, "corridor-1");
|
||||
PlaceAt(school, second.Id, "corridor-1");
|
||||
|
||||
Assert.True(school.TryStartAction(first.Id, TalkActions.PhoneChat));
|
||||
Assert.Equal(TalkActions.PhoneChat, ActivityOf(school, first.Id));
|
||||
Assert.Equal(TalkActions.PhoneChat, ActivityOf(school, second.Id));
|
||||
}
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public void WithoutPhone_CannotPhoneChat()
|
||||
{
|
||||
var (school, first, second) = TwoPupilsOnBreak();
|
||||
using (school)
|
||||
{
|
||||
Assert.False(TalkCircles.HasPhone(first.Items));
|
||||
PlaceAt(school, first.Id, "corridor-1");
|
||||
PlaceAt(school, second.Id, "corridor-1");
|
||||
|
||||
Assert.False(school.TryStartAction(first.Id, TalkActions.PhoneChat));
|
||||
}
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public void TalkCircle_WritesTopicToLogAndShiftsOpinion()
|
||||
{
|
||||
var (school, first, second) = TwoPupilsOnBreak();
|
||||
using (school)
|
||||
{
|
||||
OpinionStore.Set(first, second.Id, 10);
|
||||
PlaceAt(school, first.Id, "corridor-1");
|
||||
PlaceAt(school, second.Id, "corridor-1");
|
||||
Assert.True(school.TryStartAction(first.Id, TalkActions.Chat));
|
||||
|
||||
while (ActivityOf(school, first.Id) is not null)
|
||||
{
|
||||
school.Tick(0.2d, 5d);
|
||||
}
|
||||
|
||||
Assert.NotEqual(10, OpinionStore.Get(first, second.Id));
|
||||
Assert.Contains(
|
||||
school.DayLog,
|
||||
row => row.PersonId == first.Id
|
||||
&& row.Type.Equals(PersonLogTypes.TalkEnded, StringComparison.Ordinal)
|
||||
&& row.ThingDef is not null);
|
||||
}
|
||||
}
|
||||
|
||||
private static (School School, Person First, Person Second) TwoPupilsWithPhonesOnBreak()
|
||||
{
|
||||
var (catalog, map) = Vanilla();
|
||||
var school = OpenSchool(catalog, map, seed: 45);
|
||||
var withPhone = school.Roster!.People
|
||||
.Where(person => person.IsStudent && TalkCircles.HasPhone(person.Items))
|
||||
.Take(2)
|
||||
.ToArray();
|
||||
Assert.Equal(2, withPhone.Length);
|
||||
return (school, withPhone[0], withPhone[1]);
|
||||
}
|
||||
|
||||
private static (School School, Person First, Person Second) TwoPupilsOnBreak()
|
||||
{
|
||||
var (catalog, map) = Vanilla();
|
||||
var school = OpenSchool(catalog, map, seed: 42);
|
||||
var schoolClass = school.Roster!.Classes.First(row => row.RoomId == "classroom-101");
|
||||
var pupils = schoolClass.PupilIds
|
||||
.Select(id => school.Roster.People.First(person => person.Id == id))
|
||||
.Take(2)
|
||||
.ToArray();
|
||||
return (school, pupils[0], pupils[1]);
|
||||
}
|
||||
|
||||
private static (School School, IReadOnlyList<Person> Pupils) FivePupilsOnBreak()
|
||||
{
|
||||
var (catalog, map) = Vanilla();
|
||||
var school = OpenSchool(catalog, map, seed: 43);
|
||||
var schoolClass = school.Roster!.Classes.First(row => row.RoomId == "classroom-101");
|
||||
var pupils = schoolClass.PupilIds
|
||||
.Select(id => school.Roster.People.First(person => person.Id == id))
|
||||
.Take(5)
|
||||
.ToArray();
|
||||
return (school, pupils);
|
||||
}
|
||||
|
||||
private static (School School, Person[] Staff) TwoStaffOnBreak()
|
||||
{
|
||||
const float cap = 100_000f;
|
||||
var (catalog, map) = Vanilla();
|
||||
var school = OpenSchool(catalog, map, seed: 44);
|
||||
var roster = school.Roster!;
|
||||
var pool = school.Applicants!;
|
||||
for (var i = 0; i < 2; i++)
|
||||
{
|
||||
var hired = Staffing.Hire(
|
||||
catalog,
|
||||
map,
|
||||
roster,
|
||||
pool,
|
||||
pool.Applicants[0].Person.Id,
|
||||
Staffing.TeacherPosition,
|
||||
cap);
|
||||
Assert.Equal(StaffingError.None, hired.Error);
|
||||
roster = hired.Roster;
|
||||
pool = hired.Pool;
|
||||
}
|
||||
|
||||
school.ApplyStaffing(roster, pool);
|
||||
var staff = school.Roster!.People.Where(person => person.IsStaff).Take(2).ToArray();
|
||||
Assert.Equal(2, staff.Length);
|
||||
return (school, staff);
|
||||
}
|
||||
|
||||
private static School OpenSchool(DefCatalog catalog, MapLayout map, int seed)
|
||||
{
|
||||
var roster = RosterGenerator.Generate(catalog, map, seed, "Russia", TuesdayMorning);
|
||||
var pool = ApplicantPool.Create(catalog, roster, seed, "Russia", TuesdayMorning);
|
||||
var schoolClass = roster.Classes.First(row => row.RoomId == "classroom-101");
|
||||
var school = School.Create(seed, "Talk", TuesdayMorning, catalog, map);
|
||||
school.InstallPeople(roster, seed, "Russia", pool);
|
||||
school.SetTimetable(new HSchool.Schedule.Timetable(
|
||||
[
|
||||
new HSchool.Schedule.LessonPlacement(schoolClass.Id, "Mathematics", "t1", schoolClass.RoomId, Day: 1, Period: 1),
|
||||
],
|
||||
[]));
|
||||
school.ConfigurePresence(weekDays: 5, maxDecisionsPerTick: 10_000);
|
||||
AdvanceTo(school, new DateTime(2012, 4, 3, 9, 18, 0, DateTimeKind.Utc));
|
||||
return school;
|
||||
}
|
||||
|
||||
private static void AdvanceTo(School school, DateTime until)
|
||||
{
|
||||
while (school.Clock.Time < until)
|
||||
{
|
||||
school.Tick(0.2d, 5d);
|
||||
}
|
||||
}
|
||||
|
||||
private static void PlaceAt(School school, string personId, string nodeId)
|
||||
{
|
||||
var query = new QueryDescription().WithAll<PersonIdentity, Presence, PersonActivity, Intent>();
|
||||
school.World.Query(in query, (ref PersonIdentity identity, ref Presence presence, ref PersonActivity activity, ref Intent intent) =>
|
||||
{
|
||||
if (identity.Id.Equals(personId, StringComparison.Ordinal))
|
||||
{
|
||||
presence = new Presence(nodeId, 0f, nodeId, false, []);
|
||||
activity = PersonActivity.Idle;
|
||||
intent = Intent.None;
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
private static string? ActivityOf(School school, string personId) =>
|
||||
school.CapturePresence().Single(row => row.PersonId == personId).ActionId;
|
||||
|
||||
private static (DefCatalog Catalog, MapLayout Map) Vanilla()
|
||||
{
|
||||
var root = Path.Combine(AppContext.BaseDirectory, "vanilla");
|
||||
var documents = new List<ContentDocument>();
|
||||
foreach (var path in Directory.EnumerateFiles(root, "*.*", SearchOption.AllDirectories))
|
||||
{
|
||||
if (!path.EndsWith(".jsonc", StringComparison.OrdinalIgnoreCase)
|
||||
&& !path.EndsWith(".json", StringComparison.OrdinalIgnoreCase))
|
||||
{
|
||||
continue;
|
||||
}
|
||||
|
||||
documents.Add(new ContentDocument(
|
||||
CatalogLoader.CorePackId,
|
||||
Path.GetRelativePath(root, path).Replace('\\', '/'),
|
||||
File.ReadAllText(path)));
|
||||
}
|
||||
|
||||
var catalog = new CatalogLoader().Load([CatalogLoader.CorePackId], documents);
|
||||
var map = CatalogLoader.LastDefaultMap([CatalogLoader.CorePackId], documents);
|
||||
Assert.NotNull(map);
|
||||
return (catalog, map);
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user