Skip to content

Commit

Permalink
change csv package name
Browse files Browse the repository at this point in the history
  • Loading branch information
kenanfarukcakir committed Apr 19, 2023
1 parent 2646607 commit 6ac2ca3
Show file tree
Hide file tree
Showing 3 changed files with 48 additions and 4 deletions.
4 changes: 2 additions & 2 deletions core/engine.go
Original file line number Diff line number Diff line change
Expand Up @@ -30,7 +30,7 @@ import (
"go.ddosify.com/ddosify/core/proxy"
"go.ddosify.com/ddosify/core/report"
"go.ddosify.com/ddosify/core/scenario"
"go.ddosify.com/ddosify/core/scenario/testdata"
"go.ddosify.com/ddosify/core/scenario/data"
"go.ddosify.com/ddosify/core/types"
)

Expand Down Expand Up @@ -378,7 +378,7 @@ var readTestData = func(testDataConf map[string]types.CsvConf) (map[string]types
for k, conf := range testDataConf {
var rows []map[string]interface{}
var err error
rows, err = testdata.ReadCsv(conf)
rows, err = data.ReadCsv(conf)
if err != nil {
return nil, err
}
Expand Down
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
package testdata
package data

import (
"encoding/csv"
Expand Down
Original file line number Diff line number Diff line change
@@ -1,8 +1,10 @@
package testdata
package data

import (
"errors"
"fmt"
"net/http"
"net/http/httptest"
"reflect"
"strings"
"testing"
Expand Down Expand Up @@ -79,6 +81,48 @@ func TestWrapAsRemoteCsvError(t *testing.T) {
}
}

func TestReadCsvFromRemote(t *testing.T) {
// Test server
handler := func(w http.ResponseWriter, r *http.Request) {
w.WriteHeader(http.StatusUnauthorized)
}

path := "/csv"
mux := http.NewServeMux()
mux.HandleFunc(path, handler)

server := httptest.NewServer(mux)
defer server.Close()

conf := types.CsvConf{
Path: server.URL + path,
Delimiter: ";",
SkipFirstLine: true,
Vars: map[string]types.Tag{
"0": {Tag: "name", Type: "string"},
"3": {Tag: "payload", Type: "json"},
"4": {Tag: "age", Type: "int"},
"5": {Tag: "percent", Type: "float"},
"6": {Tag: "boolField", Type: "bool"},
},
SkipEmptyLine: true,
AllowQuota: true,
Order: "sequential",
}

_, err := ReadCsv(conf)

if err == nil {
t.Errorf("TestReadCsvFromRemote %v", err)
}

var remoteCsvErr RemoteCsvError
if !errors.As(err, &remoteCsvErr) {
t.Errorf("Expected: %v, Found: %v", remoteCsvErr, err)
}

}

func TestReadCsv(t *testing.T) {
t.Parallel()
conf := types.CsvConf{
Expand Down

0 comments on commit 6ac2ca3

Please sign in to comment.