yujunjun/datahub
2
0
Fork 0
mirror of https://github.com/datahub-project/datahub.git synced 2025-11-08 07:23:34 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(test): fix bug on verification for EbeanEntityService (#3829)

Browse Source 
Co-authored-by: Ryan Holstien <ryan@acryl.io>
This commit is contained in:
RyanHolstien 2022-01-05 16:05:54 -06:00 committed by GitHub
parent 81f14f9a84
commit 9a237e9c92
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 11 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
metadata-io/src/test/java/com/linkedin/metadata/entity/EbeanEntityServiceTest.java
View File

@ -59,16 +59,8 @@ import org.mockito.Mockito;
import org.testng.annotations.BeforeMethod;
import org.testng.annotations.Test;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.reset;
import static org.mockito.Mockito.times;
import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.verifyNoMoreInteractions;
import static org.testng.Assert.assertEquals;
import static org.testng.Assert.assertFalse;
import static org.testng.Assert.assertNotNull;
import static org.testng.Assert.assertNull;
import static org.testng.Assert.assertTrue;
import static org.mockito.Mockito.*;
import static org.testng.Assert.*;
public class EbeanEntityServiceTest {
@ -337,6 +329,12 @@ public class EbeanEntityServiceTest {
verify(_mockProducer, times(2)).produceMetadataAuditEvent(Mockito.eq(entityUrn2), Mockito.eq(null), Mockito.any(),
Mockito.any(), Mockito.any(), Mockito.eq(MetadataAuditOperation.UPDATE));
verify(_mockProducer, times(2)).produceMetadataChangeLog(Mockito.eq(entityUrn1),
Mockito.any(), Mockito.any());
verify(_mockProducer, times(2)).produceMetadataChangeLog(Mockito.eq(entityUrn2),
Mockito.any(), Mockito.any());
verifyNoMoreInteractions(_mockProducer);
}
@ -443,6 +441,9 @@ public class EbeanEntityServiceTest {
verify(_mockProducer, times(1)).produceMetadataAuditEvent(Mockito.eq(entityUrn), Mockito.notNull(), Mockito.any(),
Mockito.any(), Mockito.any(), Mockito.eq(MetadataAuditOperation.UPDATE));
verify(_mockProducer, times(2)).produceMetadataChangeLog(Mockito.eq(entityUrn),
Mockito.any(), Mockito.any());
verifyNoMoreInteractions(_mockProducer);
}