Function TestLoadLocalities

zipcodes/zipcodes_test.go:184–247

(t *testing.T)

}

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

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

	t.Run("load valid localities file", func(t *testing.T) {
		localities = make(map[string]*Localidade)

		localitiesFile := filepath.Join(tmpDir, "LOG_LOCALIDADE.TXT")
		content := "001@SP@São Paulo@01000-000@1@M@@SP@3550308\n" +
			"002@RJ@Rio de Janeiro@20000-000@1@M@@RJ@3304557\n" +
			"003@MG@Belo Horizonte@30000-000@1@M@@BH@3106200\n"

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

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

		err = importer.loadLocalities(localitiesFile)
		assert.NoError(t, err)
		assert.Equal(t, 3, len(localities))

		sp := localities["001"]
		assert.NotNil(t, sp)
		assert.Equal(t, "SP", sp.UF)
		assert.Equal(t, "São Paulo", sp.Nome)
		assert.Equal(t, "01000000", sp.CEP)
		assert.Equal(t, "3550308", sp.CodigoIBGE)
	})

	t.Run("skip invalid lines", func(t *testing.T) {
		localities = make(map[string]*Localidade)

		localitiesFile := filepath.Join(tmpDir, "LOG_LOCALIDADE_INVALID.TXT")
		content := "001@SP@São Paulo@01000-000@1@M@@SP@3550308\n" +
			"invalid line\n" +
			"002@RJ\n" +
			"003@MG@Belo Horizonte@30000-000@1@M@@BH@3106200\n"

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

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

		err = importer.loadLocalities(localitiesFile)
		assert.NoError(t, err)
		assert.Equal(t, 2, len(localities))
	})