mdb: package for interacting with game's local database

2026-06-02 12:39:58 -04:00
parent 96503e40f6
commit f40fea0ec5
28 changed files with 2906 additions and 3 deletions
--- a/mdb/character_test.go
+++ b/mdb/character_test.go
@@ -0,0 +1,224 @@
+package mdb_test
+
+import (
+	_ "embed"
+	"testing"
+
+	"git.sunturtle.xyz/zephyr/horse"
+	"git.sunturtle.xyz/zephyr/horse/mdb"
+	"github.com/google/go-cmp/cmp"
+)
+
+//go:embed testdata/character.sql
+var characterSQL string
+
+func TestCharacters(t *testing.T) {
+	db := testdb(t.Context(), "file:TestCharacters?mode=memory&cache=shared", characterSQL)
+	got, err := mdb.Characters(t.Context(), db)
+	if err != nil {
+		t.Error(err)
+	}
+	want := []horse.Character{
+		{ID: 1001, Name: "Special Week"},
+		{ID: 1002, Name: "Silence Suzuka"},
+		{ID: 1078, Name: "Yamanin Zephyr"},
+		{ID: 2001, Name: "Happy Meek"},
+		{ID: 9001, Name: "Tazuna Hayakawa"},
+	}
+	if diff := cmp.Diff(want, got); diff != "" {
+		t.Errorf("wrong characters (+got/-want):\n%s", diff)
+	}
+}
+
+func TestAffinitySummary(t *testing.T) {
+	db := testdb(t.Context(), "file:TestAffinitySummary?mode=memory&cache=shared", characterSQL)
+	got, err := mdb.AffinitySummary(t.Context(), db)
+	if err != nil {
+		t.Error(err)
+	}
+	want := []horse.AffinityRelation{
+		{IDA: 1001, IDB: 1002, Affinity: 23},
+	}
+	if diff := cmp.Diff(want, got); diff != "" {
+		t.Errorf("wrong affinity (+got/-want):\n%s", diff)
+	}
+}
+
+func TestUmas(t *testing.T) {
+	db := testdb(t.Context(), "file:TestUmas?mode=memory&cache=shared", characterSQL)
+	got, err := mdb.Umas(t.Context(), db)
+	if err != nil {
+		t.Error(err)
+	}
+	want := []horse.Uma{
+		{
+			ID:          100101,
+			CharacterID: 1001,
+			Name:        `[Special Dreamer] Special Week`,
+			Variant:     `[Special Dreamer]`,
+			Sprint:      2,
+			Mile:        5,
+			Medium:      7,
+			Long:        7,
+			Front:       1,
+			Pace:        7,
+			Late:        7,
+			End:         5,
+			Turf:        7,
+			Dirt:        1,
+			Unique:      100011,
+			Skill1:      200512,
+			Skill2:      201352,
+			Skill3:      200732,
+			SkillPL2:    200162,
+			SkillPL3:    201351,
+			SkillPL4:    200612,
+			SkillPL5:    200511,
+		},
+		{
+			ID:          100102,
+			CharacterID: 1001,
+			Name:        `[Hopp'n♪Happy Heart] Special Week`,
+			Variant:     `[Hopp'n♪Happy Heart]`,
+			Sprint:      2,
+			Mile:        5,
+			Medium:      7,
+			Long:        7,
+			Front:       1,
+			Pace:        7,
+			Late:        7,
+			End:         5,
+			Turf:        7,
+			Dirt:        1,
+			Unique:      110011,
+			Skill1:      200462,
+			Skill2:      200592,
+			Skill3:      201132,
+			SkillPL2:    200212,
+			SkillPL3:    200591,
+			SkillPL4:    201611,
+			SkillPL5:    200461,
+		},
+		{
+			ID:          100201,
+			CharacterID: 1002,
+			Name:        `[Innocent Silence] Silence Suzuka`,
+			Variant:     `[Innocent Silence]`,
+			Sprint:      4,
+			Mile:        7,
+			Medium:      7,
+			Long:        3,
+			Front:       7,
+			Pace:        5,
+			Late:        3,
+			End:         1,
+			Turf:        7,
+			Dirt:        1,
+			Unique:      100021,
+			Skill1:      200432,
+			Skill2:      200552,
+			Skill3:      200712,
+			SkillPL2:    200022,
+			SkillPL3:    200431,
+			SkillPL4:    200542,
+			SkillPL5:    200551,
+		},
+	}
+	if diff := cmp.Diff(want, got); diff != "" {
+		t.Errorf("wrong umas (+got/-want):\n%s", diff)
+	}
+}
+
+func TestConversations(t *testing.T) {
+	db := testdb(t.Context(), "file:TestConversations?mode=memory&cache=shared", characterSQL)
+	got, err := mdb.Conversations(t.Context(), db)
+	if err != nil {
+		t.Error(err)
+	}
+	want := []horse.Conversation{
+		{
+			CharacterID: 1001,
+			Number:      1,
+			Location:    410,
+			Chara1:      1001,
+		},
+		{
+			CharacterID:   1001,
+			Number:        2,
+			Location:      510,
+			Chara1:        1001,
+			ConditionType: 1,
+		},
+		{
+			CharacterID:   1001,
+			Number:        3,
+			Location:      310,
+			Chara1:        1001,
+			ConditionType: 1,
+		},
+		{
+			CharacterID:   1001,
+			Number:        4,
+			Location:      120,
+			Chara1:        1001,
+			Chara2:        1002,
+			ConditionType: 2,
+		},
+		{
+			CharacterID:   1001,
+			Number:        5,
+			Location:      520,
+			Chara1:        1003,
+			Chara2:        1001,
+			ConditionType: 3,
+		},
+		{
+			CharacterID:   1001,
+			Number:        6,
+			Location:      430,
+			Chara1:        1001,
+			Chara2:        1014,
+			Chara3:        1011,
+			ConditionType: 1,
+		},
+		{
+			CharacterID: 1002,
+			Number:      1,
+			Location:    310,
+			Chara1:      1002,
+		},
+		{
+			CharacterID:   1002,
+			Number:        2,
+			Location:      210,
+			Chara1:        1002,
+			ConditionType: 1,
+		},
+		{
+			CharacterID:   1002,
+			Number:        3,
+			Location:      110,
+			Chara1:        1002,
+			ConditionType: 1,
+		},
+		{
+			CharacterID:   1002,
+			Number:        4,
+			Location:      520,
+			Chara1:        1010,
+			Chara2:        1002,
+			ConditionType: 3,
+		},
+		{
+			CharacterID:   1002,
+			Number:        5,
+			Location:      220,
+			Chara1:        1002,
+			Chara2:        1018,
+			ConditionType: 2,
+		},
+	}
+	if diff := cmp.Diff(want, got); diff != "" {
+		t.Errorf("wrong conversations (+got/-want):\n%s", diff)
+	}
+}