FilesExpand file tree

master

/

dktest_test.go

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

113 lines (99 loc) · 2.64 KB

master

/

dktest_test.go

Copy path

Top

File metadata and controls

• Code
• Blame

113 lines (99 loc) · 2.64 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

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

package dktest_test

import (

"context"

"errors"

"fmt"

"net/http"

"net/url"

"testing"

"github.com/dhui/dktest"

"github.com/docker/go-connections/nat"

)

const (

testImage = "alpine"

testNetworkImage = "nginx:alpine"

)

// ready functions

func nginxReady(ctx context.Context, c dktest.ContainerInfo) bool {

ip, port, err := c.FirstPort()

if err != nil {

return false

}

u := url.URL{Scheme: "http", Host: ip + ":" + port}

fmt.Println(u.String())

req, err := http.NewRequest(http.MethodGet, u.String(), nil)

if err != nil {

fmt.Println("req err", err)

return false

}

req = req.WithContext(ctx)

if resp, err := http.DefaultClient.Do(req); err != nil {

fmt.Println("do err:", err)

return false

} else if resp.StatusCode != 200 {

return false

}

return true

}

// test functions

func noop(*testing.T, dktest.ContainerInfo) {}

func TestRun(t *testing.T) {

dktest.Run(t, testImage, dktest.Options{}, noop)

}

func TestRunContext(t *testing.T) {

t.Run("success", func(t *testing.T) {

err := dktest.RunContext(context.Background(), t, testImage, dktest.Options{}, func(dktest.ContainerInfo) error {

return nil

})

if err != nil {

t.Fatal("failed", err)

}

})

t.Run("test func returns error", func(t *testing.T) {

var errForTest = errors.New("testFunc failed")

err := dktest.RunContext(context.Background(), t, testImage, dktest.Options{}, func(dktest.ContainerInfo) error {

return errForTest

})

if err == nil {

t.Fatal("expected error")

}

if errors.Unwrap(err) != errForTest {

t.Fatal("test func error not propagated with cause, got error:", err)

}

})

}

func TestRunParallel(t *testing.T) {

numTests := 10

for i := 0; i < numTests; i++ {

t.Run("", func(t *testing.T) {

t.Parallel()

dktest.Run(t, testImage, dktest.Options{}, noop)

})

}

}

func TestRunWithNetwork(t *testing.T) {

dktest.Run(t, testNetworkImage, dktest.Options{ReadyFunc: nginxReady, PortRequired: true}, noop)

}

func TestRunWithNetworkPortBinding(t *testing.T) {

port, err := nat.NewPort("tcp", "80")

if err != nil {

t.Fatal("Invalid port:", err)

}

dktest.Run(t, testNetworkImage, dktest.Options{ReadyFunc: nginxReady, PortRequired: true,

PortBindings: nat.PortMap{port: []nat.PortBinding{{HostPort: "8181"}}}}, noop)

}

func TestRunWithExposesPorts(t *testing.T) {

port, err := nat.NewPort("tcp", "8080")

if err != nil {

t.Fatal("Invalid port:", err)

}

dktest.Run(t, testNetworkImage, dktest.Options{

ReadyFunc: nginxReady,

PortRequired: true,

ExposedPorts: nat.PortSet{port: struct{}{}},

}, func(t *testing.T, info dktest.ContainerInfo) {

if _, ok := info.Ports[port]; !ok {

t.Fatal("port not exposed")

}

})

}