Skip to content

Commit be5e172

Browse files
paskalumputun
authored andcommitted
rewrite TestGetContentCustom to work offline
1 parent d681d21 commit be5e172

1 file changed

Lines changed: 14 additions & 2 deletions

File tree

backend/extractor/readability_test.go

Lines changed: 14 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -152,14 +152,26 @@ func (m RulesMock) All(_ context.Context) []datastore.Rule { retu
152152
func TestGetContentCustom(t *testing.T) {
153153
lr := UReadability{TimeOut: 30, SnippetSize: 200, Rules: RulesMock{}}
154154
httpClient := &http.Client{Timeout: time.Second * 30}
155-
resp, err := httpClient.Get("https://p.umputun.com/2015/09/25/poiezdka-s-apple-maps/")
155+
ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
156+
if r.URL.String() == "/2015/09/25/poiezdka-s-apple-maps/" {
157+
fh, err := os.Open("testdata/poiezdka-s-apple-maps.html")
158+
testHTML, err := io.ReadAll(fh)
159+
assert.NoError(t, err)
160+
assert.NoError(t, fh.Close())
161+
_, err = w.Write(testHTML)
162+
assert.NoError(t, err)
163+
return
164+
}
165+
}))
166+
defer ts.Close()
167+
resp, err := httpClient.Get(ts.URL + "/2015/09/25/poiezdka-s-apple-maps/")
156168
assert.NoError(t, err)
157169
defer resp.Body.Close()
158170
dataBytes, err := io.ReadAll(resp.Body)
159171
assert.NoError(t, err)
160172
body := string(dataBytes)
161173

162-
content, rich, err := lr.getContent(context.Background(), body, "https://p.umputun.com/2015/09/25/poiezdka-s-apple-maps/")
174+
content, rich, err := lr.getContent(context.Background(), body, ts.URL+"/2015/09/25/poiezdka-s-apple-maps/")
163175
assert.NoError(t, err)
164176
assert.Equal(t, 6988, len(content))
165177
assert.Equal(t, 7169, len(rich))

0 commit comments

Comments
 (0)