1 files changed, 23 insertions, 5 deletions

diff --git a/tests/spec/tests.js b/tests/spec/tests.js
index b307f15..2685fa6 100644
--- a/tests/spec/tests.js
+++ b/tests/spec/tests.js
@@ -73,16 +73,23 @@ else if (browser == "chrome") {
 
 // Helper functions
 
-function testNetwork(done, params) {
+function testNetwork(done, params, comparePub = false) {
     var phrase = params.phrase || 'abandon abandon ability';
     driver.findElement(By.css('.phrase'))
         .sendKeys(phrase);
     selectNetwork(params.selectText);
     driver.sleep(generateDelay).then(function() {
-        getFirstAddress(function(address) {
-            expect(address).toBe(params.firstAddress);
-            done();
-        });
+        if (!comparePub) {
+            getFirstAddress(function(address) {
+                expect(address).toBe(params.firstAddress);
+                done();
+            });
+        } else {
+            getFirstPublicKey(function(pubkey) {
+                expect(pubkey).toBe(params.firstPubKey);
+                done();
+            });
+        }
     });
 }
 
@@ -98,6 +105,10 @@ function getFirstAddress(handler) {
     getFirstRowValue(handler, ".address");
 }
 
+function getFirstPublicKey(handler) {
+    getFirstRowValue(handler, ".pubkey");
+}
+
 function getFirstPath(handler) {
     getFirstRowValue(handler, ".index");
 }
@@ -867,6 +878,13 @@ it('Allows selection of Einsteinium', function(done) {
     };
     testNetwork(done, params);
 });
+it('Allows selection of EOSIO', function(done) {
+    var params = {
+        selectText: "EOS - EOSIO",
+        firstPubKey: "EOS692VJTBK3Rmw93onNnpnZ8ZtmE9PdxjDStArvbyzoe11QUTNoy",
+    };
+    testNetowrk(done, params, true);
+});
 it('Allows selection of Europecoin', function(done) {
     var params = {
         selectText: "ERC - Europecoin",