Only switch off pushback once we know that we are not pushing anything else

Signed-off-by: Andrew Thornton <[email protected]>

Author: zeripath

modules/queue/queue_disk_channel_test.go

@@ -207,7 +207,6 @@ func TestPersistableChannelQueue_Pause(t *testing.T) {
 				log.Info("pausing")
 				pausable.Pause()
 			}
-			pushBack = false
 			lock.Unlock()
 			return data
 		}
@@ -333,6 +332,10 @@ func TestPersistableChannelQueue_Pause(t *testing.T) {
 		return
 	}
 
+	lock.Lock()
+	pushBack = false
+	lock.Unlock()
+
 	pausable.Resume()
 
 	_, resumed = pausable.IsPausedIsResumed()
@@ -451,6 +454,10 @@ func TestPersistableChannelQueue_Pause(t *testing.T) {
 	default:
 	}
 
+	lock.Lock()
+	pushBack = false
+	lock.Unlock()
+
 	pausable.Resume()
 	_, resumed = pausable.IsPausedIsResumed()
 	select {