add testing for divid and bugfix

Author: imura81gt

kadai3/imura81gt/rget/go.mod

@@ -2,4 +2,7 @@ module github.com/gopherdojo/dojo7/kadai3/imura81gt/rget
 
 go 1.13
 
-require golang.org/x/sync v0.0.0-20190911185100-cd5d95a43a6e
+require (
+	github.com/google/go-cmp v0.3.1
+	golang.org/x/sync v0.0.0-20190911185100-cd5d95a43a6e
+)

kadai3/imura81gt/rget/go.sum

@@ -1,2 +1,4 @@
+github.com/google/go-cmp v0.3.1 h1:Xye71clBPdm5HgqGwUkwhbynsUJZhDbS20FvLhQ2izg=
+github.com/google/go-cmp v0.3.1/go.mod h1:8QqcDgzrUqlUb/G2PQTWiueGozuR1884gddMywk6iLU=
 golang.org/x/sync v0.0.0-20190911185100-cd5d95a43a6e h1:vcxGaoTs7kV8m5Np9uUNQin4BrLOthgV7252N8V+FwY=
 golang.org/x/sync v0.0.0-20190911185100-cd5d95a43a6e/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=

kadai3/imura81gt/rget/rget.go

@@ -22,9 +22,18 @@ type Option struct {
 }
 
 type Unit struct {
-	RangeStart   int64
-	RangeEnd     int64
-	TempFileName string
+	RangeStart     int64
+	RangeEnd       int64
+	TempFileName   string
+	DownloadedSize int64
+}
+
+func (u *Unit) Write(data []byte) (int, error) {
+	d := len(data)
+	u.DownloadedSize += int64(d)
+	// fmt.Printf("%v is downloaded %v/%v \n",
+	// 	u.TempFileName, u.DownloadedSize, u.RangeEnd-u.RangeStart+1)
+	return d, nil
 }
 
 type Units []Unit
@@ -105,7 +114,7 @@ func (o *Option) divide() {
 	}
 
 	// TODO: should distribute the remainder to each unit
-	units[len(units)-1].RangeEnd = o.ContentLength
+	units[len(units)-1].RangeEnd = (o.ContentLength - 1)
 
 	o.Units = units
 }
@@ -178,7 +187,7 @@ func (o *Option) downloadWithContext(
 		return nil
 	}()
 
-	_, err = io.Copy(w, resp.Body)
+	_, err = io.Copy(w, io.TeeReader(resp.Body, &o.Units[i]))
 	if err != nil {
 		return fmt.Errorf("Error: %v", err)
 	}

kadai3/imura81gt/rget/rget_test.go

@@ -1 +1,115 @@
 package rget
+
+import (
+	"testing"
+
+	"github.com/google/go-cmp/cmp"
+)
+
+func TestWrite(t *testing.T) {
+}
+
+func TestRun(t *testing.T) {
+}
+
+func TestContentLength(t *testing.T) {
+}
+
+func TestAcceptRangesHeaderCheck(t *testing.T) {
+}
+
+//func divide(contentLength int64, concurrency int) Units {
+func TestDivide(t *testing.T) {
+
+	const (
+		url           = "https://example.com/test.iso"
+		contentLength = 5
+	)
+
+	testCases := []struct {
+		caseName    string
+		concurrency uint
+		expected    Option
+	}{
+		{
+			caseName:    "ContentLength:5/5",
+			concurrency: 5,
+			expected: Option{
+				URL:           url,
+				ContentLength: contentLength,
+				Concurrency:   5,
+				Units: []Unit{
+					{TempFileName: "0_test.iso", RangeStart: 0, RangeEnd: 0},
+					{TempFileName: "1_test.iso", RangeStart: 1, RangeEnd: 1},
+					{TempFileName: "2_test.iso", RangeStart: 2, RangeEnd: 2},
+					{TempFileName: "3_test.iso", RangeStart: 3, RangeEnd: 3},
+					{TempFileName: "4_test.iso", RangeStart: 4, RangeEnd: 4},
+				},
+			},
+		},
+		{
+			caseName:    "ContentLength:5/1",
+			concurrency: 1,
+			expected: Option{
+				URL:           url,
+				ContentLength: contentLength,
+				Concurrency:   1,
+				Units: []Unit{
+					{TempFileName: "0_test.iso", RangeStart: 0, RangeEnd: 4},
+				},
+			},
+		},
+		{
+			caseName:    "ContentLength:5/10",
+			concurrency: 10,
+			expected: Option{
+				URL:           url,
+				ContentLength: contentLength,
+				Concurrency:   5,
+				Units: []Unit{
+					{TempFileName: "0_test.iso", RangeStart: 0, RangeEnd: 0},
+					{TempFileName: "1_test.iso", RangeStart: 1, RangeEnd: 1},
+					{TempFileName: "2_test.iso", RangeStart: 2, RangeEnd: 2},
+					{TempFileName: "3_test.iso", RangeStart: 3, RangeEnd: 3},
+					{TempFileName: "4_test.iso", RangeStart: 4, RangeEnd: 4},
+				},
+			},
+		},
+		{
+			caseName:    "ContentLength:5/0",
+			concurrency: 0,
+			expected: Option{
+				URL:           url,
+				ContentLength: contentLength,
+				Concurrency:   1,
+				Units: []Unit{
+					{TempFileName: "0_test.iso", RangeStart: 0, RangeEnd: 4},
+				},
+			},
+		},
+	}
+
+	for _, tc := range testCases {
+		tc := tc
+		t.Run(tc.caseName, func(t *testing.T) {
+			t.Parallel()
+
+			o := Option{URL: url, ContentLength: contentLength, Concurrency: tc.concurrency}
+			o.divide()
+
+			if !cmp.Equal(o, tc.expected) {
+				t.Errorf("actual: %+v\nexpected: %+v\n", o, tc.expected)
+			}
+
+		})
+	}
+}
+
+func TestParallelDownload(t *testing.T) {
+}
+
+func TestDownloadWithContext(t *testing.T) {
+}
+
+func TestCombine(t *testing.T) {
+}