fix(git): Added headers for each option (#13)

durandj · Mar 14, 2019 · 7dec028 · 7dec028
1 parent 4a9f59b
commit 7dec028
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/pkg/gitignore/git_adapter.go b/pkg/gitignore/git_adapter.go
@@ -81,6 +81,7 @@ func (adapter *GitAdapter) Generate(options []string) (string, error) {
 			return "", errors.Wrap(err, message)
 		}
 
+		builder.WriteString(fmt.Sprintf("### %s ###\n", option))
 		builder.Write(contents)
 		builder.WriteString("\n")
 	}

diff --git a/pkg/gitignore/git_adapter_test.go b/pkg/gitignore/git_adapter_test.go
@@ -98,7 +98,9 @@ var _ = Describe("GitAdapter", func() {
 
 			Expect(err).To(BeNil())
 
+			Expect(contents).To(ContainSubstring("### C ###"))
 			Expect(contents).To(ContainSubstring("*.o"))
+			Expect(contents).To(ContainSubstring("### Python ###"))
 			Expect(contents).To(ContainSubstring("__pycache__/"))
 		})