refactoring

Author: tomhrr

src/main/java/net/apnic/rdap/conformance/attributetest/Autnum.java

@@ -0,0 +1,124 @@
+package net.apnic.rdap.conformance.attributetest;
+
+import java.math.BigInteger;
+import java.math.BigDecimal;
+import java.util.Arrays;
+import java.util.HashSet;
+import java.util.Set;
+import java.util.Map;
+import java.util.List;
+
+import com.google.common.collect.Sets;
+
+import net.apnic.rdap.conformance.Result;
+import net.apnic.rdap.conformance.Utils;
+import net.apnic.rdap.conformance.Context;
+import net.apnic.rdap.conformance.ObjectTest;
+import net.apnic.rdap.conformance.AttributeTest;
+
+/**
+ * <p>Autnum class.</p>
+ *
+ * @author Tom Harrison <[email protected]>
+ * @version 0.2
+ */
+public final class Autnum implements AttributeTest {
+    private String autnum = null;
+    private Set<String> knownAttributes = new HashSet<String>();
+
+    /**
+     * <p>Constructor for Autnum.</p>
+     */
+    public Autnum() { }
+
+    /**
+     * <p>Constructor for Autnum.</p>
+     *
+     * @param autnum a {@link java.lang.String} object.
+     */
+    public Autnum(final String autnum) {
+        this.autnum = autnum;
+    }
+
+    private BigInteger processAutnum(final Context context,
+                                     final Result proto,
+                                     final Map<String, Object> data,
+                                     final String key) {
+        Result asnres = new Result(proto);
+        Object asnObj = data.get(key);
+        if (asnObj == null) {
+            asnres.setStatus(Result.Status.Failure);
+            asnres.addNode(key);
+            asnres.setInfo("not present");
+            context.addResult(asnres);
+            return null;
+        } else {
+            asnres.setStatus(Result.Status.Success);
+            asnres.addNode(key);
+            asnres.setInfo("present");
+            context.addResult(asnres);
+        }
+
+        Result asn2res = new Result(proto);
+        asn2res.setStatus(Result.Status.Success);
+        asn2res.addNode(key);
+        asn2res.setInfo("valid");
+        BigInteger asn = null;
+        try {
+            asn = BigDecimal.valueOf((Double) data.get(key))
+                            .toBigIntegerExact();
+        } catch (Exception e) {
+            asn2res.setStatus(Result.Status.Failure);
+            asn2res.setInfo("invalid: " + e.toString());
+            context.addResult(asn2res);
+            return null;
+        }
+        context.addResult(asn2res);
+
+        return asn;
+    }
+
+    /** {@inheritDoc} */
+    public boolean run(final Context context, final Result proto,
+                       final Map<String, Object> data) {
+        BigInteger startAddress =
+            processAutnum(context, proto, data, "startAutnum");
+        BigInteger endAddress =
+            processAutnum(context, proto, data, "endAutnum");
+
+        if ((startAddress != null) && (endAddress != null)) {
+            Result r = new Result(proto);
+            r.addNode("startAutnum");
+            r.setStatus(Result.Status.Success);
+            r.setInfo("startAutnum less then or equal to endAutnum");
+            if (startAddress.compareTo(endAddress) > 0) {
+                r.setStatus(Result.Status.Failure);
+                r.setInfo("startAutnum more than endAutnum");
+            }
+            context.addResult(r);
+        }
+
+        knownAttributes = Sets.newHashSet("startAutnum",
+                                          "endAutnum");
+
+        return Utils.runTestList(
+            context, proto, data, knownAttributes, true,
+            Arrays.asList(
+                new ScalarAttribute("name"),
+                new ScalarAttribute("handle"),
+                new ScalarAttribute("type"),
+                new Country(),
+                new StandardResponse()
+            )
+        );
+    }
+
+    /**
+     * <p>Getter for the field <code>knownAttributes</code>.</p>
+     *
+     * @return a {@link java.util.Set} object.
+     */
+    public Set<String> getKnownAttributes() {
+        return knownAttributes;
+    }
+}