Proof of concept #1026

Paul Martin

2016-04-16 eecaad8b8e2c447429c31a01d49260ddd6b4ee03

 src/test/java/com/gitblit/tests/SshKeysDispatcherTest.java

@@ -103,6 +103,20 @@
   }

   @Test
   public void testKeysAddBlankCommand() throws Exception {
      testSshCommand("keys add --permission R", "\n");
      List<SshKey> keys = getKeyManager().getKeys(username);
      assertEquals(String.format("There are %d keys!", keys.size()), 2, keys.size());
   }

   @Test
   public void testKeysAddInvalidCommand() throws Exception {
      testSshCommand("keys add --permission R", "My invalid key\n");
      List<SshKey> keys = getKeyManager().getKeys(username);
      assertEquals(String.format("There are %d keys!", keys.size()), 2, keys.size());
   }

   @Test
   public void testKeysCommentCommand() throws Exception {
      List<SshKey> keys = getKeyManager().getKeys(username);
      assertTrue(StringUtils.isEmpty(keys.get(0).getComment()));
@@ -112,4 +126,20 @@
      keys = getKeyManager().getKeys(username);
      assertEquals(comment, keys.get(0).getComment());
   }

   @Test
   public void testKeysPermissionCommand() throws Exception {
      List<SshKey> keys = getKeyManager().getKeys(username);
      assertTrue(StringUtils.isEmpty(keys.get(0).getComment()));
      testSshCommand(String.format("keys permission 1 %s", AccessPermission.CLONE));

      keys = getKeyManager().getKeys(username);
      assertEquals(AccessPermission.CLONE, keys.get(0).getPermission());

      testSshCommand(String.format("keys permission 1 %s", AccessPermission.PUSH));

      keys = getKeyManager().getKeys(username);
      assertEquals(AccessPermission.PUSH, keys.get(0).getPermission());

   }
}