| 8 | ) |
| 9 | |
| 10 | func TestGetPorts(t *testing.T) { |
| 11 | dir := t.TempDir() |
| 12 | |
| 13 | wdBackup, err := os.Getwd() |
| 14 | if err != nil { |
| 15 | t.Fatalf("Error getting current working directory: %v", err) |
| 16 | } |
| 17 | err = os.Chdir(dir) |
| 18 | if err != nil { |
| 19 | t.Fatalf("Error changing working directory: %v", err) |
| 20 | } |
| 21 | |
| 22 | // 8. Delete temp folder |
| 23 | defer func() { |
| 24 | err = os.Chdir(wdBackup) |
| 25 | if err != nil { |
| 26 | t.Fatalf("Error changing dir back: %v", err) |
| 27 | } |
| 28 | }() |
| 29 | |
| 30 | file, err := os.Create("Dockerfile") |
| 31 | if err != nil { |
| 32 | t.Fatalf("Error creating Dockerfile: %v", err) |
| 33 | } |
| 34 | _, err = file.Write([]byte(`FROM mysql |
| 35 | EXPOSE 8080 |
| 36 | EXPOSE `)) |
| 37 | if err != nil { |
| 38 | t.Fatalf("Error creating Dockerfile: %v", err) |
| 39 | } |
| 40 | err = file.Close() |
| 41 | if err != nil { |
| 42 | t.Fatalf("Error creating Dockerfile: %v", err) |
| 43 | } |
| 44 | |
| 45 | ports, err := GetPorts("Dockerfile") |
| 46 | if err != nil { |
| 47 | t.Fatalf("Error receiving ports: %v", err) |
| 48 | } |
| 49 | assert.Equal(t, 1, len(ports), "Wrong number of ports returned") |
| 50 | assert.Equal(t, 8080, ports[0], "Wrong port returned") |
| 51 | |
| 52 | } |