Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

US947955 - Increase unit test (cont.) #2247

Merged
merged 1 commit into from
Mar 12, 2024
Merged

US947955 - Increase unit test (cont.) #2247

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
271 changes: 271 additions & 0 deletions ...engine/src/test/java/org/eclipse/lsp/cobol/core/visitor/CobolDataDivisionVisitorTest.java
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,271 @@
/*
* Copyright (c) 2024 Broadcom.
* The term "Broadcom" refers to Broadcom Inc. and/or its subsidiaries.
*
* This program and the accompanying materials are made
* available under the terms of the Eclipse Public License 2.0
* which is available at https://www.eclipse.org/legal/epl-2.0/
*
* SPDX-License-Identifier: EPL-2.0
*
* Contributors:
* Broadcom, Inc. - initial API and implementation
*
*/
package org.eclipse.lsp.cobol.core.visitor;

import com.google.common.collect.ImmutableList;
import org.antlr.v4.runtime.CharStream;
import org.antlr.v4.runtime.Token;
import org.antlr.v4.runtime.tree.TerminalNode;
import org.eclipse.lsp.cobol.common.mapping.ExtendedDocument;
import org.eclipse.lsp.cobol.common.message.MessageService;
import org.eclipse.lsp.cobol.common.model.NodeType;
import org.eclipse.lsp.cobol.common.model.tree.Node;
import org.eclipse.lsp.cobol.core.CobolDataDivisionParser.*;
import org.eclipse.lsp.cobol.core.CobolParser;
import org.eclipse.lsp.cobol.core.semantics.CopybooksRepository;
import org.eclipse.lsp4j.Location;
import org.eclipse.lsp4j.Position;
import org.eclipse.lsp4j.Range;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.extension.ExtendWith;
import org.mockito.Mock;
import org.mockito.junit.jupiter.MockitoExtension;

import java.util.List;
import java.util.Map;

import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.mockito.ArgumentMatchers.any;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.when;

