Tests writeJSONs

Author: cuducos

transformnext/writer_test.go

@@ -0,0 +1,82 @@
+package transformnext
+
+import (
+	"context"
+	"sync"
+	"testing"
+)
+
+type testDB struct {
+	lock sync.Mutex
+	data map[string]string
+}
+
+func (db *testDB) PreLoad() error {
+	db.lock.Lock()
+	defer db.lock.Unlock()
+	db.data = make(map[string]string)
+	return nil
+}
+
+func (db *testDB) CreateCompanies(companies [][]string) error {
+	db.lock.Lock()
+	defer db.lock.Unlock()
+	for _, c := range companies {
+		db.data[c[0]] = c[1]
+	}
+	return nil
+}
+
+func (db *testDB) PostLoad() error {
+	return nil
+}
+
+func (db *testDB) CreateExtraIndexes(indexes []string) error {
+	return nil
+}
+
+func (db *testDB) MetaSave(key, value string) error {
+	db.lock.Lock()
+	defer db.lock.Unlock()
+	db.data[key] = value
+	return nil
+}
+
+func TestWriteJSONs(t *testing.T) {
+	ctx := context.Background()
+	srcs := sources()
+	kv, err := newBadger(t.TempDir(), false)
+	if err != nil {
+		t.Fatalf("expected no error creating badger, got %s", err)
+	}
+	defer func() {
+		if err := kv.db.Close(); err != nil {
+			t.Errorf("expected no error closing badger, got %s", err)
+		}
+	}()
+	for key, src := range srcs {
+		if key == "est" {
+			continue
+		}
+		if err := loadCSVs(ctx, "../testdata", src, nil, kv); err != nil {
+			t.Fatalf("expected no error loading %s data, got %s", key, err)
+		}
+	}
+	db := &testDB{}
+	if err := db.PreLoad(); err != nil {
+		t.Fatalf("expected no error calling PreLoad, got %s", err)
+	}
+	err = writeJSONs(ctx, srcs, kv, db, 16, 8192, "../testdata", false)
+	if err != nil {
+		t.Fatalf("expected no error processing test data, got %s", err)
+	}
+	db.lock.Lock()
+	defer db.lock.Unlock()
+	if len(db.data) != 1 {
+		t.Errorf("expected 1 company to be persisted, got %d", len(db.data))
+	}
+	exp := "33683111000280"
+	if _, ok := db.data[exp]; !ok {
+		t.Errorf("expected CNPJ %s to be persisted, got nil", exp)
+	}
+}