Function TestImportOperationalUnits

zipcodes/zipcodes_test.go:534–593

(t *testing.T)

}

func TestImportOperationalUnits(t *testing.T) {
	importer, cleanup := setupImporter(t)
	defer cleanup()

	tmpDir, err := os.MkdirTemp("", "uop-test-*")
	require.NoError(t, err)
	defer os.RemoveAll(tmpDir)

	t.Run("import operational units successfully", func(t *testing.T) {
		localities = make(map[string]*Localidade)
		districts = make(map[string]*Bairro)
		seenCEPs = make(map[string]bool)

		localities["001"] = &Localidade{
			Codigo:     "001",
			UF:         "RJ",
			Nome:       "Rio de Janeiro",
			CodigoIBGE: "3304557",
		}
		districts["002"] = &Bairro{
			Codigo: "002",
			Nome:   "Copacabana",
		}

		uopFile := filepath.Join(tmpDir, "LOG_UNID_OPER.TXT")
		content := "001@RJ@001@002@@Agência Central@Av Atlântica 1000@22070-000@CP@Ag Ctr\n"

		// Encode content to ISO-8859-1
		encoder := charmap.ISO8859_1.NewEncoder()
		encodedContent, err := encoder.String(content)
		require.NoError(t, err)

		err = os.WriteFile(uopFile, []byte(encodedContent), 0644)
		require.NoError(t, err)

		count, err := importer.importOperationalUnits(uopFile)
		assert.NoError(t, err)
		assert.Equal(t, 1, count)

		err = importer.db.View(func(txn *badger.Txn) error {
			item, err := txn.Get([]byte("cep:22070000"))
			if err != nil {
				return err
			}

			return item.Value(func(val []byte) error {
				var cep CEPCompleto
				err := json.Unmarshal(val, &cep)
				assert.NoError(t, err)
				assert.Equal(t, "22070000", cep.CEP)
				assert.Equal(t, "Av Atlântica 1000", cep.Logradouro)
				assert.Equal(t, "Copacabana", cep.Bairro)
				assert.Equal(t, "unid_oper", cep.TipoOrigem)
				assert.Equal(t, "Agência Central", cep.NomeOrigem)
				return nil
			})
		})
		assert.NoError(t, err)