/**
* Test for {@link CobolDataDivisionVisitor}
*/
@ExtendWith(MockitoExtension.class)
class CobolDataDivisionVisitorTest {
@Mock
private ExtendedDocument extendedDocument;
@Mock
private CopybooksRepository copybooks;
@Mock
private MessageService messageService;
@Mock
private Map<String, CobolParser.FileControlEntryContext> fileControls;

private CobolDataDivisionVisitor visitor;
private Location location;

@BeforeEach
void init() {
visitor = new CobolDataDivisionVisitor(extendedDocument, copybooks,
messageService, fileControls);
Range range = new Range(new Position(1, 1), new Position(2, 2));
location = new Location("URI", range);
}

@Test
void testVisitDataDivision() {
DataDivisionContext ctx = mock(DataDivisionContext.class);
ctx.start = mock(Token.class);
ctx.stop = mock(Token.class);
Token token = mock(Token.class);
when(ctx.getStart()).thenReturn(token);
when(token.getText()).thenReturn("TEXT");
when(extendedDocument.mapLocation(any())).thenReturn(location);
List<Node> result = visitor.visitDataDivision(ctx);

assertEquals(NodeType.DIVISION, result.get(0).getNodeType());
assertEquals(location, result.get(0).getLocality().toLocation());
}

@Test
void testVisitQualifiedDataName() {
QualifiedDataNameContext ctx = mock(QualifiedDataNameContext.class);
ctx.start = mock(Token.class);
ctx.stop = mock(Token.class);
when(extendedDocument.mapLocation(any())).thenReturn(location);
List<Node> result = visitor.visitQualifiedDataName(ctx);

assertEquals(NodeType.QUALIFIED_REFERENCE_NODE, result.get(0).getNodeType());
assertEquals(location, result.get(0).getLocality().toLocation());
}

@Test
void testVisitVariableUsageName() {
VariableUsageNameContext ctx = mock(VariableUsageNameContext.class);
ctx.start = mock(Token.class);
ctx.stop = mock(Token.class);
when(extendedDocument.mapLocation(any())).thenReturn(location);
List<Node> result = visitor.visitVariableUsageName(ctx);

assertEquals(NodeType.VARIABLE_USAGE, result.get(0).getNodeType());
assertEquals(location, result.get(0).getLocality().toLocation());
}

@Test
void testVisitDataDivisionSection() {
DataDivisionSectionContext ctx = mock(DataDivisionSectionContext.class);
Token token = mock(Token.class);
when(ctx.getStart()).thenReturn(token);
when(token.getText()).thenReturn("TEXT");
when(extendedDocument.mapLocation(any())).thenReturn(location);
List<Node> result = visitor.visitDataDivisionSection(ctx);

assertEquals(ImmutableList.of(), result);
}

@Test
void testVisitFileSection() {
FileSectionContext ctx = mock(FileSectionContext.class);
ctx.start = mock(Token.class);
ctx.stop = mock(Token.class);
when(extendedDocument.mapLocation(any())).thenReturn(location);
List<Node> result = visitor.visitFileSection(ctx);

assertEquals(NodeType.SECTION, result.get(0).getNodeType());
assertEquals(location, result.get(0).getLocality().toLocation());
}

@Test
void testVisitWorkingStorageSection() {
WorkingStorageSectionContext ctx = mock(WorkingStorageSectionContext.class);
ctx.start = mock(Token.class);
ctx.stop = mock(Token.class);
when(extendedDocument.mapLocation(any())).thenReturn(location);
List<Node> result = visitor.visitWorkingStorageSection(ctx);

assertEquals(NodeType.SECTION, result.get(0).getNodeType());
assertEquals(location, result.get(0).getLocality().toLocation());
}

@Test
void testVisitLinkageSection() {
LinkageSectionContext ctx = mock(LinkageSectionContext.class);
ctx.start = mock(Token.class);
ctx.stop = mock(Token.class);
when(extendedDocument.mapLocation(any())).thenReturn(location);
List<Node> result = visitor.visitLinkageSection(ctx);

assertEquals(NodeType.SECTION, result.get(0).getNodeType());
assertEquals(location, result.get(0).getLocality().toLocation());
}

@Test
void testVisitLocalStorageSection() {
LocalStorageSectionContext ctx = mock(LocalStorageSectionContext.class);
ctx.start = mock(Token.class);
ctx.stop = mock(Token.class);
when(extendedDocument.mapLocation(any())).thenReturn(location);
List<Node> result = visitor.visitLocalStorageSection(ctx);

assertEquals(NodeType.SECTION, result.get(0).getNodeType());
assertEquals(location, result.get(0).getLocality().toLocation());
}

@Test
void testVisitDataDescriptionEntryFormat1() {
DataDescriptionEntryFormat1Context ctx = mock(DataDescriptionEntryFormat1Context.class);
when(ctx.levelNumber()).thenReturn(mock(LevelNumberContext.class));
TerminalNode terminalNode = mock(TerminalNode.class);
when(ctx.levelNumber().LEVEL_NUMBER()).thenReturn(terminalNode);
when(terminalNode.getText()).thenReturn("1");
Token token = mock(Token.class);
when(token.getText()).thenReturn("TEXT");
when(terminalNode.getSymbol()).thenReturn(token);
when(extendedDocument.mapLocation(any())).thenReturn(mock(Location.class));
ctx.start = mock(Token.class);
ctx.stop = mock(Token.class);
when(ctx.start.getLine()).thenReturn(1);
List<Node> result = visitor.visitDataDescriptionEntryFormat1(ctx);

assertEquals(result.get(0).getNodeType(), NodeType.VARIABLE_DEFINITION);
}

@Test
void testVisitDataDescriptionEntryFormat1_exception() {
DataDescriptionEntryFormat1Context ctx = mock(DataDescriptionEntryFormat1Context.class);
when(ctx.levelNumber()).thenReturn(mock(LevelNumberContext.class));
TerminalNode terminalNode = mock(TerminalNode.class);
when(ctx.levelNumber().LEVEL_NUMBER()).thenReturn(terminalNode);
when(terminalNode.getText()).thenReturn("1");
Token token = mock(Token.class);
when(token.getText()).thenReturn("TEXT");
when(terminalNode.getSymbol()).thenReturn(token);
when(extendedDocument.mapLocation(any())).thenThrow(new IllegalStateException("Illegal State"));
ctx.start = mock(Token.class);
ctx.stop = mock(Token.class);
try {
visitor.visitDataDescriptionEntryFormat1(ctx);
} catch (IllegalStateException e) {
assertEquals("Illegal State", e.getMessage());
}
}

@Test
void testVisitFileDescriptionEntry() {
FileDescriptionEntryContext ctx = mock(FileDescriptionEntryContext.class);
CobolWordContext cobolWordContext = mock(CobolWordContext.class);
FileDescriptionEntryClausesContext entryClausesContext = mock(FileDescriptionEntryClausesContext.class);
Token token = mock(Token.class);
when(ctx.getStart()).thenReturn(token);
when(token.getText()).thenReturn("TEXT");
when((ctx.fileDescriptionEntryClauses())).thenReturn(entryClausesContext);
when(ctx.fileDescriptionEntryClauses().cobolWord()).thenReturn(cobolWordContext);
cobolWordContext.start = mock(Token.class);
cobolWordContext.stop = mock(Token.class);
when(ctx.fileDescriptionEntryClauses().getStart()).thenReturn(mock(Token.class));
when(ctx.fileDescriptionEntryClauses().getStop()).thenReturn(mock(Token.class));
when(ctx.fileDescriptionEntryClauses().getStart().getInputStream()).thenReturn(mock(CharStream.class));
when(extendedDocument.mapLocation(any())).thenReturn(location);
List<Node> result = visitor.visitFileDescriptionEntry(ctx);

assertEquals(result.get(0).getLocality().getRange().getStart(), new Position(1, 1));
assertEquals(result.get(0).getLocality().getRange().getEnd(), new Position(2, 2));
}

@Test
void testVisitDataDescriptionEntryFormat2() {
DataDescriptionEntryFormat2Context ctx = mock(DataDescriptionEntryFormat2Context.class);
Token token = mock(Token.class);
when(token.getText()).thenReturn("TEXT");
TerminalNode terminalNode = mock(TerminalNode.class);
when(ctx.LEVEL_NUMBER_66()).thenReturn(terminalNode);
when(terminalNode.getSymbol()).thenReturn(token);
ctx.start = mock(Token.class);
ctx.stop = mock(Token.class);
when(extendedDocument.mapLocation(any())).thenReturn(location);
List<Node> result = visitor.visitDataDescriptionEntryFormat2(ctx);

assertEquals(result.get(0).getLocality().getRange().getStart(), new Position(1, 1));
assertEquals(result.get(0).getLocality().getRange().getEnd(), new Position(2, 2));
}

@Test
void testVisitDataDescriptionEntryFormat1Level77() {
DataDescriptionEntryFormat1Level77Context ctx = mock(DataDescriptionEntryFormat1Level77Context.class);
Token token = mock(Token.class);
when(token.getText()).thenReturn("TEXT");
TerminalNode terminalNode = mock(TerminalNode.class);
when(ctx.LEVEL_NUMBER_77()).thenReturn(terminalNode);
when(terminalNode.getSymbol()).thenReturn(token);
ctx.start = mock(Token.class);
ctx.stop = mock(Token.class);
when(extendedDocument.mapLocation(any())).thenReturn(location);
List<Node> result = visitor.visitDataDescriptionEntryFormat1Level77(ctx);

assertEquals(result.get(0).getLocality().getRange().getStart(), new Position(1, 1));
assertEquals(result.get(0).getLocality().getRange().getEnd(), new Position(2, 2));
}

@Test
void testVisitDataDescriptionEntryFormat3() {
DataDescriptionEntryFormat3Context ctx = mock(DataDescriptionEntryFormat3Context.class);
List<Node> result = visitor.visitDataDescriptionEntryFormat3(ctx);

assertEquals(ImmutableList.of(), result);
}
}
103 changes: 103 additions & 0 deletions server/engine/src/test/java/org/eclipse/lsp/cobol/lsp/analysis/AsyncAnalysisServiceTest.java
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,103 @@
/*
* Copyright (c) 2024 Broadcom.
* The term "Broadcom" refers to Broadcom Inc. and/or its subsidiaries.
*
* This program and the accompanying materials are made
* available under the terms of the Eclipse Public License 2.0
* which is available at https://www.eclipse.org/legal/epl-2.0/
*
* SPDX-License-Identifier: EPL-2.0
*
* Contributors:
* Broadcom, Inc. - initial API and implementation
*
*/
package org.eclipse.lsp.cobol.lsp.analysis;

