Example #1
0
 @Override
 protected void execute() {
   FloatingIpService service = get(FloatingIpService.class);
   try {
     FloatingIp floatingIpObj =
         new DefaultFloatingIp(
             FloatingIpId.of(id),
             TenantId.tenantId(tenantId),
             TenantNetworkId.networkId(networkId),
             VirtualPortId.portId(portId),
             RouterId.valueOf(routerId),
             floatingIp == null ? null : IpAddress.valueOf(floatingIp),
             fixedIp == null ? null : IpAddress.valueOf(fixedIp),
             status == null ? Status.ACTIVE : Status.valueOf(status));
     Set<FloatingIp> floatingIpSet = Sets.newHashSet(floatingIpObj);
     service.createFloatingIps(floatingIpSet);
   } catch (Exception e) {
     print(null, e.getMessage());
   }
 }
/** Unit tests for flow classifier REST APIs. */
public class FlowClassifierResourceTest extends VtnResourceTest {

  final FlowClassifierService flowClassifierService = createMock(FlowClassifierService.class);

  FlowClassifierId flowClassifierId1 = FlowClassifierId.of("4a334cd4-fe9c-4fae-af4b-321c5e2eb051");
  TenantId tenantId1 = TenantId.tenantId("1814726e2d22407b8ca76db5e567dcf1");
  VirtualPortId srcPortId1 = VirtualPortId.portId("dace4513-24fc-4fae-af4b-321c5e2eb3d1");
  VirtualPortId dstPortId1 = VirtualPortId.portId("aef3478a-4a56-2a6e-cd3a-9dee4e2ec345");

  final MockFlowClassifier flowClassifier1 =
      new MockFlowClassifier(
          flowClassifierId1,
          tenantId1,
          "flowClassifier1",
          "Mock flow classifier",
          "IPv4",
          "IP",
          1001,
          1500,
          5001,
          6000,
          IpPrefix.valueOf("1.1.1.1/16"),
          IpPrefix.valueOf("22.12.34.45/16"),
          srcPortId1,
          dstPortId1);

  /** Mock class for a flow classifier. */
  private static class MockFlowClassifier implements FlowClassifier {

    private final FlowClassifierId flowClassifierId;
    private final TenantId tenantId;
    private final String name;
    private final String description;
    private final String etherType;
    private final String protocol;
    private final int minSrcPortRange;
    private final int maxSrcPortRange;
    private final int minDstPortRange;
    private final int maxDstPortRange;
    private final IpPrefix srcIpPrefix;
    private final IpPrefix dstIpPrefix;
    private final VirtualPortId srcPort;
    private final VirtualPortId dstPort;

    public MockFlowClassifier(
        FlowClassifierId flowClassifierId,
        TenantId tenantId,
        String name,
        String description,
        String etherType,
        String protocol,
        int minSrcPortRange,
        int maxSrcPortRange,
        int minDstPortRange,
        int maxDstPortRange,
        IpPrefix srcIpPrefix,
        IpPrefix dstIpPrefix,
        VirtualPortId srcPort,
        VirtualPortId dstPort) {
      this.flowClassifierId = flowClassifierId;
      this.tenantId = tenantId;
      this.name = name;
      this.description = description;
      this.etherType = etherType;
      this.protocol = protocol;
      this.minSrcPortRange = minSrcPortRange;
      this.maxSrcPortRange = maxSrcPortRange;
      this.minDstPortRange = minDstPortRange;
      this.maxDstPortRange = maxDstPortRange;
      this.srcIpPrefix = srcIpPrefix;
      this.dstIpPrefix = dstIpPrefix;
      this.srcPort = srcPort;
      this.dstPort = dstPort;
    }

    @Override
    public FlowClassifierId flowClassifierId() {
      return flowClassifierId;
    }

    @Override
    public TenantId tenantId() {
      return tenantId;
    }

    @Override
    public String name() {
      return name;
    }

    @Override
    public String description() {
      return description;
    }

    @Override
    public String etherType() {
      return etherType;
    }

    @Override
    public String protocol() {
      return protocol;
    }

    @Override
    public int minSrcPortRange() {
      return minSrcPortRange;
    }

    @Override
    public int maxSrcPortRange() {
      return maxSrcPortRange;
    }

    @Override
    public int minDstPortRange() {
      return minDstPortRange;
    }

    @Override
    public int maxDstPortRange() {
      return maxDstPortRange;
    }

