godownloadthat/filedownloader_test.go at master · abrad1212/godownloadthat · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
package godownloadthat_test

import (
	"os"
	"testing"

	"github.com/abrad1212/godownloadthat"
)

func BenchmarkDownloadFiles(b *testing.B) {
	urls := []string{
		"https://img.icons8.com/cotton/512/000000/chrome.png",
		"https://img.icons8.com/cotton/512/000000/safari.png",
	}

	fileNames := []string{
		"chrome.png",
		"safari.png",
	}

	downloader := godownloadthat.Downloader{
		Debug: false,
	}

	for n := 0; n < b.N; n++ {
		downloader.DownloadFiles(urls, fileNames)
	}
}

func TestMain(m *testing.M) {
	// Setup

	// Run Test
	code := m.Run()

	// Teardown
	os.Remove("google.png")
	os.Remove("chrome.png")

	// Exit
	os.Exit(code)
}

func TestDownloadFiles(t *testing.T) {
	t.Run("Main", func(t *testing.T) {
		t.Parallel()
		downloader := godownloadthat.Downloader{
			Debug: true,
		}

		t.Run("1 URL", func(t *testing.T) {
			url := []string{
				"http://pluspng.com/img-png/google-logo-png-open-2000.png",
			}
			fileName := []string{
				"google.png",
			}
			err := downloader.DownloadFiles(url, fileName)

			if err != nil {
				t.Error(err)
			}
		})

		t.Run("2 URLs", func(t *testing.T) {
			url := []string{
				"http://pluspng.com/img-png/google-logo-png-open-2000.png",
				"https://img.icons8.com/cotton/512/000000/chrome.png",
			}
			fileName := []string{
				"google.png",
				"chrome.png",
			}
			err := downloader.DownloadFiles(url, fileName)

			if err != nil {
				t.Error(err)
			}
		})

		t.Run("Mismatch URL Array Length", func(t *testing.T) {
			urls := []string{
				"http://pluspng.com/img-png/google-logo-png-open-2000.png",
			}
			fileNames := []string{
				"google.png",
				"chrome.png",
			}
			err := downloader.DownloadFiles(urls, fileNames)

			// Expecting the error not to be nil
			if err == nil {
				t.Error("Expected an error, the URL's length doesn't match the filename's length")
			}
		})

		t.Run("Mismatch Filename Array Length", func(t *testing.T) {
			urls := []string{
				"http://pluspng.com/img-png/google-logo-png-open-2000.png",
				"http://pluspng.com/img-png/google-logo-png-open-2000.png",
			}
			fileNames := []string{
				"google.png",
			}
			err := downloader.DownloadFiles(urls, fileNames)

			// Expecting error here
			if err == nil {
				t.Error("Expected an error, the length of the URL array doesn't match the length of the filename array")
			}
		})

		t.Run("URL Doesn't Exist", func(t *testing.T) {
			url := []string{
				"http://randomImageWebsiteOrSomething.com",
			}
			fileName := []string{
				"google.png",
			}
			err := downloader.DownloadFiles(url, fileName)

			if err == nil {
				t.Error("Expected a non 200 status code error")
			}
		})
	})
}