More tests

modules/testsuite/shared-rest-tests/src/test/java/org/jboss/test/jaxrs/examples/ex12_2/CustomerResourceTest.java

@@ -27,8 +27,6 @@
 import javax.ws.rs.client.Client;
 import javax.ws.rs.client.ClientBuilder;
 import javax.ws.rs.client.Entity;
-import javax.ws.rs.core.CacheControl;
-import javax.ws.rs.core.HttpHeaders;
 import javax.ws.rs.core.Response;
 
 import org.jboss.arquillian.container.test.api.Deployment;
@@ -87,10 +85,8 @@ public void testCustomerResource() throws Exception
                  .request().post(Entity.xml(newCustomer));
          if (response.getStatus() != 201) throw new RuntimeException("Failed to create");
          String location = response.getLocation().toString();
-         System.out.println("Location: " + location);
          response.close();
 
-         System.out.println("*** GET Created Customer **");
          response = client.target(location).request().get();
          String md5 = response.getHeaderString("Content-MD5");
          Assert.assertEquals("uPYTHd3xAeXEk8NMp7A6hw==", md5);

modules/testsuite/shared-rest-tests/src/test/java/org/jboss/test/jaxrs/examples/ex15_1/CustomerResourceTest.java

@@ -0,0 +1,133 @@
+/*
+ * JBoss, Home of Professional Open Source.
+ * Copyright 2015, Red Hat Middleware LLC, and individual contributors
+ * as indicated by the @author tags. See the copyright.txt file in the
+ * distribution for a full listing of individual contributors.
+ *
+ * This is free software; you can redistribute it and/or modify it
+ * under the terms of the GNU Lesser General Public License as
+ * published by the Free Software Foundation; either version 2.1 of
+ * the License, or (at your option) any later version.
+ *
+ * This software is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+ * Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public
+ * License along with this software; if not, write to the Free
+ * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
+ * 02110-1301 USA, or see the FSF site: http://www.fsf.org.
+ */
+package org.jboss.test.jaxrs.examples.ex15_1;
+
+import java.io.File;
+import java.net.URL;
+
+import javax.ws.rs.NotAuthorizedException;
+import javax.ws.rs.client.Client;
+import javax.ws.rs.client.ClientBuilder;
+import javax.ws.rs.client.Entity;
+import javax.ws.rs.client.WebTarget;
+import javax.ws.rs.core.Response;
+
+import org.jboss.arquillian.container.test.api.Deployment;
+import org.jboss.arquillian.container.test.api.RunAsClient;
+import org.jboss.arquillian.junit.Arquillian;
+import org.jboss.arquillian.test.api.ArquillianResource;
+import org.jboss.shrinkwrap.api.ShrinkWrap;
+import org.jboss.shrinkwrap.api.asset.StringAsset;
+import org.jboss.shrinkwrap.api.spec.WebArchive;
+import org.jboss.test.jaxrs.examples.ex10_2.domain.Customer;
+import org.jboss.test.jaxrs.examples.ex15_1.features.OneTimePasswordGenerator;
+import org.jboss.wsf.test.JBossWSTest;
+import org.jboss.wsf.test.JBossWSTestHelper;
+import org.junit.Assert;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+
+/**
+ * @author <a href="mailto:bill@burkecentral.com">Bill Burke</a>
+ * @author <a href="mailto:alessio.soldano@jboss.com">Alessio Soldano</a>
+ */
+@RunWith(Arquillian.class)
+public class CustomerResourceTest extends JBossWSTest
+{
+   @ArquillianResource
+   private URL baseURL;
+
+   @Deployment(testable = false)
+   public static WebArchive createDeployments() {
+      WebArchive archive = ShrinkWrap.create(WebArchive.class, "jaxrs-examples-ex15_1.war");
+         archive
+               .setManifest(new StringAsset("Manifest-Version: 1.0\n" + "Dependencies: org.jboss.ws.common\n"))
+               .addClass(org.jboss.test.jaxrs.examples.ex15_1.domain.Customer.class)
+               .addClass(org.jboss.test.jaxrs.examples.ex15_1.features.AllowedPerDay.class)
+               .addClass(org.jboss.test.jaxrs.examples.ex15_1.features.OneTimePasswordAuthenticator.class)
+               .addClass(org.jboss.test.jaxrs.examples.ex15_1.features.OTPAuthenticated.class)
+               .addClass(org.jboss.test.jaxrs.examples.ex15_1.features.PerDayAuthorizer.class)
+               .addClass(org.jboss.test.jaxrs.examples.ex15_1.features.OneTimePasswordGenerator.class)
+               .addClass(org.jboss.test.jaxrs.examples.ex15_1.features.OTP.class)
+               .addClass(org.jboss.test.jaxrs.examples.ex15_1.services.CustomerResource.class)
+               .addClass(org.jboss.test.jaxrs.examples.ex15_1.services.ShoppingApplication.class)
+               .setWebXML(new File(JBossWSTestHelper.getTestResourcesDir() + "/jaxrs/examples/ex15_x/WEB-INF/web.xml"));
+      return archive;
+   }
+
+   @Test
+   @RunAsClient
+   public void testCustomerResource() throws Exception
+   {
+      Client client = ClientBuilder.newClient();
+      try {
+         Customer newCustomer = new Customer();
+         newCustomer.setFirstName("Bill");
+         newCustomer.setLastName("Burke");
+         newCustomer.setStreet("256 Clarendon Street");
+         newCustomer.setCity("Boston");
+         newCustomer.setState("MA");
+         newCustomer.setZip("02115");
+         newCustomer.setCountry("USA");
+
+         Response response = client.target(baseURL + "services/customers")
+                 .request().post(Entity.xml(newCustomer));
+         if (response.getStatus() != 201) throw new RuntimeException("Failed to create");
+         String location = response.getLocation().toString();
+         response.close();
+
+         Customer customer = null;
+         WebTarget target = client.target(location);
+         try
+         {
+            customer = target.request().get(Customer.class);
+            Assert.fail(); // should have thrown an exception
+         }
+         catch (NotAuthorizedException e)
+         {
+            //OK
+         }
+
+         target.register(new OneTimePasswordGenerator("bburke", "geheim"));
+
+         customer = target.request().get(Customer.class);
+         Assert.assertEquals("Bill", customer.getFirstName());
+
+         customer.setFirstName("William");
+         response = target.request().put(Entity.xml(customer));
+         if (response.getStatus() != 204) throw new RuntimeException("Failed to update");
+
+         // Show the update
+         customer = target.request().get(Customer.class);
+         Assert.assertEquals("William", customer.getFirstName());
+
+         // only allowed to update once per day
+         customer.setFirstName("Bill");
+         response = target.request().put(Entity.xml(customer));
+         Assert.assertEquals(Response.Status.FORBIDDEN.getStatusCode(), response.getStatusInfo().getStatusCode());
+         Assert.assertEquals(Response.Status.FORBIDDEN.getReasonPhrase(), response.getStatusInfo().getReasonPhrase());
+
+      } finally {
+         client.close();
+      }
+   }
+}

