@Test(dependsOnMethods = "testCreatePortForwardingRule") public void testCreateFirewallRule() { if (networksDisabled) return; AsyncCreateResponse job = client.getFirewallApi().createFirewallRuleForIpAndProtocol( ip.getId(), FirewallRule.Protocol.TCP, CreateFirewallRuleOptions.Builder.startPort(30).endPort(35)); assertTrue(jobComplete.apply(job.getJobId())); firewallRule = client.getFirewallApi().getFirewallRule(job.getId()); assertEquals(firewallRule.getStartPort(), 30); assertEquals(firewallRule.getEndPort(), 35); assertEquals(firewallRule.getProtocol(), FirewallRule.Protocol.TCP); checkFirewallRule(firewallRule); }
protected void checkFirewallRule(FirewallRule rule) { assertEquals(rule, client.getFirewallApi().getFirewallRule(rule.getId())); assert rule.getId() != null : rule; assert rule.getStartPort() > 0 : rule; assert rule.getEndPort() >= rule.getStartPort() : rule; assert rule.getProtocol() != null; }
public void testGetFirewallRuleWhenResponseIs404() { FirewallApi client = requestSendsResponse( HttpRequest.builder() .method("GET") .endpoint("http://localhost:8080/client/api") .addQueryParam("response", "json") .addQueryParam("command", "listFirewallRules") .addQueryParam("listAll", "true") .addQueryParam("id", "4") .addQueryParam("apiKey", "identity") .addQueryParam("signature", "rYd8gr7ptdSZlIehBEMQEKsm07Q=") .addHeader("Accept", "application/json") .build(), HttpResponse.builder() .statusCode(404) .build()); assertNull(client.getFirewallRule("4")); }
public void testGetFirewallRuleWhenResponseIs2xx() { FirewallApi client = requestSendsResponse( HttpRequest.builder() .method("GET") .endpoint("http://localhost:8080/client/api") .addQueryParam("response", "json") .addQueryParam("command", "listFirewallRules") .addQueryParam("listAll", "true") .addQueryParam("id", "2017") .addQueryParam("apiKey", "identity") .addQueryParam("signature", "6coh9Qdwla94TN1Dl008WlhzZUY=") .addHeader("Accept", "application/json") .build(), HttpResponse.builder() .statusCode(200) .payload(payloadFromResource("/getfirewallrulesresponse.json")) .build()); assertEquals(client.getFirewallRule("2017"), FirewallRule.builder().id("2017").protocol(FirewallRule.Protocol.TCP).startPort(30) .endPort(35).ipAddressId("2").ipAddress("10.27.27.51").state(FirewallRule.State.ACTIVE) .CIDRs(ImmutableSet.of("0.0.0.0/0")).build() ); }