fix: get right datasource in ConceptExists

pieterlukasse · pieterlukasse · commit ce8bb4c82699 · 2024-10-04T14:01:02.000+02:00
diff --git a/tests/testutils.go b/tests/testutils.go
@@ -195,7 +195,8 @@ func RemoveConcept(sourceType models.SourceType, conceptId int64) {
 }
 
 func ConceptExists(sourceType models.SourceType, conceptId int64) bool {
-	dataSource := db.GetAtlasDB()
+	var dataSourceModel = new(models.Source)
+	dataSource := dataSourceModel.GetDataSource(GetTestSourceId(), sourceType)
 	count := 0
 	query := fmt.Sprintf("SELECT COUNT(*) FROM %s.concept WHERE concept_id = ?", GetSchemaNameForType(sourceType))
 	dataSource.Db.Raw(query, conceptId).Scan(&count)

Original file line number	Diff line number	Diff line change
`@@ -195,7 +195,8 @@ func RemoveConcept(sourceType models.SourceType, conceptId int64) {`
`195`	`195`	`}`
`196`	`196`
`197`	`197`	`func ConceptExists(sourceType models.SourceType, conceptId int64) bool {`
`198`		`- dataSource := db.GetAtlasDB()`
	`198`	`+ var dataSourceModel = new(models.Source)`
	`199`	`+ dataSource := dataSourceModel.GetDataSource(GetTestSourceId(), sourceType)`
`199`	`200`	`count := 0`
`200`	`201`	`query := fmt.Sprintf("SELECT COUNT(*) FROM %s.concept WHERE concept_id = ?", GetSchemaNameForType(sourceType))`
`201`	`202`	`dataSource.Db.Raw(query, conceptId).Scan(&count)`