Adding check for short flag in the longer running tests

scheduler/scheduler_test.go

@@ -13,6 +13,11 @@ const cycleConfigFile = "test/cycle_test.yml"
 const expectedColletion = "a-collection"
 
 func TestSchedulerShouldStartWhenEnabled(t *testing.T) {
+	if testing.Short() {
+		t.Skip("Skipping - this test can take several seconds.")
+		return
+	}
+
 	db := new(native.MockDB)
 	s := NewScheduler(db, &tasks.MockTask{}, &MockMetadataRW{}, 1*time.Minute)
 
@@ -39,6 +44,10 @@ func TestSchedulerShouldStartWhenEnabled(t *testing.T) {
 }
 
 func TestSchedulerDoNotStartWhenDisabled(t *testing.T) {
+	if testing.Short() {
+		t.Skip("Skipping - this test can take several seconds.")
+		return
+	}
 
 	db := new(native.MockDB)
 	s := NewScheduler(db, &tasks.MockTask{}, &MockMetadataRW{}, 1*time.Minute)

scheduler/throttle_test.go

@@ -8,6 +8,11 @@ import (
 )
 
 func TestDynamicThrottle(t *testing.T) {
+	if testing.Short() {
+		t.Skip("Skipping - this test can take several seconds.")
+		return
+	}
+
 	throttle, _ := NewDynamicThrottle(1*time.Second, 1*time.Second, 1, 1)
 	start := time.Now()
 	throttle.Queue()
@@ -25,6 +30,11 @@ func TestDynamicThrottle(t *testing.T) {
 }
 
 func TestCappedThrottle(t *testing.T) {
+	if testing.Short() {
+		t.Skip("Skipping - this test can take several seconds.")
+		return
+	}
+
 	throttle, _ := NewCappedDynamicThrottle(time.Minute, time.Millisecond, 100*time.Millisecond, 1, 1)
 	start := time.Now()
 	throttle.Queue()

scheduler/throttled_whole_collection_cycle.go

@@ -27,7 +27,6 @@ func (l *ThrottledWholeCollectionCycle) Start() {
 }
 
 func (l *ThrottledWholeCollectionCycle) start(ctx context.Context) {
-
 	skip := l.PublishedItems()
 
 	b := true

scheduler/throttled_whole_collection_cycle_test.go

@@ -12,6 +12,11 @@ import (
 )
 
 func TestWholeCollectionCycleRun(t *testing.T) {
+	if testing.Short() {
+		t.Skip("Skipping - this test can take several seconds.")
+		return
+	}
+
 	db := new(native.MockDB)
 	mockTx := new(native.MockTX)
 	iter := new(native.MockDBIter)
@@ -49,6 +54,11 @@ func TestWholeCollectionCycleRun(t *testing.T) {
 }
 
 func TestWholeCollectionCycleRunMongoDBConnectionError(t *testing.T) {
+	if testing.Short() {
+		t.Skip("Skipping - this test can take several seconds.")
+		return
+	}
+
 	db := new(native.MockDB)
 	mockTx := new(native.MockTX)
 	db.On("Open").Return(mockTx, errors.New("error in DB connection")).After(1 * time.Second)
@@ -75,6 +85,11 @@ func TestWholeCollectionCycleRunMongoDBConnectionError(t *testing.T) {
 }
 
 func TestWholeCollectionCycleRunEmptyCollection(t *testing.T) {
+	if testing.Short() {
+		t.Skip("Skipping - this test can take several seconds.")
+		return
+	}
+
 	db := new(native.MockDB)
 	mockTx := new(native.MockTX)
 	iter := new(native.MockDBIter)