diff --git a/internal/toukibo/stock_converter.go b/internal/toukibo/stock_converter.go index 2e52847..20fefa7 100644 --- a/internal/toukibo/stock_converter.go +++ b/internal/toukibo/stock_converter.go @@ -93,6 +93,10 @@ func GetHoujinStock(stock string) HoujinStock { continue } + // trim (%d) + // sample1042用のハック + stock = regexp.MustCompile(`([0-9]+)`).ReplaceAllString(stock, "") + // *優先株式 or *種類株式で始まる場合 pattern := fmt.Sprintf("([%s]+-[0-9]種優先株式|[%s]+[0-9]種優先株式|[%s]+[0-9]優先株式|[%s]+優先株式|[%s]+種類株式|[%s]+種株式)", ZenkakuNoNumberStringPattern, ZenkakuNoNumberStringPattern, ZenkakuNoNumberStringPattern, ZenkakuNoNumberStringPattern, ZenkakuNoNumberStringPattern, ZenkakuNoNumberStringPattern) diff --git a/toukibo_parser_test.go b/toukibo_parser_test.go index b087fb3..eaafb9e 100644 --- a/toukibo_parser_test.go +++ b/toukibo_parser_test.go @@ -142,9 +142,8 @@ func TestToukiboParser(t *testing.T) { } stock := h.GetHoujinStock() - if stock.Total != stock.Sum() { - // TODO: make it error - fmt.Println("stock.Total != stock.Sum()") + if stock.Sum() != 0 && stock.Total != stock.Sum() { + t.Errorf("stock.Total != stock.Sum(), want: %d, got: %d, detail: %v", stock.Total, stock.Sum(), stock) } if fmt.Sprint(stock.Total) != td.HoujinStock { t.Fatalf("stock is not match,\nwant : %s,\ngot : %d,", td.HoujinStock, stock.Total)