Merge pull request #2460 from fjl/whisper-expiration-test-delay

whisper: deflake Test*MessageExpiration
This commit is contained in:
Péter Szilágyi 2016-04-15 14:44:36 +03:00
commit 5c17b2f521
2 changed files with 5 additions and 9 deletions

View File

@ -255,7 +255,7 @@ func (self *Whisper) add(envelope *Envelope) error {
defer self.poolMu.Unlock() defer self.poolMu.Unlock()
// short circuit when a received envelope has already expired // short circuit when a received envelope has already expired
if envelope.Expiry <= uint32(time.Now().Unix()) { if envelope.Expiry < uint32(time.Now().Unix()) {
return nil return nil
} }
@ -278,7 +278,6 @@ func (self *Whisper) add(envelope *Envelope) error {
go self.postEvent(envelope) go self.postEvent(envelope)
} }
glog.V(logger.Detail).Infof("cached whisper envelope %x\n", envelope) glog.V(logger.Detail).Infof("cached whisper envelope %x\n", envelope)
return nil return nil
} }

View File

@ -179,9 +179,7 @@ func TestMessageExpiration(t *testing.T) {
node := startTestCluster(1)[0] node := startTestCluster(1)[0]
message := NewMessage([]byte("expiring message")) message := NewMessage([]byte("expiring message"))
envelope, err := message.Wrap(DefaultPoW, Options{ envelope, err := message.Wrap(DefaultPoW, Options{TTL: time.Second})
TTL: time.Second,
})
if err != nil { if err != nil {
t.Fatalf("failed to wrap message: %v", err) t.Fatalf("failed to wrap message: %v", err)
} }
@ -197,17 +195,17 @@ func TestMessageExpiration(t *testing.T) {
t.Fatalf("message not found in cache") t.Fatalf("message not found in cache")
} }
// Wait for expiration and check cache again // Wait for expiration and check cache again
time.Sleep(time.Second) // wait for expiration time.Sleep(time.Second) // wait for expiration
time.Sleep(expirationCycle) // wait for cleanup cycle time.Sleep(2 * expirationCycle) // wait for cleanup cycle
node.poolMu.RLock() node.poolMu.RLock()
_, found = node.messages[envelope.Hash()] _, found = node.messages[envelope.Hash()]
node.poolMu.RUnlock() node.poolMu.RUnlock()
if found { if found {
t.Fatalf("message not expired from cache") t.Fatalf("message not expired from cache")
} }
// Check that adding an expired envelope doesn't do anything.
node.add(envelope) node.add(envelope)
node.poolMu.RLock() node.poolMu.RLock()
_, found = node.messages[envelope.Hash()] _, found = node.messages[envelope.Hash()]
@ -215,5 +213,4 @@ func TestMessageExpiration(t *testing.T) {
if found { if found {
t.Fatalf("message was added to cache") t.Fatalf("message was added to cache")
} }
} }