77 lines
2.6 KiB
Go
77 lines
2.6 KiB
Go
|
package cache
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"errors"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"chis_osi/osi"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestDictionarySnapshotLooksUpMasterDataByName(t *testing.T) {
|
||
|
|
snapshot := NewDictionarySnapshot(
|
||
|
|
[]osi.GridAddress{{RegionCode: "441625000000", RegionName: "测试网格"}},
|
||
|
|
[]osi.Doctor{{PersonID: "doc-001", PersonName: "测试医生"}},
|
||
|
|
[]osi.Org{{OrganizCode: "123456789", OrganizName: "测试机构"}},
|
||
|
|
)
|
||
|
|
|
||
|
|
if got, ok := snapshot.RegionCodeByName("测试网格"); !ok || got != "441625000000" {
|
||
|
|
t.Fatalf("RegionCodeByName = %q, %v", got, ok)
|
||
|
|
}
|
||
|
|
if got, ok := snapshot.DoctorIDByName("测试医生"); !ok || got != "doc-001" {
|
||
|
|
t.Fatalf("DoctorIDByName = %q, %v", got, ok)
|
||
|
|
}
|
||
|
|
if got, ok := snapshot.ManaUnitIDByName("测试机构"); !ok || got != "123456789" {
|
||
|
|
t.Fatalf("ManaUnitIDByName = %q, %v", got, ok)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestDictionaryServiceRefreshKeepsMemoryWhenPersistentStoreFails(t *testing.T) {
|
||
|
|
client := fakePublicClient{
|
||
|
|
grids: []osi.GridAddress{{RegionCode: "441625000000", RegionName: "测试网格"}},
|
||
|
|
doctors: []osi.Doctor{{PersonID: "doc-001", PersonName: "测试医生"}},
|
||
|
|
orgs: []osi.Org{{OrganizCode: "123456789", OrganizName: "测试机构"}},
|
||
|
|
}
|
||
|
|
service := NewDictionaryService(client, failingStore{})
|
||
|
|
|
||
|
|
snapshot, err := service.Refresh(context.Background(), RefreshParams{
|
||
|
|
Grid: osi.GridAddressQuery{ParentCode: "441625", PageNo: 1, OperateUser: "712041"},
|
||
|
|
Doctors: osi.DoctorQuery{ManaUnitID: "123456789", OperateUser: "712041"},
|
||
|
|
Orgs: osi.OrgQuery{OrganizCode: "123456789"},
|
||
|
|
})
|
||
|
|
if err != nil {
|
||
|
|
t.Fatalf("Refresh: %v", err)
|
||
|
|
}
|
||
|
|
if got, ok := snapshot.DoctorIDByName("测试医生"); !ok || got != "doc-001" {
|
||
|
|
t.Fatalf("snapshot doctor = %q, %v", got, ok)
|
||
|
|
}
|
||
|
|
cached := service.Snapshot()
|
||
|
|
if got, ok := cached.RegionCodeByName("测试网格"); !ok || got != "441625000000" {
|
||
|
|
t.Fatalf("memory snapshot region = %q, %v", got, ok)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
type fakePublicClient struct {
|
||
|
|
grids []osi.GridAddress
|
||
|
|
doctors []osi.Doctor
|
||
|
|
orgs []osi.Org
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c fakePublicClient) QueryGridAddress(context.Context, osi.GridAddressQuery) ([]osi.GridAddress, osi.Result, error) {
|
||
|
|
return c.grids, osi.Result{Code: "01", Success: true}, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c fakePublicClient) QueryDoctors(context.Context, osi.DoctorQuery) ([]osi.Doctor, osi.Result, error) {
|
||
|
|
return c.doctors, osi.Result{Code: "01", Success: true}, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c fakePublicClient) QueryOrgs(context.Context, osi.OrgQuery) ([]osi.Org, osi.Result, error) {
|
||
|
|
return c.orgs, osi.Result{Code: "01", Success: true}, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
type failingStore struct{}
|
||
|
|
|
||
|
|
func (failingStore) SaveDictionarySnapshot(context.Context, DictionarySnapshot) error {
|
||
|
|
return errors.New("redis unavailable")
|
||
|
|
}
|