    @Override
    public IpPrefix srcIpPrefix() {
      return srcIpPrefix;
    }

    @Override
    public IpPrefix dstIpPrefix() {
      return dstIpPrefix;
    }

    @Override
    public VirtualPortId srcPort() {
      return srcPort;
    }

    @Override
    public VirtualPortId dstPort() {
      return dstPort;
    }

    @Override
    public boolean exactMatch(FlowClassifier flowClassifier) {
      return this.equals(flowClassifier)
          && Objects.equals(this.flowClassifierId, flowClassifier.flowClassifierId())
          && Objects.equals(this.tenantId, flowClassifier.tenantId());
    }
  }

  /** Sets up the global values for all the tests. */
  @Before
  public void setUpTest() {
    SfcCodecContext context = new SfcCodecContext();

    ServiceDirectory testDirectory =
        new TestServiceDirectory()
            .add(FlowClassifierService.class, flowClassifierService)
            .add(CodecService.class, context.codecManager());
    BaseResource.setServiceDirectory(testDirectory);
  }

  /** Cleans up. */
  @After
  public void tearDownTest() {}

  /** Tests the result of the rest api GET when there are no flow classifiers. */
  @Test
  public void testFlowClassifiersEmpty() {

    expect(flowClassifierService.getFlowClassifiers()).andReturn(null).anyTimes();
    replay(flowClassifierService);
    final WebResource rs = resource();
    final String response = rs.path("flow_classifiers").get(String.class);
    assertThat(response, is("{\"flow_classifiers\":[]}"));
  }

  /** Tests the result of a rest api GET for flow classifier id. */
  @Test
  public void testGetFlowClassifierId() {

    final Set<FlowClassifier> flowClassifiers = new HashSet<>();
    flowClassifiers.add(flowClassifier1);

    expect(flowClassifierService.exists(anyObject())).andReturn(true).anyTimes();
    expect(flowClassifierService.getFlowClassifier(anyObject()))
        .andReturn(flowClassifier1)
        .anyTimes();
    replay(flowClassifierService);

    final WebResource rs = resource();
    final String response =
        rs.path("flow_classifiers/4a334cd4-fe9c-4fae-af4b-321c5e2eb051").get(String.class);
    final JsonObject result = JsonObject.readFrom(response);
    assertThat(result, notNullValue());
  }

  /** Tests that a fetch of a non-existent flow classifier object throws an exception. */
  @Test
  public void testBadGet() {
    expect(flowClassifierService.getFlowClassifier(anyObject())).andReturn(null).anyTimes();
    replay(flowClassifierService);
    WebResource rs = resource();
    try {
      rs.path("flow_classifiers/78dcd363-fc23-aeb6-f44b-56dc5aafb3ae").get(String.class);
      fail("Fetch of non-existent flow classifier did not throw an exception");
    } catch (UniformInterfaceException ex) {
      assertThat(ex.getMessage(), containsString("returned a response status of"));
    }
  }

  /** Tests creating a flow classifier with POST. */
  @Test
  public void testPost() {

    expect(flowClassifierService.createFlowClassifier(anyObject())).andReturn(true).anyTimes();
    replay(flowClassifierService);

    WebResource rs = resource();
    InputStream jsonStream =
        FlowClassifierResourceTest.class.getResourceAsStream("post-FlowClassifier.json");

    ClientResponse response =
        rs.path("flow_classifiers")
            .type(MediaType.APPLICATION_JSON_TYPE)
            .post(ClientResponse.class, jsonStream);
    assertThat(response.getStatus(), is(HttpURLConnection.HTTP_OK));
  }

  /** Tests deleting a flow classifier. */
  @Test
  public void testDelete() {
    expect(flowClassifierService.removeFlowClassifier(anyObject())).andReturn(true).anyTimes();
    replay(flowClassifierService);

    WebResource rs = resource();

    String location = "flow_classifiers/4a334cd4-fe9c-4fae-af4b-321c5e2eb051";

    ClientResponse deleteResponse =
        rs.path(location).type(MediaType.APPLICATION_JSON_TYPE).delete(ClientResponse.class);
    assertThat(deleteResponse.getStatus(), is(HttpURLConnection.HTTP_NO_CONTENT));
  }
}
 @Override
 protected void execute() {
   VirtualPortService service = get(VirtualPortService.class);
   Set<VirtualPortId> virtualPorts = Sets.newHashSet(VirtualPortId.portId(id));
   service.removePorts(virtualPorts);
 }