Function TestDownstreamServer

helper/server/downstream_test.go:18–181

(t *testing.T)

)

func TestDownstreamServer(t *testing.T) {
	fromDir := t.TempDir()
	toDir := t.TempDir()

	err := createFiles(fromDir, fileStructure)
	if err != nil {
		t.Fatal(err)
	}

	err = createFiles(toDir, overwriteFileStructure)
	if err != nil {
		t.Fatal(err)
	}

	// Upload tar with client
	clientReader, clientWriter := io.Pipe()
	serverReader, serverWriter := io.Pipe()

	go func() {
		err := StartDownstreamServer(serverReader, clientWriter, &DownstreamOptions{
			RemotePath:   fromDir,
			ExcludePaths: []string{"emptydir"},
			ExitOnClose:  false,
			Polling:      true,
		})
		if err != nil {
			panic(err)
		}
	}()

	conn, err := util.NewClientConnection(clientReader, serverWriter)
	if err != nil {
		t.Fatal(err)
	}

	// Count changes
	client := remote.NewDownstreamClient(conn)
	amount, err := client.ChangesCount(context.Background(), &remote.Empty{})
	if err != nil {
		t.Fatal(err)
	}
	if amount.Amount == 0 {
		t.Fatalf("Unexpected change amount, expected >0, got %d", amount.Amount)
	}

	changesClient, err := client.Changes(context.Background(), &remote.Empty{})
	if err != nil {
		t.Fatal(err)
	}

	log.Println("Created downstream server & client")

	changes, err := getAllChanges(changesClient)
	if err != nil {
		t.Fatal(err)
	}

	log.Println("Got all changes")