feature:WEOS-1378

- Fixed e2e step
wepala · Mar 28, 2022 · 1c4f99e · 1c4f99e
1 parent 7905068
commit 1c4f99e
Showing 1 changed file with 11 additions and 4 deletions.
diff --git a/end2end_test.go b/end2end_test.go
@@ -1813,11 +1813,18 @@ func theFieldShouldHaveTodaysDate(field string) error {
 
 	switch dbconfig.Driver {
 	case "postgres", "mysql":
-		date := contentEntity[field].(time.Time).Format("2006-01-02")
-		if !strings.Contains(date, todaysDate) {
-			return fmt.Errorf("expected the %s date: %s to contain the current date: %s ", field, date, todaysDate)
+		switch contentEntity[field].(type) {
+		case *time.Time:
+			date := contentEntity[field].(*time.Time).Format("2006-01-02")
+			if !strings.Contains(date, todaysDate) {
+				return fmt.Errorf("expected the %s date: %s to contain the current date: %s ", field, date, todaysDate)
+			}
+		case time.Time:
+			date := contentEntity[field].(time.Time).Format("2006-01-02")
+			if !strings.Contains(date, todaysDate) {
+				return fmt.Errorf("expected the %s date: %s to contain the current date: %s ", field, date, todaysDate)
+			}
 		}
-
 	case "sqlite3":
 		if date, ok := contentEntity[field].(*time.Time); ok {
 			if !strings.Contains(date.Format("2006-01-02"), todaysDate) {