atualizando testes

dadosjusbr · Dec 19, 2024 · f6caec4 · f6caec4
1 parent 8fd1caa
commit f6caec4
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/repo/database/postgres.go b/repo/database/postgres.go
@@ -326,7 +326,6 @@ func (p *PostgresDB) GetMonthlyInfo(agencies []models.Agency, year int) (map[str
 }
 
 func (p *PostgresDB) GetAnnualSummary(agency string) ([]models.AnnualSummary, error) {
-	// var dtoAgmi dto.AgencyMonthlyInfoDTO
 	var dtoAmis []dto.AnnualSummaryDTO
 	agency = strings.ToLower(agency)
 

diff --git a/repo/database/postgres_test.go b/repo/database/postgres_test.go
@@ -788,13 +788,15 @@ func (g getMonthlyInfo) testWhenMonthlyInfoExists(t *testing.T) {
 			Month:             1,
 			CrawlingTimestamp: timestamppb.Now(),
 			ManualCollection:  false,
+			Inconsistent:      false,
 		},
 		{
 			AgencyID:          "tjsp",
 			Year:              2020,
 			Month:             1,
 			CrawlingTimestamp: timestamppb.Now(),
 			ManualCollection:  true,
+			Inconsistent:      false,
 		},
 	}
 	if err := insertMonthlyInfos(agmis); err != nil {
@@ -1191,6 +1193,7 @@ func (g getOMA) testWhenDataExists(t *testing.T) {
 	assert.Equal(t, agmi.Month, returnedAgmi.Month)
 	assert.Equal(t, agmi.Year, returnedAgmi.Year)
 	assert.Equal(t, agencies[0], *agency)
+	assert.Equal(t, returnedAgmi.Inconsistent, false)
 	truncateTables()
 }