Function TestImportCPC

zipcodes/zipcodes_test.go:595–645

(t *testing.T)

}

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

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

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

		localities["001"] = &Localidade{
			Codigo:     "001",
			UF:         "MG",
			Nome:       "Belo Horizonte",
			CodigoIBGE: "3106200",
		}

		cpcFile := filepath.Join(tmpDir, "LOG_CPC.TXT")
		content := "001@MG@001@CPC Savassi@Rua Pernambuco 1000@30130-150\n"

		err := os.WriteFile(cpcFile, []byte(content), 0644)
		require.NoError(t, err)

		count, err := importer.importCPC(cpcFile)
		assert.NoError(t, err)
		assert.Equal(t, 1, count)

		err = importer.db.View(func(txn *badger.Txn) error {
			item, err := txn.Get([]byte("cep:30130150"))
			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, "30130150", cep.CEP)
				assert.Equal(t, "Rua Pernambuco 1000", cep.Logradouro)
				assert.Equal(t, "Belo Horizonte", cep.Cidade)
				assert.Equal(t, "cpc", cep.TipoOrigem)
				assert.Equal(t, "CPC Savassi", cep.NomeOrigem)
				assert.Empty(t, cep.Bairro)
				return nil
			})
		})
		assert.NoError(t, err)
	})
}