Compare commits
1 Commits
Author | SHA1 | Date | |
---|---|---|---|
6128f82149 |
@@ -1,29 +0,0 @@
|
|||||||
package script_test
|
|
||||||
|
|
||||||
import (
|
|
||||||
"slices"
|
|
||||||
"testing"
|
|
||||||
|
|
||||||
"code.uint32.ru/dmitry/script"
|
|
||||||
)
|
|
||||||
|
|
||||||
func TestCSVReader(t *testing.T) {
|
|
||||||
t.Parallel()
|
|
||||||
r, err := script.NewCSVReader("testdata/sample_csv.csv")
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
want := []string{"one", "two", "three"}
|
|
||||||
|
|
||||||
for range 2 {
|
|
||||||
row, err := r.Read()
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
if !slices.Equal(row, want) {
|
|
||||||
t.Fatalf("rows not equal, want: %v, have: %v", want, row)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
@@ -1,42 +0,0 @@
|
|||||||
package script
|
|
||||||
|
|
||||||
import (
|
|
||||||
"bytes"
|
|
||||||
"os"
|
|
||||||
"testing"
|
|
||||||
)
|
|
||||||
|
|
||||||
func TestCSVWriter(t *testing.T) {
|
|
||||||
path := "testdata/output_csv.csv"
|
|
||||||
|
|
||||||
os.Remove(path)
|
|
||||||
defer os.Remove(path)
|
|
||||||
|
|
||||||
w, err := newCSVwriter(path)
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
want := []byte("one,two,three\none,two,three\n")
|
|
||||||
|
|
||||||
row := []string{"one", "two", "three"}
|
|
||||||
|
|
||||||
for range 2 {
|
|
||||||
if err := w.Write(row); err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if err := w.Close(); err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
b, err := os.ReadFile(path)
|
|
||||||
if err != nil {
|
|
||||||
t.Fatalf("err reading output file: %s", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
if !bytes.Equal(want, b) {
|
|
||||||
t.Errorf("incorrect result, want: %s, have: %s", string(want), string(b))
|
|
||||||
}
|
|
||||||
}
|
|
2
testdata/sample_csv.csv
vendored
2
testdata/sample_csv.csv
vendored
@@ -1,2 +0,0 @@
|
|||||||
one,two,three
|
|
||||||
one,two,three
|
|
|
Reference in New Issue
Block a user