modules/testsuite/shared-rest-tests/src/test/java/org/jboss/test/jaxrs/examples/ex15_1/domain/Customer.java

@@ -0,0 +1,142 @@
+/*
+ * JBoss, Home of Professional Open Source.
+ * Copyright 2015, Red Hat Middleware LLC, and individual contributors
+ * as indicated by the @author tags. See the copyright.txt file in the
+ * distribution for a full listing of individual contributors.
+ *
+ * This is free software; you can redistribute it and/or modify it
+ * under the terms of the GNU Lesser General Public License as
+ * published by the Free Software Foundation; either version 2.1 of
+ * the License, or (at your option) any later version.
+ *
+ * This software is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+ * Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public
+ * License along with this software; if not, write to the Free
+ * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
+ * 02110-1301 USA, or see the FSF site: http://www.fsf.org.
+ */
+package org.jboss.test.jaxrs.examples.ex15_1.domain;
+
+import javax.xml.bind.annotation.XmlAttribute;
+import javax.xml.bind.annotation.XmlElement;
+import javax.xml.bind.annotation.XmlRootElement;
+
+@XmlRootElement(name = "customer")
+public class Customer
+{
+   private int id;
+   private String firstName;
+   private String lastName;
+   private String street;
+   private String city;
+   private String state;
+   private String zip;
+   private String country;
+
+   @XmlAttribute
+   public int getId()
+   {
+      return id;
+   }
+
+   public void setId(int id)
+   {
+      this.id = id;
+   }
+
+   @XmlElement(name = "first-name")
+   public String getFirstName()
+   {
+      return firstName;
+   }
+
+   public void setFirstName(String firstName)
+   {
+      this.firstName = firstName;
+   }
+
+   @XmlElement(name = "last-name")
+   public String getLastName()
+   {
+      return lastName;
+   }
+
+   public void setLastName(String lastName)
+   {
+      this.lastName = lastName;
+   }
+
+   @XmlElement
+   public String getStreet()
+   {
+      return street;
+   }
+
+   public void setStreet(String street)
+   {
+      this.street = street;
+   }
+
+   @XmlElement
+   public String getCity()
+   {
+      return city;
+   }
+
+   public void setCity(String city)
+   {
+      this.city = city;
+   }
+
+   @XmlElement
+   public String getState()
+   {
+      return state;
+   }
+
+   public void setState(String state)
+   {
+      this.state = state;
+   }
+
+   @XmlElement
+   public String getZip()
+   {
+      return zip;
+   }
+
+   public void setZip(String zip)
+   {
+      this.zip = zip;
+   }
+
+   @XmlElement
+   public String getCountry()
+   {
+      return country;
+   }
+
+   public void setCountry(String country)
+   {
+      this.country = country;
+   }
+
+   @Override
+   public String toString()
+   {
+      return "Customer{" +
+              "id=" + id +
+              ", firstName='" + firstName + '\'' +
+              ", lastName='" + lastName + '\'' +
+              ", street='" + street + '\'' +
+              ", city='" + city + '\'' +
+              ", state='" + state + '\'' +
+              ", zip='" + zip + '\'' +
+              ", country='" + country + '\'' +
+              '}';
+   }
+}

