Fix IRB container package

- implement docker API changes

Signed-off-by: Marcus Brandenburger <bur@zurich.ibm.com>

Author: mbrandenburger

samples/demos/Makefile

@@ -8,6 +8,6 @@ include $(TOP)/build.mk
 
 DEMOS = irb
 
-build clean:
+build test clean:
 	$(foreach DIR, $(DEMOS), $(MAKE) -C $(DIR) $@ || exit;)
 

samples/demos/irb/.gitignore

@@ -1,3 +1,3 @@
-cmd/**
+cmd/
 users/**
 pkg/protos/**

samples/demos/irb/Makefile

@@ -7,7 +7,7 @@ include $(TOP)/build.mk
 
 COMPONENTS = protos chaincode experimenter
 
-all: pull-images ercc build test
+all: ercc build test
 
 pull-images:
 	$(DOCKER) pull redis:latest
@@ -23,11 +23,11 @@ ercc:
 build: ercc
 	$(foreach DIR, $(COMPONENTS), $(MAKE) -C $(DIR) $@ || exit;)
 
-test:
-	$(GO) test -v ./...
+test: pull-images
+	$(GO) test -v -failfast ./...
 
 run:
-	$(GO) test -v .
+	$(GO) test -v -failfast .
 
 clean:
 	$(GO) clean

samples/demos/irb/experimenter/Makefile

@@ -17,4 +17,4 @@ run: build
 	$(MAKE) -C worker stop-docker run-docker
 
 test:
-	$(GO) test -v ./...
+	$(GO) test -v -failfast ./...

samples/demos/irb/experimenter/test/worker_test.go

@@ -8,23 +8,24 @@ package test
 
 import (
 	"bytes"
+	"context"
 	"fmt"
 	"io"
 	"net/http"
 	"testing"
-	"time"
 
-	"github.com/hyperledger/fabric-private-chaincode/samples/demos/irb/pkg/container"
 	"github.com/hyperledger/fabric-private-chaincode/samples/demos/irb/pkg/crypto"
 	pb "github.com/hyperledger/fabric-private-chaincode/samples/demos/irb/pkg/protos"
 	"github.com/hyperledger/fabric-private-chaincode/samples/demos/irb/pkg/storage"
-	"github.com/hyperledger/fabric-private-chaincode/samples/demos/irb/pkg/utils"
 	"github.com/pkg/errors"
+	"github.com/stretchr/testify/require"
+	"github.com/testcontainers/testcontainers-go"
+	"github.com/testcontainers/testcontainers-go/network"
+	"github.com/testcontainers/testcontainers-go/wait"
 	"google.golang.org/protobuf/proto"
 )
 
 func requestAttestation() ([]byte, error) {
-
 	resp, err := http.Get("http://localhost:5000/attestation")
 	if err != nil {
 		return nil, err
@@ -57,7 +58,6 @@ func requestAttestation() ([]byte, error) {
 }
 
 func submitEvaluationPack(pk []byte, req *pb.RegisterDataRequest) error {
-
 	epm := &pb.EvaluationPackMessage{}
 	epm.RegisteredData = []*pb.RegisterDataRequest{req}
 
@@ -134,7 +134,7 @@ func upload() (*pb.RegisterDataRequest, error) {
 		PublicKey: []byte("some verification key"),
 	}
 
-	//build request
+	// build request
 	registerDataRequest := &pb.RegisterDataRequest{
 		Participant:   &userIdentity,
 		DecryptionKey: sk,
@@ -148,70 +148,66 @@ func upload() (*pb.RegisterDataRequest, error) {
 const networkID = "mytestnetwork"
 
 func TestWorker(t *testing.T) {
-
-	network := &container.Network{Name: networkID}
-	err := network.Create()
-	defer network.Remove()
-	if err != nil {
-		panic(err)
-	}
-
-	// setup redis
-	redis := &container.Container{
-		Image:    "redis",
-		Name:     "redis-container",
-		HostIP:   "localhost",
-		HostPort: "6379",
-		Network:  networkID,
-	}
-	err = redis.Start()
-	defer redis.Stop()
-	if err != nil {
-		panic(err)
-	}
-
-	// setup experiment container
-	experiment := &container.Container{
-		Image:    "irb-experimenter-worker",
-		Name:     "experiment-container",
-		HostIP:   "localhost",
-		HostPort: "5000",
-		Env:      []string{"REDIS_HOST=redis-container"},
-		Network:  networkID,
-	}
-	err = experiment.Start()
-	defer experiment.Stop()
-	if err != nil {
-		panic(err)
-	}
-
-	// let's wait until experiment service is up an running
-	err = utils.Retry(func() bool {
-		resp, err := http.Get("http://localhost:5000/info")
-		if err != nil {
-			return false
-		}
-		return resp.StatusCode == 200
-	}, 5, 60*time.Second, 2*time.Second)
-	if err != nil {
-		panic(err)
-	}
+	ctx := context.Background()
+
+	// network
+	net, err := network.New(ctx)
+	require.NoError(t, err)
+	defer func() {
+		err := net.Remove(ctx)
+		require.NoError(t, err)
+	}()
+	networkName := net.Name
+
+	// redis
+	redisImageName := "redis"
+	redis, err := testcontainers.GenericContainer(ctx, testcontainers.GenericContainerRequest{
+		ContainerRequest: testcontainers.ContainerRequest{
+			Image:        redisImageName,
+			ExposedPorts: []string{fmt.Sprintf("%d/tcp", storage.DefaultRedisPort)},
+			Networks:     []string{networkName},
+			WaitingFor:   wait.ForLog("* Ready to accept connections"),
+		},
+		Started: true,
+	})
+	require.NoError(t, err)
+	defer func() {
+		err := redis.Terminate(ctx)
+		require.NoError(t, err)
+	}()
+
+	redisHost, err := redis.Host(ctx)
+	require.NoError(t, err)
+
+	// experiment
+	experimentImageName := "irb-experimenter-worker"
+	experimentPort := "5000/tcp"
+	experiment, err := testcontainers.GenericContainer(ctx, testcontainers.GenericContainerRequest{
+		ContainerRequest: testcontainers.ContainerRequest{
+			Image:        experimentImageName,
+			ExposedPorts: []string{experimentPort},
+			Networks:     []string{networkName},
+			Env:          map[string]string{"REDIS_HOST": redisHost},
+			WaitingFor:   wait.ForExposedPort(),
+		},
+		Started: true,
+	})
+	require.NoError(t, err)
+	defer func() {
+		err := experiment.Terminate(ctx)
+		require.NoError(t, err)
+	}()
 
 	req, err := upload()
-	if err != nil {
-		panic(err)
-	}
+	require.NoError(t, err)
 
 	fmt.Println("Testing attestation...")
 	pk, err := requestAttestation()
-	if err != nil {
-		panic(err)
-	}
+	require.NoError(t, err)
 
 	fmt.Println("Testing evaluation pack...")
-	if err := submitEvaluationPack(pk, req); err != nil {
-		panic(err)
-	}
+	err = submitEvaluationPack(pk, req)
+	require.NoError(t, err)
 
 	fmt.Println("Test done.")
 }