dadata/api/model/model_test.go

package model

import (
	"encoding/json"
	"os"
	"testing"

	"github.com/stretchr/testify/assert"
)

func TestBank_Recursive(t *testing.T) {
	testBankReader, err := os.Open("test/bank.json")
	if err != nil {
		t.Error(err)
	}
	defer testBankReader.Close()

	bank := Bank{}

	err = json.NewDecoder(testBankReader).Decode(&bank)
	assert.NoError(t, err)
	assert.Equal(t, "111", bank.Bic)
	assert.Equal(t, "222", bank.Rkc.Bic)
	assert.Equal(t, "333", bank.Rkc.Rkc.Bic)
	assert.Equal(t, "444", bank.Rkc.Rkc.Rkc.Bic)
}

func TestParty(t *testing.T) {
	testPartyReader, err := os.Open("test/party.json")
	if err != nil {
		t.Error(err)
	}
	defer testPartyReader.Close()

	model := Party{}

	err = json.NewDecoder(testPartyReader).Decode(&model)
	assert.NoError(t, err)
	assert.Equal(t, 2, len(model.Founders))
	assert.Equal(t, PartyFounderTypeLegal, model.Founders[0].Type)
	assert.Equal(t, PartySMBCategorySmall, model.Documents.Smb.Category)
}