Merge pull request #166 from OpenHFT/codex/manipulate-affinity-and-assert-value

Add unit test for Affinity.resetToBaseAffinity

Author: peter-lawrey-admin

affinity/src/test/java/net/openhft/affinity/AffinityResetToBaseAffinityTest.java

@@ -0,0 +1,33 @@
+package net.openhft.affinity;
+
+import net.openhft.affinity.impl.VanillaCpuLayout;
+import org.junit.Test;
+
+import java.io.File;
+
+import static org.junit.Assert.assertEquals;
+
+public class AffinityResetToBaseAffinityTest extends BaseAffinityTest {
+
+    @Test
+    public void resettingShouldRestoreBaseAffinity() throws Exception {
+        if (!new File("/proc/cpuinfo").exists()) {
+            System.out.println("Cannot run affinity test as this system doesn't have a /proc/cpuinfo file");
+            return;
+        }
+
+        // initialise CPU layout from the running machine so acquireLock works
+        AffinityLock.cpuLayout(VanillaCpuLayout.fromCpuInfo());
+
+        assertEquals(AffinityLock.BASE_AFFINITY, Affinity.getAffinity());
+        AffinityLock lock = AffinityLock.acquireLock();
+        try {
+            assertEquals(1, Affinity.getAffinity().cardinality());
+
+            Affinity.resetToBaseAffinity();
+            assertEquals(AffinityLock.BASE_AFFINITY, Affinity.getAffinity());
+        } finally {
+            lock.release();
+        }
+    }
+}