diff --git a/test/connector.jl b/test/connector.jl
index 5de9feb..fac18bc 100644
--- a/test/connector.jl
+++ b/test/connector.jl
@@ -12,16 +12,21 @@
                 @test  batch[1, j] == df[i, j] 
             end
         end
-
+        
         @test EasyStream.hasnext(conn) == false
 
+        @test EasyStream.next(conn) === nothing
+
         @test length(conn) == size(df, 1)
 
         EasyStream.reset!(conn)
 
         @test conn.state == 0
-    
 
+        array = [1 2 3 4 5; 6 7 8 9 10]
+
+        @test_throws ArgumentError EasyStream.TablesConnector(array)
+        
         @testset "sort functionalities" begin
             
             conn = EasyStream.TablesConnector(df, :x)
@@ -58,4 +63,19 @@
             @test diff_elements > 0
         end
     end
+
+    @testset "Generator Connector" begin
+        function generator(n_features::Int)        
+            return DataFrame(permutedims([rand() for _ = 1:n_features ]))
+        end
+        
+        n_features = 5
+        conn = EasyStream.GeneratorConnector(generator, n_features)
+        
+        data = EasyStream.next(cnn)
+        
+        @test size(data, 1) == 1
+        @test size(data, 1) == n_features
+        
+    end
 end
\ No newline at end of file