add: ifaces testing
This commit is contained in:
4
interfaces/18-testing/.gitignore
vendored
Normal file
4
interfaces/18-testing/.gitignore
vendored
Normal file
@ -0,0 +1,4 @@
|
||||
rosie.unknown
|
||||
rosie.png
|
||||
_other.go
|
||||
toc
|
40
interfaces/18-testing/main.go
Normal file
40
interfaces/18-testing/main.go
Normal file
@ -0,0 +1,40 @@
|
||||
// Copyright © 2018 Inanc Gumus
|
||||
// Learn Go Programming Course
|
||||
// License: https://creativecommons.org/licenses/by-nc-sa/4.0/
|
||||
//
|
||||
// For more tutorials : https://learngoprogramming.com
|
||||
// In-person training : https://www.linkedin.com/in/inancgumus/
|
||||
// Follow me on twitter: https://twitter.com/inancgumus
|
||||
|
||||
package main
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"io"
|
||||
"net/http"
|
||||
"os"
|
||||
)
|
||||
|
||||
func main() {
|
||||
// resp, err := http.Get("https://inancgumus.github.com/x/rosie.jpg")
|
||||
resp, err := http.Get("https://inancgumus.github.com/x/rosie.unknown")
|
||||
if err != nil {
|
||||
fmt.Fprintln(os.Stderr, err)
|
||||
return
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
|
||||
file, err := os.Create("rosie.png")
|
||||
if err != nil {
|
||||
fmt.Fprintln(os.Stderr, err)
|
||||
return
|
||||
}
|
||||
defer file.Close()
|
||||
|
||||
n, err := io.Copy(file, pngReader(resp.Body))
|
||||
if err != nil {
|
||||
fmt.Fprintln(os.Stderr, err)
|
||||
return
|
||||
}
|
||||
fmt.Printf("%d bytes transferred.\n", n)
|
||||
}
|
48
interfaces/18-testing/reader.go
Normal file
48
interfaces/18-testing/reader.go
Normal file
@ -0,0 +1,48 @@
|
||||
// Copyright © 2018 Inanc Gumus
|
||||
// Learn Go Programming Course
|
||||
// License: https://creativecommons.org/licenses/by-nc-sa/4.0/
|
||||
//
|
||||
// For more tutorials : https://learngoprogramming.com
|
||||
// In-person training : https://www.linkedin.com/in/inancgumus/
|
||||
// Follow me on twitter: https://twitter.com/inancgumus
|
||||
|
||||
package main
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"fmt"
|
||||
"io"
|
||||
)
|
||||
|
||||
// reader reads from `r` if the stream starts with a given signature.
|
||||
// otherwise it stops and returns with an error.
|
||||
type signatureReader struct {
|
||||
r io.Reader // reads from the response.Body (or from any reader)
|
||||
signature []byte // stream should start with this initial signature
|
||||
}
|
||||
|
||||
// Read implements the io.Reader interface.
|
||||
func (sr *signatureReader) Read(b []byte) (n int, err error) {
|
||||
n, err = sr.r.Read(b)
|
||||
|
||||
l := len(sr.signature)
|
||||
if l == 0 {
|
||||
return
|
||||
}
|
||||
if lb := len(b); lb < l {
|
||||
l = lb
|
||||
}
|
||||
if got, want := b[:l], sr.signature[:l]; !bytes.Equal(got, want) {
|
||||
err = fmt.Errorf("signature doesn't match, got: % x, want: % x", got, want)
|
||||
}
|
||||
sr.signature = sr.signature[l:]
|
||||
return n, err
|
||||
}
|
||||
|
||||
// create a reader for detecting only the png signatures.
|
||||
func pngReader(r io.Reader) io.Reader {
|
||||
return &signatureReader{
|
||||
r: r,
|
||||
signature: []byte("\x89PNG\r\n\x1a\n"),
|
||||
}
|
||||
}
|
34
interfaces/18-testing/reader_test.go
Normal file
34
interfaces/18-testing/reader_test.go
Normal file
@ -0,0 +1,34 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"io"
|
||||
"strings"
|
||||
"testing"
|
||||
)
|
||||
|
||||
func Test_PNG_Reader_Correct_Signature(t *testing.T) {
|
||||
const input = "\x89PNG\r\n\x1a\nHELLO"
|
||||
|
||||
out, err := readSignature(input)
|
||||
if err != nil {
|
||||
t.Fatalf("got: %q; want: <nil> err", err)
|
||||
}
|
||||
if out != input {
|
||||
t.Fatalf("invalid output, got: '%x'; want: '%x'", out, input)
|
||||
}
|
||||
}
|
||||
|
||||
func Test_PNG_Reader_Incorrect_Signature(t *testing.T) {
|
||||
_, err := readSignature("\x89INCORRECT")
|
||||
if err == nil {
|
||||
t.Fatal("got: nil; want: !nil err")
|
||||
}
|
||||
}
|
||||
|
||||
func readSignature(in string) (string, error) {
|
||||
r := strings.NewReader(in)
|
||||
w := strings.Builder{}
|
||||
// _, err := io.Copy(&w, pngReader(r))
|
||||
_, err := io.CopyBuffer(&w, pngReader(r), make([]byte, 1))
|
||||
return w.String(), err
|
||||
}
|
Reference in New Issue
Block a user