-
Notifications
You must be signed in to change notification settings - Fork 0
/
parser_test.go
75 lines (54 loc) · 6.33 KB
/
parser_test.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
package stock_parser
import (
"fmt"
"strings"
"testing"
"github.com/longbridgeapp/assert"
)
func assert_matches_code(t *testing.T, expected string, input string) {
t.Helper()
res := input
cb := func(code, market, match string) string {
counterId := fmt.Sprintf("ST/%s/%s", market, code)
name := code
s := fmt.Sprintf(`<span type="security-tag" counter_id="%s" name="%s">$%s.%s</span>`, counterId, name, name, market)
res = strings.ReplaceAll(res, match, s)
return res
}
out := Parse(input, cb)
assert.Equal(t, expected, out)
}
func TestParse(t *testing.T) {
assert_matches_code(t, `Alibaba<span type="security-tag" counter_id="ST/US/BABA" name="BABA">$BABA.US</span> published its Q2 results`, "Alibaba BABA.US published its Q2 results")
assert_matches_code(t, `Alibaba <span type="security-tag" counter_id="ST/US/BABA" name="BABA">$BABA.US</span> published its Q2 results`, "Alibaba $BABA$ published its Q2 results")
assert_matches_code(t, `阿里巴巴<span type="security-tag" counter_id="ST/US/BABA" name="BABA">$BABA.US</span> 发布财报`, "阿里巴巴$BABA.US 发布财报")
assert_matches_code(t, `阿里巴巴<span type="security-tag" counter_id="ST/US/BABA" name="BABA">$BABA.US</span>发布财报`, "阿里巴巴$BABA.US$发布财报")
assert_matches_code(t, `阿里巴巴<span type="security-tag" counter_id="ST/US/BABA" name="BABA">$BABA.US</span>$发布财报`, "阿里巴巴 BABA.US$发布财报")
assert_matches_code(t, `阿里巴巴<span type="security-tag" counter_id="ST/US/BABA" name="BABA">$BABA.US</span> 发布财报`, "阿里巴巴 BABA.US 发布财报")
assert_matches_code(t, "阿里巴巴 BABA$发布财报", "阿里巴巴 BABA$发布财报")
assert_matches_code(t, "腾讯 700 发布财报", "腾讯 700 发布财报")
assert_matches_code(t, "阿里巴巴 [BABA] 发布财报", "阿里巴巴 [BABA] 发布财报")
assert_matches_code(t, "腾讯 (700) 发布财报", "腾讯 (700) 发布财报")
assert_matches_code(t, `腾讯<span type="security-tag" counter_id="ST/HK/00700" name="00700">$00700.HK</span> 发布财报`, "腾讯 00700.HK 发布财报")
assert_matches_code(t, "Only the fortune of Tesla's (TSLA)", "Only the fortune of Tesla's (TSLA)")
assert_matches_code(t, `阿里巴巴<span type="security-tag" counter_id="ST/US/BABA" name="BABA">$BABA.US</span>和腾讯<span type="security-tag" counter_id="ST/HK/700" name="700">$700.HK</span>发布财报`, "阿里巴巴$BABA.US$和腾讯$700.HK$发布财报")
assert_matches_code(t, `阿里巴巴 <span type="security-tag" counter_id="ST/US/BABA" name="BABA">$BABA.US</span> 发布财报`, "阿里巴巴 $BABA 发布财报")
assert_matches_code(t, `吉利交付 <span type="security-tag" counter_id="ST/HK/00175" name="00175">$00175.HK</span> 股票, 哈哈`, `吉利交付 $00175.HK 股票, 哈哈`)
assert_matches_code(t, `理想汽車 <span type="security-tag" counter_id="ST/HK/2015" name="2015">$2015.HK</span> 升不足 1%，阿里巴巴 <span type="security-tag" counter_id="ST/US/BABA" name="BABA">$BABA.US</span> 升 1.06%。`, "理想汽車 (HK:2015) 升不足 1%，阿里巴巴 (US:BABA) 升 1.06%。")
assert_matches_code(t, `<span type="security-tag" counter_id="ST/US/XPEV" name="XPEV">$XPEV.US</span> 小鹏汽车除了吹牛还能干什么`, "$XPeng (XPEV.US)$ 小鹏汽车除了吹牛还能干什么")
}
func TestXueqiuLaohuFutu(t *testing.T) {
// 老虎雪球
assert_matches_code(t, `啊哈哈哈 <span type="security-tag" counter_id="ST/US/BABA" name="BABA">$BABA.US</span> <span type="security-tag" counter_id="ST/SH/601318" name="601318">$601318.SH</span> <span type="security-tag" counter_id="ST/HK/00700" name="00700">$00700.HK</span> 这些天，江苏省 100 多家外贸企业在境外参展招商，开拓市场。这家大型显示设备的制造企业前期团队已在国外开展对接，这次他们带了 20 多项新技术。`, "啊哈哈哈 $阿里巴巴 (BABA)$ $中国平安 (SH601318)$ $腾讯控股 (00700)$ 这些天，江苏省 100 多家外贸企业在境外参展招商，开拓市场。这家大型显示设备的制造企业前期团队已在国外开展对接，这次他们带了 20 多项新技术。")
assert_matches_code(t, `<span type="security-tag" counter_id="ST/HK/09988" name="09988">$09988.HK</span> 中概股要发财报了，第一枪 今晚 百度来开，密切关注。`, `$百度集团-SW(09988)$ 中概股要发财报了，第一枪 今晚 百度来开，密切关注。`)
assert_matches_code(t, `<span type="security-tag" counter_id="ST/HK/HSTECH" name="HSTECH">$HSTECH.HK</span> your grandpa is still your grandpa! 又高又硬`, `$恒生科技指数(HKHSTECH)$ your grandpa is still your grandpa! 又高又硬`)
assert_matches_code(t, `<span type="security-tag" counter_id="ST/US/COIN" name="COIN">$COIN.US</span> 开户现货交易了!`, `$CoinBase Global(COIN)$ 开户现货交易了!`)
assert_matches_code(t, `<span type="security-tag" counter_id="ST/SH/603200" name="603200">$603200.SH</span> 哈哈`, "$上海洗霸 (SH603200)$ 哈哈")
assert_matches_code(t, `<p><span type="security-tag" counter_id="ST/US/.IXIC" name=".IXIC">$.IXIC.US</span><br/>都说等美股回调买入，回调7%了，咱就说如果出现10%以上的回调，你敢买入吗？<br/>0%持仓美股的投资者，可以考虑开始定投ETF（溢价也都掉下来了）啦，埋下投资美股的第一粒种子～</p>`, `<p>$纳斯达克综合指数(.IXIC)$<br/>都说等美股回调买入，回调7%了，咱就说如果出现10%以上的回调，你敢买入吗？<br/>0%持仓美股的投资者，可以考虑开始定投ETF（溢价也都掉下来了）啦，埋下投资美股的第一粒种子～</p>`)
// 富途
assert_matches_code(t, `<span type="security-tag" counter_id="ST/US/BABA" name="BABA">$BABA.US</span> 不错的哈哈哈 <span type="security-tag" counter_id="ST/HK/00700" name="00700">$00700.HK</span> 看好 <span type="security-tag" counter_id="ST/SZ/002241" name="002241">$002241.SZ</span> 也不错`, "$阿里巴巴 (BABA.US)$ 不错的哈哈哈 $腾讯控股 (00700.HK)$ 看好 $歌尔股份 (002241.SZ)$ 也不错")
}
func TestSpecialMarket(t *testing.T) {
assert_matches_code(t, `美股简短的股票代码<span type="security-tag" counter_id="ST/US/Q" name="Q">$Q.US</span><span type="security-tag" counter_id="ST/US/AB" name="AB">$AB.US</span>`, "美股简短的股票代码 Q.US AB.US")
assert_matches_code(t, "$ Google-A.US", "$ Google-A.US")
}