Improve the burndown test

burndown_test.go

@@ -18,25 +18,29 @@ func TestBurndownMeta(t *testing.T) {
 	}
 }
 
-func TestBurndownConsume(t *testing.T) {
+func TestBurndownConsumeFinalize(t *testing.T) {
 	burndown := BurndownAnalysis{
 		Granularity:  30,
 		Sampling:     30,
-		PeopleNumber: 1,
+		PeopleNumber: 2,
 	}
 	burndown.Initialize(testRepository)
 	deps := map[string]interface{}{}
+
+	// stage 1
 	deps["author"] = 0
 	deps["day"] = 0
 	cache := map[plumbing.Hash]*object.Blob{}
 	hash := plumbing.NewHash("291286b4ac41952cbd1389fda66420ec03c1a9fe")
 	cache[hash], _ = testRepository.BlobObject(hash)
-	hash = plumbing.NewHash("334cde09da4afcb74f8d2b3e6fd6cce61228b485")
+	hash = plumbing.NewHash("c29112dbd697ad9b401333b80c18a63951bc18d9")
+	cache[hash], _ = testRepository.BlobObject(hash)
+	hash = plumbing.NewHash("baa64828831d174f40140e4b3cfa77d1e917a2c1")
 	cache[hash], _ = testRepository.BlobObject(hash)
 	hash = plumbing.NewHash("dc248ba2b22048cc730c571a748e8ffcf7085ab9")
 	cache[hash], _ = testRepository.BlobObject(hash)
 	deps["blob_cache"] = cache
-	changes := make(object.Changes, 2)
+	changes := make(object.Changes, 3)
 	treeFrom, _ := testRepository.TreeObject(plumbing.NewHash(
 		"a1eb2ea76eb7f9bfbde9b243861474421000eb96"))
 	treeTo, _ := testRepository.TreeObject(plumbing.NewHash(
@@ -55,10 +59,20 @@ func TestBurndownConsume(t *testing.T) {
 		TreeEntry: object.TreeEntry{
 			Name: "analyser.go",
 			Mode: 0100644,
-			Hash: plumbing.NewHash("334cde09da4afcb74f8d2b3e6fd6cce61228b485"),
+			Hash: plumbing.NewHash("baa64828831d174f40140e4b3cfa77d1e917a2c1"),
 		},
 	}}
 	changes[1] = &object.Change{From: object.ChangeEntry{}, To: object.ChangeEntry{
+		Name: "cmd/hercules/main.go",
+		Tree: treeTo,
+		TreeEntry: object.TreeEntry{
+			Name: "cmd/hercules/main.go",
+			Mode: 0100644,
+			Hash: plumbing.NewHash("c29112dbd697ad9b401333b80c18a63951bc18d9"),
+		},
+	},
+	}
+	changes[2] = &object.Change{From: object.ChangeEntry{}, To: object.ChangeEntry{
 		Name: ".travis.yml",
 		Tree: treeTo,
 		TreeEntry: object.TreeEntry{
@@ -77,22 +91,149 @@ func TestBurndownConsume(t *testing.T) {
 	assert.Nil(t, result)
 	assert.Nil(t, err)
 	assert.Equal(t, burndown.previousDay, 0)
-	assert.Equal(t, len(burndown.files), 2)
+	assert.Equal(t, len(burndown.files), 3)
+	assert.Equal(t, burndown.files["cmd/hercules/main.go"].Len(), 207)
+	assert.Equal(t, burndown.files["analyser.go"].Len(), 926)
 	assert.Equal(t, burndown.files[".travis.yml"].Len(), 12)
-	assert.Equal(t, burndown.files["analyser.go"].Len(), 309)
-	assert.Equal(t, len(burndown.people), 1)
-	assert.Equal(t, burndown.people[0][0], int64(12+309))
+	assert.Equal(t, len(burndown.people), 2)
+	assert.Equal(t, burndown.people[0][0], int64(12+207+926))
 	assert.Equal(t, len(burndown.globalStatus), 1)
-	assert.Equal(t, burndown.globalStatus[0], int64(12+309))
+	assert.Equal(t, burndown.globalStatus[0], int64(12+207+926))
 	assert.Equal(t, len(burndown.globalHistory), 0)
 	assert.Equal(t, len(burndown.fileHistories), 0)
-	burndown = BurndownAnalysis{
+	burndown2 := BurndownAnalysis{
 		Granularity:  30,
 		Sampling:     0,
 		PeopleNumber: 0,
 	}
-	burndown.Initialize(testRepository)
-	_, err = burndown.Consume(deps)
+	burndown2.Initialize(testRepository)
+	_, err = burndown2.Consume(deps)
+	assert.Nil(t, err)
+	assert.Equal(t, len(burndown2.people), 0)
+
+	// stage 2
+	// 2b1ed978194a94edeabbca6de7ff3b5771d4d665
+	deps["author"] = 1
+	deps["day"] = 30
+	cache = map[plumbing.Hash]*object.Blob{}
+	hash = plumbing.NewHash("291286b4ac41952cbd1389fda66420ec03c1a9fe")
+	cache[hash], _ = testRepository.BlobObject(hash)
+	hash = plumbing.NewHash("baa64828831d174f40140e4b3cfa77d1e917a2c1")
+	cache[hash], _ = testRepository.BlobObject(hash)
+	hash = plumbing.NewHash("29c9fafd6a2fae8cd20298c3f60115bc31a4c0f2")
+	cache[hash], _ = testRepository.BlobObject(hash)
+	hash = plumbing.NewHash("c29112dbd697ad9b401333b80c18a63951bc18d9")
+	cache[hash], _ = testRepository.BlobObject(hash)
+	hash = plumbing.NewHash("f7d918ec500e2f925ecde79b51cc007bac27de72")
+	cache[hash], _ = testRepository.BlobObject(hash)
+	deps["blob_cache"] = cache
+	changes = make(object.Changes, 3)
+	treeFrom, _ = testRepository.TreeObject(plumbing.NewHash(
+		"96c6ece9b2f3c7c51b83516400d278dea5605100"))
+	treeTo, _ = testRepository.TreeObject(plumbing.NewHash(
+		"251f2094d7b523d5bcc60e663b6cf38151bf8844"))
+	changes[0] = &object.Change{From: object.ChangeEntry{
+		Name: "analyser.go",
+		Tree: treeFrom,
+		TreeEntry: object.TreeEntry{
+			Name: "analyser.go",
+			Mode: 0100644,
+			Hash: plumbing.NewHash("baa64828831d174f40140e4b3cfa77d1e917a2c1"),
+		},
+	}, To: object.ChangeEntry{
+		Name: "burndown.go",
+		Tree: treeTo,
+		TreeEntry: object.TreeEntry{
+			Name: "burndown.go",
+			Mode: 0100644,
+			Hash: plumbing.NewHash("29c9fafd6a2fae8cd20298c3f60115bc31a4c0f2"),
+		},
+	},
+	}
+	changes[1] = &object.Change{From: object.ChangeEntry{
+		Name: "cmd/hercules/main.go",
+		Tree: treeFrom,
+		TreeEntry: object.TreeEntry{
+			Name: "cmd/hercules/main.go",
+			Mode: 0100644,
+			Hash: plumbing.NewHash("c29112dbd697ad9b401333b80c18a63951bc18d9"),
+		},
+	}, To: object.ChangeEntry{
+		Name: "cmd/hercules/main.go",
+		Tree: treeTo,
+		TreeEntry: object.TreeEntry{
+			Name: "cmd/hercules/main.go",
+			Mode: 0100644,
+			Hash: plumbing.NewHash("f7d918ec500e2f925ecde79b51cc007bac27de72"),
+		},
+	},
+	}
+	changes[2] = &object.Change{From: object.ChangeEntry{
+		Name: ".travis.yml",
+		Tree: treeTo,
+		TreeEntry: object.TreeEntry{
+			Name: ".travis.yml",
+			Mode: 0100644,
+			Hash: plumbing.NewHash("291286b4ac41952cbd1389fda66420ec03c1a9fe"),
+		},
+	}, To: object.ChangeEntry{},
+	}
+	deps["renamed_changes"] = changes
+	fd = fixtureFileDiff()
+	result, err = fd.Consume(deps)
 	assert.Nil(t, err)
-	assert.Equal(t, len(burndown.people), 0)
+	deps["file_diff"] = result["file_diff"]
+	result, err = burndown.Consume(deps)
+	assert.Nil(t, result)
+	assert.Nil(t, err)
+	assert.Equal(t, burndown.previousDay, 30)
+	assert.Equal(t, len(burndown.files), 2)
+	assert.Equal(t, burndown.files["cmd/hercules/main.go"].Len(), 290)
+	assert.Equal(t, burndown.files["burndown.go"].Len(), 543)
+	assert.Equal(t, len(burndown.people), 2)
+	assert.Equal(t, len(burndown.globalStatus), 2)
+	assert.Equal(t, burndown.globalStatus[0], int64(464))
+	assert.Equal(t, burndown.globalStatus[1], int64(0))
+	assert.Equal(t, len(burndown.globalHistory), 1)
+	assert.Equal(t, len(burndown.globalHistory[0]), 2)
+	assert.Equal(t, len(burndown.fileHistories), 3)
+	out := burndown.Finalize().(BurndownResult)
+	/*
+	GlobalHistory   [][]int64
+	FileHistories   map[string][][]int64
+	PeopleHistories [][][]int64
+	PeopleMatrix    [][]int64
+	*/
+	assert.Equal(t, len(out.GlobalHistory), 2)
+	for i := 0; i < 2; i++ {
+		assert.Equal(t, len(out.GlobalHistory[i]), 2)
+	}
+	assert.Equal(t, len(out.GlobalHistory), 2)
+	assert.Equal(t, out.GlobalHistory[0][0], int64(1145))
+	assert.Equal(t, out.GlobalHistory[0][1], int64(0))
+	assert.Equal(t, out.GlobalHistory[1][0], int64(464))
+	assert.Equal(t, out.GlobalHistory[1][1], int64(369))
+	assert.Equal(t, len(out.FileHistories), 2)
+	assert.Equal(t, len(out.FileHistories["cmd/hercules/main.go"]), 2)
+	assert.Equal(t, len(out.FileHistories["burndown.go"]), 2)
+	assert.Equal(t, len(out.FileHistories["cmd/hercules/main.go"][0]), 2)
+	assert.Equal(t, len(out.FileHistories["burndown.go"][0]), 2)
+	assert.Equal(t, len(out.PeopleMatrix), 2)
+	assert.Equal(t, len(out.PeopleMatrix[0]), 4)
+	assert.Equal(t, len(out.PeopleMatrix[1]), 4)
+	assert.Equal(t, out.PeopleMatrix[0][0], int64(1145))
+	assert.Equal(t, out.PeopleMatrix[0][1], int64(0))
+	assert.Equal(t, out.PeopleMatrix[0][2], int64(0))
+	assert.Equal(t, out.PeopleMatrix[0][3], int64(-681))
+	assert.Equal(t, out.PeopleMatrix[1][0], int64(369))
+	assert.Equal(t, out.PeopleMatrix[1][1], int64(0))
+	assert.Equal(t, out.PeopleMatrix[1][2], int64(0))
+	assert.Equal(t, out.PeopleMatrix[1][3], int64(0))
+	assert.Equal(t, len(out.PeopleHistories), 2)
+	for i := 0; i < 2; i++ {
+		assert.Equal(t, len(out.PeopleHistories[i]), 2)
+		assert.Equal(t, len(out.PeopleHistories[i][0]), 2)
+		assert.Equal(t, len(out.PeopleHistories[i][1]), 2)
+	}
 }
+