@@ -25,52 +25,52 @@ import org.junit.Test
25
25
class WorkerEnvironmentTest {
26
26
27
27
@Test
28
- fun send () {
29
- val give = ( 1 .. 5 ).map { WorkRequest .newBuilder().setRequestId(it ).addArguments(" foo" ).build() }
28
+ fun sendOne () {
29
+ val give = WorkRequest .newBuilder().setRequestId(1 ).addArguments(" foo" ).build()
30
30
val got = WorkerEnvironment .inProcess {
31
31
task { stdIn, stdOut ->
32
32
generateSequence {
33
- WorkRequest .parseDelimitedFrom(stdIn)
33
+ WorkRequest .parseDelimitedFrom(stdIn). also { println ( " got $it " ) }
34
34
}.forEach { req ->
35
35
WorkerProtocol .WorkResponse .newBuilder().setRequestId(req.requestId).build()
36
+ .also { println (" sent $it " ) }
36
37
.writeDelimitedTo(stdOut)
37
38
}
38
39
}
39
40
40
- give.forEach { writeStdIn(it) }
41
+ writeStdIn(give)
41
42
closeStdIn()
42
43
43
- return @inProcess generateSequence { readStdOut() }.toSet ()
44
+ return @inProcess readStdOut()
44
45
}
45
46
46
- assertThat(got).containsExactlyElementsIn(
47
- (1 .. 5 ).map { id ->
48
- WorkerProtocol .WorkResponse .newBuilder().setRequestId(id).build()
49
- }
50
- )
47
+ assertThat(got)
48
+ .isEqualTo(WorkerProtocol .WorkResponse .newBuilder().setRequestId(1 ).build())
51
49
}
52
50
53
51
@Test
54
- fun sendOne () {
55
- val give = WorkRequest .newBuilder().setRequestId(1 ).addArguments(" foo" ).build()
52
+ fun send () {
53
+ val give = ( 1 .. 5 ).map { WorkRequest .newBuilder().setRequestId(it ).addArguments(" foo" ).build() }
56
54
val got = WorkerEnvironment .inProcess {
57
55
task { stdIn, stdOut ->
58
56
generateSequence {
59
- WorkRequest .parseDelimitedFrom(stdIn). also { println ( " got $it " ) }
57
+ WorkRequest .parseDelimitedFrom(stdIn)
60
58
}.forEach { req ->
61
59
WorkerProtocol .WorkResponse .newBuilder().setRequestId(req.requestId).build()
62
- .also { println (" sent $it " ) }
63
60
.writeDelimitedTo(stdOut)
64
61
}
65
62
}
66
63
67
- writeStdIn(give)
64
+ give.forEach { writeStdIn(it) }
68
65
closeStdIn()
69
66
70
- return @inProcess readStdOut()
67
+ return @inProcess generateSequence { readStdOut() }.toSet ()
71
68
}
72
69
73
- assertThat(got)
74
- .isEqualTo(WorkerProtocol .WorkResponse .newBuilder().setRequestId(1 ).build())
70
+ assertThat(got).containsExactlyElementsIn(
71
+ (1 .. 5 ).map { id ->
72
+ WorkerProtocol .WorkResponse .newBuilder().setRequestId(id).build()
73
+ }
74
+ )
75
75
}
76
76
}
0 commit comments