modules/testsuite/shared-rest-tests/src/test/java/org/jboss/test/jaxrs/examples/ex15_1/features/AllowedPerDay.java

@@ -0,0 +1,39 @@
+/*
+ * JBoss, Home of Professional Open Source.
+ * Copyright 2015, Red Hat Middleware LLC, and individual contributors
+ * as indicated by the @author tags. See the copyright.txt file in the
+ * distribution for a full listing of individual contributors.
+ *
+ * This is free software; you can redistribute it and/or modify it
+ * under the terms of the GNU Lesser General Public License as
+ * published by the Free Software Foundation; either version 2.1 of
+ * the License, or (at your option) any later version.
+ *
+ * This software is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+ * Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public
+ * License along with this software; if not, write to the Free
+ * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
+ * 02110-1301 USA, or see the FSF site: http://www.fsf.org.
+ */
+package org.jboss.test.jaxrs.examples.ex15_1.features;
+
+import javax.ws.rs.NameBinding;
+import java.lang.annotation.ElementType;
+import java.lang.annotation.Retention;
+import java.lang.annotation.RetentionPolicy;
+import java.lang.annotation.Target;
+
+/**
+ * @author <a href="mailto:bill@burkecentral.com">Bill Burke</a>
+ */
+@Target({ElementType.METHOD, ElementType.TYPE})
+@Retention(RetentionPolicy.RUNTIME)
+@NameBinding
+public @interface AllowedPerDay
+{
+   int value();
+}

modules/testsuite/shared-rest-tests/src/test/java/org/jboss/test/jaxrs/examples/ex15_1/features/OTP.java

@@ -0,0 +1,51 @@
+/*
+ * JBoss, Home of Professional Open Source.
+ * Copyright 2015, Red Hat Middleware LLC, and individual contributors
+ * as indicated by the @author tags. See the copyright.txt file in the
+ * distribution for a full listing of individual contributors.
+ *
+ * This is free software; you can redistribute it and/or modify it
+ * under the terms of the GNU Lesser General Public License as
+ * published by the Free Software Foundation; either version 2.1 of
+ * the License, or (at your option) any later version.
+ *
+ * This software is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+ * Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public
+ * License along with this software; if not, write to the Free
+ * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
+ * 02110-1301 USA, or see the FSF site: http://www.fsf.org.
+ */
+package org.jboss.test.jaxrs.examples.ex15_1.features;
+
+import java.nio.charset.Charset;
+import java.security.MessageDigest;
+import java.security.NoSuchAlgorithmException;
+
+import org.jboss.ws.common.utils.Base64;
+
+/**
+ * @author <a href="mailto:bill@burkecentral.com">Bill Burke</a>
+ */
+public class OTP
+{
+   public static String generateToken(String secret)
+   {
+      long minutes = System.currentTimeMillis() / 1000 / 60;
+      String concat = secret + minutes;
+      MessageDigest digest = null;
+      try
+      {
+         digest = MessageDigest.getInstance("MD5");
+      }
+      catch (NoSuchAlgorithmException e)
+      {
+         throw new IllegalArgumentException(e);
+      }
+      byte[] hash = digest.digest(concat.getBytes(Charset.forName("UTF-8")));
+      return Base64.encodeBytes(hash);
+   }
+}