import org.eclipse.lsp.cobol.common.SubroutineService;
import org.eclipse.lsp.cobol.lsp.SourceUnitGraph;
import org.eclipse.lsp.cobol.service.AnalysisService;
import org.eclipse.lsp.cobol.service.CobolDocumentModel;
import org.eclipse.lsp.cobol.service.DocumentModelService;
import org.eclipse.lsp.cobol.service.copybooks.CopybookServiceImpl;
import org.eclipse.lsp.cobol.service.delegates.communications.Communications;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.extension.ExtendWith;
import org.mockito.Mock;
import org.mockito.junit.jupiter.MockitoExtension;

import java.lang.reflect.Field;
import java.util.*;
import java.util.stream.Collectors;
import java.util.stream.Stream;

import static org.mockito.Mockito.*;

/**
* Test {@link AsyncAnalysisService}
*/
@ExtendWith(MockitoExtension.class)
class AsyncAnalysisServiceTest {
@Mock
private DocumentModelService documentModelService;
@Mock
private AnalysisService analysisService;
@Mock
private CopybookServiceImpl copybookService;
@Mock
private SubroutineService subroutineService;
@Mock
private Communications communication;

private AsyncAnalysisService asyncAnalysisService;

@BeforeEach
void setUp() {
asyncAnalysisService = new AsyncAnalysisService(documentModelService, analysisService, copybookService,
subroutineService, communication);
}

@Test
void testReanalyseOpenedPrograms() {
try {
asyncAnalysisService.reanalyseOpenedPrograms();
} catch (InterruptedException e) {
throw new RuntimeException(e);
}

verify(copybookService, times(1)).invalidateCache();
verify(subroutineService, times(1)).invalidateCache();
}

@Test
void testReanalyseCopybooksAssociatedPrograms() {
Map<String, Integer> analysisResultsRevisionsMock = mock(Map.class);
try {
Field field = AsyncAnalysisService.class.getDeclaredField("analysisResultsRevisions");
field.setAccessible(true);
field.set(asyncAnalysisService, analysisResultsRevisionsMock);
} catch (Exception e) {
throw new RuntimeException(e);
}
List<String> uris = Stream.of("URI1", "URI2").collect(Collectors.toList());
SourceUnitGraph.EventSource eventSource = mock(SourceUnitGraph.EventSource.class);
CobolDocumentModel cobolDocumentModel = mock(CobolDocumentModel.class);
when(documentModelService.get(any())).thenReturn(cobolDocumentModel);
when(analysisResultsRevisionsMock.get(cobolDocumentModel.getUri())).thenReturn(1);
asyncAnalysisService.reanalyseCopybooksAssociatedPrograms(uris, "copybookUri", "copybookContent", eventSource);

verify(cobolDocumentModel, times(2)).getText();
}

@Test
void testCancelAnalysis() {
try {
asyncAnalysisService.cancelAnalysis("URI");
} catch (InterruptedException e) {
throw new RuntimeException(e);
}

verify(communication, times(1)).publishDiagnostics(any());
}
}
Loading
Loading