Question

我已经编写了以下用于从LDAP获取结果的代码。从结果中我需要提取一个条目并对其执行操作。这段代码运行正常。但问题是当我尝试为它编写单元测试时。我需要加载测试数据或以某种方式模拟某些东西来为这段代码编写测试。有人可以帮我指导写作方向。

讨论中的代码如下：

NamingEnumeration<SearchResult> results = dirContext.search(ldapSearchBase, searchFilter, searchControls);

if (results.hasMoreElements()) {

    List<String> securityGroups = new ArrayList<String>();
    SearchResult searchResult = results.nextElement();
    NamingEnumeration ldapAttributes = searchResult.getAttributes().getAll();

    while (ldapAttributes.hasMore()) {
        Attribute attr = (Attribute) (ldapAttributes.next());
        for (int i = 0; i < attr.size(); i++) {
            if (attr.get(i).toString().startsWith("CN=GG-PaaS-logging-service")) {
                String commonName = attr.get(i).toString();
                int startIndex = commonName.indexOf("=") + 1;
                int endIndex = commonName.indexOf(",");
                commonName = commonName.substring(startIndex, endIndex);
                securityGroups.add(commonName);
            }
        }
    }
}

Answer 1

我看到（至少）需要在代码中模拟7个依赖项。

我们必须嘲笑那么多的原因是因为你的代码无法遵循 Demeter法则又名不要与陌生人交谈。如果您方法的输入直接为ldapAttributes，则可以避免一半的模拟。

我想在您的真实代码中，您实际上正在使用securityGroups执行某些操作。在您的示例中，在离开if ...

时会收集垃圾

以下是我将如何编写测试：

class LdapLoggingServiceExctractorTest{
 @Rule 
 public MockitoRule mockito = MockitoJunit.rule()
 @Mock 
 private DirContext dirContext;
 @Mock 
 NamingEnumeration<SearchResult> results;
 @Mock 
 private SearchResult loggerSearchResult;
 @Mock 
 private Attributes ldapAttributes; // same name but different object!!
 @Mock 
 private NamingEnumeration ldapAttributeEnum;
 @Mock 
 private Attribute attr;

 @Before
 public void setup(){
  doReturn(results).when(dirContext).search(any(Name.class),anyString(),any(SearchControls));
  doReturn(loggerSearchResult).when(results).nextElement();
  doReturn(ldapAttributes).when(loggerSearchResult).getAttributes();
  doReturn(ldapAttributeEnum).when(loggerSearchResult).getAll();
  doReturn(true).when(ldapAttributeEnum).hasMore();
  doReturn(attr).when(ldapAttributeEnum).next();
  doReturn(1).when(attr).size();
  doReturn(THE_VALID_LOGGER_ENTRY_STRING).when(attr).get(0);
 }


 @Test
 public void extractLogger_singleResultSingleAttribut_addsLoggersCommonNameToList(){
   LdapLoggingServiceExctractor ldapLoggingServiceExctractor = new LdapLoggingServiceExctractor();

   ldapLoggingServiceExctractor.extractLogger(dirContext);

   assertThat(ldapLoggingServiceExctractor.getSecurityGroups().get(0),equalTo(LOGGER_COMMON_NAME));
  }
}

你能解释一下“如果你的方法的输入直接是ldapAttributes，你可以避免一半的嘲弄。”我不是很明白。 - Nirmalya Guha Khasnobis

您的代码剪切以：

开头

if (results.hasMoreElements()) {
   List<String> securityGroups = new ArrayList<String>();
   SearchResult searchResult = results.nextElement();
   NamingEnumeration ldapAttributes = searchResult.getAttributes().getAll();

由于LDAP框架存储信息的方式，这是所需的“样板”代码。在我对关注点分离原理的理解中，ldapAttributes的提取是不同单位的责任。所以我的方法是我会有一个单独的类LdapLoggingServiceExctractor，其方法如下：

public void addLoggerToSecurityGroup(NamingEnumeration ldapAttributes,List<String> securityGroups){
    while (ldapAttributes.hasMore()) {
        Attribute attr = (Attribute) (ldapAttributes.next());
        for (int i = 0; i < attr.size(); i++) {
            if (attr.get(i).toString().startsWith("CN=GG-PaaS-logging-service")) {
                String commonName = attr.get(i).toString();
                int startIndex = commonName.indexOf("=") + 1;
                int endIndex = commonName.indexOf(",");
                commonName = commonName.substring(startIndex, endIndex);
                securityGroups.add(commonName);
            }
        }
    }
}

然后测试会简化为：

class LdapLoggingServiceExctractorTest{
 @Rule 
 public MockitoRule mockito = MockitoJunit.rule();
 @Mock 
 private NamingEnumeration ldapAttributeEnum;
 @Mock 
 private Attribute attr;
 @Before
 public void setup(){
  doReturn(true,false).when(ldapAttributeEnum).hasMore(); // corrected to avoid endless loop...
  doReturn(attr).when(ldapAttributeEnum).next();
  doReturn(1).when(attr).size();
  doReturn(THE_VALID_LOGGER_ENTRY_STRING).when(attr).get(0);
 }
 @Test
 public void extractLogger_singleResultSingleAttribut_addsLoggersCommonNameToList(){
   List<String> securityGroups = new ArrayList<String>();
   LdapLoggingServiceExctractor ldapLoggingServiceExctractor = new LdapLoggingServiceExctractor();

   ldapLoggingServiceExctractor.addLoggerToSecurityGroup(ldapAttributeEnum, securityGroups);

   assertThat(securityGroups.get(0),equalTo(LOGGER_COMMON_NAME));
  }
}

从原始版本中要模拟的7个依赖项中只剩下2个。

关闭原因这并不意味着不需要（大部分）其他模拟。
他们将被移动到一个独立的测试，新的类将“样板”代码移动到。

Junit测试在LDAP连接测试中测试NamingEnumeration中的数据

1 个答案: