embedded driver for scs and ncs

Author: man-zhang

jdk_8_maven/em/embedded/thrift/ncs/src/main/java/em/embedded/org/thriftncs/EmbeddedEvoMasterController.java

@@ -76,13 +76,13 @@ public Connection getConnection() {
     public ProblemInfo getProblemInfo() {
 
         return new RPCProblem(new HashMap<String, Object>() {{
-            put("org.thrift.ncs.NcsService$Iface", client);
+            put(NcsService.Iface.class.getName(), client);
         }});
     }
 
     @Override
     public SutInfoDto.OutputFormat getPreferredOutputFormat() {
-        return SutInfoDto.OutputFormat.JAVA_JUNIT_5;
+        return SutInfoDto.OutputFormat.JAVA_JUNIT_4;
     }
 
     @Override
@@ -103,6 +103,7 @@ public String startSut() {
             e.printStackTrace();
         }
 
+        // it is not necessary here for RPC
         return url;
     }
 

jdk_8_maven/em/embedded/thrift/pom.xml

@@ -14,6 +14,7 @@
 
     <modules>
         <module>ncs</module>
+        <module>scs</module>
     </modules>
 
 </project>

jdk_8_maven/em/embedded/thrift/scs/pom.xml

@@ -0,0 +1,23 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project xmlns="http://maven.apache.org/POM/4.0.0"
+         xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
+         xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
+    <parent>
+        <artifactId>evomaster-benchmark-em-embedded-thrift</artifactId>
+        <groupId>org.evomaster</groupId>
+        <version>1.3.1-SNAPSHOT</version>
+    </parent>
+    <modelVersion>4.0.0</modelVersion>
+
+    <artifactId>evomaster-benchmark-em-embedded-thrift-scs</artifactId>
+    <packaging>jar</packaging>
+
+
+    <dependencies>
+        <dependency>
+            <groupId>org.evomaster</groupId>
+            <artifactId>evomaster-benchmark-cs-rpc-thrift-artificial-scs</artifactId>
+            <version>${project.version}</version>
+        </dependency>
+    </dependencies>
+</project>

jdk_8_maven/em/embedded/thrift/scs/src/main/java/em/embedded/org/thriftscs/EmbeddedEvoMasterController.java

@@ -0,0 +1,124 @@
+package em.embedded.org.thriftscs;
+
+import org.apache.thrift.protocol.TBinaryProtocol;
+import org.apache.thrift.protocol.TProtocol;
+import org.apache.thrift.transport.THttpClient;
+import org.apache.thrift.transport.TTransport;
+import org.apache.thrift.transport.TTransportException;
+import org.evomaster.client.java.controller.EmbeddedSutController;
+import org.evomaster.client.java.controller.InstrumentedSutStarter;
+import org.evomaster.client.java.controller.api.dto.AuthenticationDto;
+import org.evomaster.client.java.controller.api.dto.SutInfoDto;
+import org.evomaster.client.java.controller.problem.ProblemInfo;
+import org.evomaster.client.java.controller.problem.RPCProblem;
+import org.springframework.boot.SpringApplication;
+import org.springframework.context.ConfigurableApplicationContext;
+import org.thrift.scs.ScsApplication;
+import org.thrift.scs.ScsService;
+
+import java.sql.Connection;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+
+/**
+ * created by manzhang on 2021/11/3
+ */
+public class EmbeddedEvoMasterController extends EmbeddedSutController {
+
+    public static void main(String[] args) {
+
+        int port = 40100;
+        if (args.length > 0) {
+            port = Integer.parseInt(args[0]);
+        }
+
+        EmbeddedEvoMasterController controller = new EmbeddedEvoMasterController(port);
+        InstrumentedSutStarter starter = new InstrumentedSutStarter(controller);
+
+        starter.start();
+    }
+
+    public EmbeddedEvoMasterController() {
+        this(40100);
+    }
+
+    public EmbeddedEvoMasterController(int port) {
+        setControllerPort(port);
+    }
+
+    private ConfigurableApplicationContext ctx;
+    private ScsService.Client client;
+
+    @Override
+    public boolean isSutRunning() {
+        return ctx != null && ctx.isRunning();
+    }
+
+
+    @Override
+    public String getPackagePrefixesToCover() {
+        return "org.thrift.scs";
+    }
+
+    @Override
+    public List<AuthenticationDto> getInfoForAuthentication() {
+        return null;
+    }
+
+    @Override
+    public Connection getConnection() {
+        return null;
+    }
+
+    @Override
+    public ProblemInfo getProblemInfo() {
+
+        return new RPCProblem(new HashMap<String, Object>() {{
+            put(ScsService.Iface.class.getName(), client);
+        }});
+    }
+
+    @Override
+    public SutInfoDto.OutputFormat getPreferredOutputFormat() {
+        return SutInfoDto.OutputFormat.JAVA_JUNIT_4;
+    }
+
+    @Override
+    public String startSut() {
+
+        ctx = SpringApplication.run(ScsApplication.class, new String[]{
+                "--server.port=0"
+        });
+
+        String url = "http://localhost:"+getSutPort()+"/scs";
+
+        try {
+            // init client
+            TTransport transport = new THttpClient(url);
+            TProtocol protocol = new TBinaryProtocol(transport);
+            client = new ScsService.Client(protocol);
+        } catch (TTransportException e) {
+            e.printStackTrace();
+        }
+
+        // it is not necessary here for RPC
+        return url;
+    }
+
+    protected int getSutPort() {
+        return (Integer) ((Map) ctx.getEnvironment()
+                .getPropertySources().get("server.ports").getSource())
+                .get("local.server.port");
+    }
+
+    @Override
+    public void stopSut() {
+        ctx.stop();
+    }
+
+    @Override
+    public void resetStateOfSUT() {
+
+    }
+}