Merge pull request #3 from madflojo/canada

Anonymous contribution

baton_test.go

@@ -13,7 +13,7 @@
  * or implied. See the License for the specific language governing
  * permissions and limitations under the License.
  */
- 
+
 package main
 
 import (

configuration.go

@@ -13,7 +13,7 @@
  * or implied. See the License for the specific language governing
  * permissions and limitations under the License.
  */
- 
+
 package main
 
 import (

csv_parsing.go

@@ -13,7 +13,7 @@
  * or implied. See the License for the specific language governing
  * permissions and limitations under the License.
  */
- 
+
 package main
 
 import (

log_writer.go

@@ -13,7 +13,7 @@
  * or implied. See the License for the specific language governing
  * permissions and limitations under the License.
  */
- 
+
 package main
 
 import "fmt"

timed_worker.go

@@ -13,7 +13,7 @@
  * or implied. See the License for the specific language governing
  * permissions and limitations under the License.
  */
- 
+
 package main
 
 import (

worker.go

@@ -19,6 +19,7 @@ package main
 import (
 	"github.com/valyala/fasthttp"
 	"math/rand"
+	"strings"
 	"time"
 )
 
@@ -97,6 +98,10 @@ func buildRequest(requests []preLoadedRequest, totalPremadeRequests int) (*fasth
 	req.Header.SetMethod(currentReq.method)
 	req.SetBodyString(currentReq.body)
 	for i := 0; i < len(currentReq.headers); i++ {
+		if strings.EqualFold(currentReq.headers[i][0], "Content-type") {
+			req.Header.SetContentType(currentReq.headers[i][1])
+			continue
+		}
 		req.Header.Add(currentReq.headers[i][0], currentReq.headers[i][1])
 	}
 	return req, resp