[Streaming] Test build fixed (#10617)

This commit is contained in:
Lixin Wei
2020-09-08 14:31:54 +08:00
committed by GitHub
parent ca8792e4ff
commit 2b95e71dac
12 changed files with 87 additions and 45 deletions
@@ -67,7 +67,8 @@ public class JobMaster {
runtimeContext = new JobMasterRuntimeContext(streamingConfig);
// load checkpoint if is recover
if (Ray.getRuntimeContext().wasCurrentActorRestarted()) {
if (!Ray.getRuntimeContext().isSingleProcess() && Ray.getRuntimeContext()
.wasCurrentActorRestarted()) {
loadMasterCheckpoint();
}
@@ -1,10 +1,25 @@
package io.ray.streaming.runtime.master;
import io.ray.api.Ray;
import io.ray.streaming.runtime.BaseUnitTest;
import java.util.HashMap;
import org.testng.Assert;
import org.testng.annotations.AfterMethod;
import org.testng.annotations.BeforeMethod;
import org.testng.annotations.Test;
public class JobMasterTest {
public class JobMasterTest extends BaseUnitTest {
@BeforeMethod
public void init() {
// ray init
Ray.init();
}
@AfterMethod
public void tearDown() {
Ray.shutdown();
}
@Test
public void testCreation() {