From 7891f27b62cbf5759ddf0c4b2138825809e4ae51 Mon Sep 17 00:00:00 2001
From: Zack Scholl <zack.scholl@gmail.com>
Date: Mon, 24 Sep 2018 08:35:24 -0700
Subject: [PATCH] fix test

---
 src/croc/croc.go      | 2 ++
 src/croc/croc_test.go | 5 +++--
 2 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/src/croc/croc.go b/src/croc/croc.go
index d0f040e6..c25c2cb2 100644
--- a/src/croc/croc.go
+++ b/src/croc/croc.go
@@ -59,6 +59,8 @@ func Init(debug bool) (c *Croc) {
 	c.RelayTCPPort = "8154"
 	c.CurveType = "siec"
 	c.Address = "198.199.67.130"
+	c.AddressWebsocketPort = "8153"
+	c.AddressTCPPort = "8154"
 	c.NoRecipientPrompt = true
 	debugLevel := "info"
 	if debug {
diff --git a/src/croc/croc_test.go b/src/croc/croc_test.go
index 95e9757a..b9783d8e 100644
--- a/src/croc/croc_test.go
+++ b/src/croc/croc_test.go
@@ -16,7 +16,8 @@ import (
 func sendAndReceive(t *testing.T, forceSend int) {
 	var startTime time.Time
 	var durationPerMegabyte float64
-	fname := generateRandomFile(100)
+	megabytes := 10
+	fname := generateRandomFile(megabytes)
 	var wg sync.WaitGroup
 	wg.Add(2)
 	go func() {
@@ -36,7 +37,7 @@ func sendAndReceive(t *testing.T, forceSend int) {
 		c.ForceSend = forceSend
 		startTime = time.Now()
 		assert.Nil(t, c.Receive("test"))
-		durationPerMegabyte = 100.0 / time.Since(startTime).Seconds()
+		durationPerMegabyte = float64(megabytes) / time.Since(startTime).Seconds()
 		assert.True(t, utils.Exists(fname))
 	}()
 	wg.Wait()