From b4c677b36e0a30ddb8768ba7265371f7b5c22a8e Mon Sep 17 00:00:00 2001
From: Takafumi Arakaki <aka.tkf@gmail.com>
Date: Tue, 28 Sep 2021 21:20:02 -0400
Subject: [PATCH] Add more tests

---
 src/ssqueue.jl                                     |  2 +-
 .../ConcurrentCollectionsTests/src/test_ssqueue.jl | 14 +++++++++++++-
 2 files changed, 14 insertions(+), 2 deletions(-)

diff --git a/src/ssqueue.jl b/src/ssqueue.jl
index 13e37b4..5a7a404 100644
--- a/src/ssqueue.jl
+++ b/src/ssqueue.jl
@@ -191,7 +191,7 @@ end
 
 Base.IteratorSize(::Type{<:NodeIterator}) = Base.SizeUnknown()
 
-Base.eltype(::Type{NodeIterator{<:DualLinkedQueue{T}}}) where {T} = SSQNode{T}
+Base.eltype(::Type{NodeIterator{DualLinkedQueue{T}}}) where {T} = SSQNode{T}
 
 function Base.iterate(
     iter::NodeIterator{DualLinkedQueue{T}},
diff --git a/test/ConcurrentCollectionsTests/src/test_ssqueue.jl b/test/ConcurrentCollectionsTests/src/test_ssqueue.jl
index 45d8f54..b35e344 100644
--- a/test/ConcurrentCollectionsTests/src/test_ssqueue.jl
+++ b/test/ConcurrentCollectionsTests/src/test_ssqueue.jl
@@ -1,7 +1,7 @@
 module TestSSQueue
 
 using ConcurrentCollections
-using ConcurrentCollections.Implementations: NodeIterator, check_invariance
+using ConcurrentCollections.Implementations: NodeIterator, SSQNode, check_invariance
 using ProgressLogging: @logprogress, @withprogress
 using Test
 using ..TestDLCRQ: check_concurrent_push_pop!
@@ -45,4 +45,16 @@ function test_concurrent_push_pop(ntrials = 100)
     end
 end
 
+function test_nodeiterator()
+    q = DualLinkedQueue{Int}()
+    push!(q, 111)
+    push!(q, 222)
+    itr = NodeIterator(q)
+    @test eltype(itr) === SSQNode{Int}
+    nodes = collect(itr)
+    @test nodes[1].value == 111
+    @test nodes[2].value == 222
+    @test check_invariance(q)
+end
+
 end  # module