@@ -35,18 +35,15 @@ import org.jboss.netty.channel.socket.nio.NioClientSocketChannelFactory
35
35
* limitations under the License.
36
36
*/
37
37
class SparkSinkSuite extends TestSuiteBase {
38
- val batchCount = 5
39
- val eventsPerBatch = 100
40
- val totalEventsPerChannel = batchCount * eventsPerBatch
38
+ val eventsPerBatch = 1000
41
39
val channelCapacity = 5000
42
- val maxAttempts = 5
43
40
44
41
test(" Success test" ) {
45
42
val (channel, sink) = initializeChannelAndSink(None )
46
43
channel.start()
47
44
sink.start()
48
45
49
- putEvents(channel, 1000 )
46
+ putEvents(channel, eventsPerBatch )
50
47
51
48
val port = sink.getPort
52
49
val address = new InetSocketAddress (" 0.0.0.0" , port)
@@ -65,7 +62,7 @@ class SparkSinkSuite extends TestSuiteBase {
65
62
val (channel, sink) = initializeChannelAndSink(None )
66
63
channel.start()
67
64
sink.start()
68
- putEvents(channel, 1000 )
65
+ putEvents(channel, eventsPerBatch )
69
66
70
67
val port = sink.getPort
71
68
val address = new InetSocketAddress (" 0.0.0.0" , port)
@@ -85,7 +82,7 @@ class SparkSinkSuite extends TestSuiteBase {
85
82
.CONF_TRANSACTION_TIMEOUT -> 1 .toString)))
86
83
channel.start()
87
84
sink.start()
88
- putEvents(channel, 1000 )
85
+ putEvents(channel, eventsPerBatch )
89
86
val port = sink.getPort
90
87
val address = new InetSocketAddress (" 0.0.0.0" , port)
91
88
@@ -112,7 +109,7 @@ class SparkSinkSuite extends TestSuiteBase {
112
109
val (channel, sink) = initializeChannelAndSink(None )
113
110
channel.start()
114
111
sink.start()
115
- (1 to 5 ).map(_ => putEvents(channel, 1000 ))
112
+ (1 to 5 ).map(_ => putEvents(channel, eventsPerBatch ))
116
113
val port = sink.getPort
117
114
val address = new InetSocketAddress (" 0.0.0.0" , port)
118
115
0 commit comments