Implement 2021/day04 puzzle solution (#62)

Author: obalunenko

internal/puzzles/common/utils/file.go

@@ -0,0 +1,26 @@
+// Package utils provide common used functionality to work with files, readers and so on.
+package utils
+
+import (
+	"io"
+	"os"
+	"path/filepath"
+	"testing"
+
+	"github.com/stretchr/testify/require"
+)
+
+// ReaderFromFile reads file from fpath and returns content as io.Reader.
+// File descriptor will be closed on tests teardown.
+func ReaderFromFile(tb testing.TB, fpath string) io.Reader {
+	tb.Helper()
+
+	file, err := os.Open(filepath.Clean(fpath))
+	require.NoError(tb, err)
+
+	tb.Cleanup(func() {
+		require.NoError(tb, file.Close())
+	})
+
+	return file
+}

internal/puzzles/input/content.go

@@ -41,6 +41,15 @@ func Get(ctx context.Context, d Date, session string) ([]byte, error) {
 
 	client := http.DefaultClient
 
+	const (
+		timeoutSecs = 5
+	)
+
+	ctx, cancel := context.WithTimeout(ctx, time.Second*timeoutSecs)
+	defer cancel()
+
+	req = req.Clone(ctx)
+
 	resp, err := client.Do(req)
 	if err != nil {
 		return nil, fmt.Errorf("send request: %w", err)
@@ -105,5 +114,8 @@ func createInputReq(ctx context.Context, d Date, sessionID string) (*http.Reques
 		Unparsed:   nil,
 	})
 
+	req.Header.Set("User-Agent",
+		"Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/96.0.4664.93 Safari/537.36")
+
 	return req, nil
 }

internal/puzzles/solutions/2019/day01/solution_test.go

@@ -1,13 +1,12 @@
 package day01
 
 import (
-	"io"
-	"os"
 	"path/filepath"
 	"testing"
 
 	"github.com/stretchr/testify/assert"
-	"github.com/stretchr/testify/require"
+
+	"github.com/obalunenko/advent-of-code/internal/puzzles/common/utils"
 )
 
 func Test_solution_Year(t *testing.T) {
@@ -200,15 +199,6 @@ func Test_calcPart2(t *testing.T) {
 	close(in)
 }
 
-func readerFromFile(tb testing.TB, fpath string) io.Reader {
-	tb.Helper()
-
-	file, err := os.Open(fpath)
-	require.NoError(tb, err)
-
-	return file
-}
-
 func Test_calc(t *testing.T) {
 	type args struct {
 		inputPath string
@@ -245,7 +235,7 @@ func Test_calc(t *testing.T) {
 		tt := tt
 
 		t.Run(tt.name, func(t *testing.T) {
-			input := readerFromFile(t, tt.args.inputPath)
+			input := utils.ReaderFromFile(t, tt.args.inputPath)
 			got, err := calc(input, tt.args.calcFn)
 
 			if tt.wantErr {