refactor: add archiver constructor

ybirader · Aug 10, 2023 · aab3217 · aab3217
1 parent a48c565
commit aab3217
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 5 deletions.
diff --git a/main.go b/main.go
@@ -7,12 +7,16 @@ import (
 
 type Archiver struct {
 	Dest *os.File
-	W    *zip.Writer
+	w    *zip.Writer
+}
+
+func NewArchiver(archive *os.File) *Archiver {
+	return &Archiver{Dest: archive, w: zip.NewWriter(archive)}
 }
 
 func (a *Archiver) Archive() {
-	a.W.Create("hello.txt")
-	a.W.Close()
+	a.w.Create("hello.txt")
+	a.w.Close()
 }
 
 func main() {

diff --git a/main_test.go b/main_test.go
@@ -14,8 +14,7 @@ func TestArchive(t *testing.T) {
 		archive, cleanup := createTempArchive(t, "testdata/archive.zip")
 		defer cleanup()
 
-		writer := zip.NewWriter(archive)
-		archiver := Archiver{Dest: archive, W: writer}
+		archiver := NewArchiver(archive)
 		archiver.Archive()
 
 		archiveReader, err := zip.OpenReader(archive.Name())