Added tests

Author: DWethmar

day07/main_test.go

@@ -54,7 +54,13 @@ func TestSumNodesWithSizeLT(t *testing.T) {
 		args args
 		want int
 	}{
-		// TODO: Add test cases.
+		{
+			name: "example",
+			args: args{
+				node: exampleTree,
+			},
+			want: 95437,
+		},
 	}
 	for _, tt := range tests {
 		t.Run(tt.name, func(t *testing.T) {
@@ -77,66 +83,7 @@ func TestSmallestFolderThatWouldFreeUpSpace(t *testing.T) {
 		{
 			name: "example",
 			args: args{
-				node: &Node{
-					Name: "",
-					Children: []*Node{
-						{
-							Name: "a",
-							Children: []*Node{
-								{
-									Name: "e",
-									Children: []*Node{
-										{
-											Name: "i",
-											Size: 584,
-										},
-									},
-								},
-								{
-									Name: "f",
-									Size: 29116,
-								},
-								{
-									Name: "g",
-									Size: 2557,
-								},
-								{
-									Name: "h.lst",
-									Size: 62596,
-								},
-							},
-						},
-						{
-							Name: "b.txt",
-							Size: 14848514,
-						},
-						{
-							Name: "c.txt",
-							Size: 8504156,
-						},
-						{
-							Name: "d",
-							Children: []*Node{
-								{
-									Name: "j",
-									Size: 4060174,
-								},
-								{
-									Name: "d.log",
-									Size: 8033020,
-								},
-								{
-									Name: "d.ext",
-									Size: 5626152,
-								},
-								{
-									Name: "k",
-									Size: 7214296,
-								},
-							},
-						},
-					},
-				},
+				node: exampleTree,
 			},
 			want: 24933642,
 		},

day07/node_test.go

@@ -96,7 +96,6 @@ func TestNewNode(t *testing.T) {
 func TestNodeSize(t *testing.T) {
 	type args struct {
 		node *Node
-		sum  int
 	}
 	tests := []struct {
 		name string
@@ -106,53 +105,9 @@ func TestNodeSize(t *testing.T) {
 		{
 			name: "test",
 			args: args{
-				node: &Node{
-					Parent: nil,
-					Children: []*Node{
-						{
-							Parent:   nil,
-							Children: []*Node{},
-							Name:     "a",
-							Size:     1,
-						},
-						{
-							Parent: nil,
-							Children: []*Node{
-								{
-									Parent:   nil,
-									Children: []*Node{},
-									Name:     "b",
-									Size:     1,
-								},
-								{
-									Parent:   nil,
-									Children: []*Node{},
-									Name:     "c",
-									Size:     1,
-								},
-								{
-									Parent: nil,
-									Children: []*Node{
-										{
-											Parent:   nil,
-											Children: []*Node{},
-											Name:     "c",
-											Size:     1,
-										},
-									},
-									Name: "c",
-									Size: 0,
-								},
-							},
-							Name: "test",
-							Size: 0,
-						},
-					},
-					Name: "x",
-					Size: 0,
-				},
+				node: exampleTree,
 			},
-			want: 4,
+			want: 48381165,
 		},
 	}
 	for _, tt := range tests {

day07/tree.go

@@ -0,0 +1,62 @@
+package main
+
+var exampleTree = &Node{
+	Name: "",
+	Children: []*Node{
+		{
+			Name: "a",
+			Children: []*Node{
+				{
+					Name: "e",
+					Children: []*Node{
+						{
+							Name: "i",
+							Size: 584,
+						},
+					},
+				},
+				{
+					Name: "f",
+					Size: 29116,
+				},
+				{
+					Name: "g",
+					Size: 2557,
+				},
+				{
+					Name: "h.lst",
+					Size: 62596,
+				},
+			},
+		},
+		{
+			Name: "b.txt",
+			Size: 14848514,
+		},
+		{
+			Name: "c.txt",
+			Size: 8504156,
+		},
+		{
+			Name: "d",
+			Children: []*Node{
+				{
+					Name: "j",
+					Size: 4060174,
+				},
+				{
+					Name: "d.log",
+					Size: 8033020,
+				},
+				{
+					Name: "d.ext",
+					Size: 5626152,
+				},
+				{
+					Name: "k",
+					Size: 7214296,
+				},
+			},
+		},
+	},
+}