Skip to content

Commit

Permalink
fix unit tests
Browse files Browse the repository at this point in the history
  • Loading branch information
rasantel committed Aug 29, 2023
1 parent e035948 commit 23e7c4e
Showing 1 changed file with 9 additions and 9 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -15,39 +15,39 @@
*/
package net.schmizz.sshj.transport;

import static org.junit.Assert.assertFalse;
import static org.junit.Assert.assertTrue;
import static org.junit.jupiter.api.Assertions.assertFalse;
import static org.junit.jupiter.api.Assertions.assertTrue;
import static org.mockito.ArgumentMatchers.any;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.when;

import java.math.BigInteger;
import java.security.GeneralSecurityException;
import java.util.Collections;
import net.schmizz.sshj.DefaultConfig;
import net.schmizz.sshj.common.Factory;
import net.schmizz.sshj.common.Message;
import net.schmizz.sshj.common.SSHPacket;
import net.schmizz.sshj.transport.kex.KeyExchange;
import net.schmizz.sshj.transport.verification.PromiscuousVerifier;
import org.junit.Before;
import org.junit.Test;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
import org.mockito.Mockito;
import org.testcontainers.shaded.com.google.common.collect.ImmutableList;

public class KeyExchangeRepeatTest {

private TransportImpl transport;
private DefaultConfig config;
private KeyExchanger keyExchanger;

@Before
@BeforeEach
public void setup() throws GeneralSecurityException, TransportException {
KeyExchange kex = mock(KeyExchange.class, Mockito.RETURNS_DEEP_STUBS);
transport = mock(TransportImpl.class, Mockito.RETURNS_DEEP_STUBS);
config = new DefaultConfig() {
@Override
protected void initKeyExchangeFactories(boolean bouncyCastleRegistered) {
setKeyExchangeFactories(ImmutableList.of(new Factory.Named<>() {
protected void initKeyExchangeFactories() {
setKeyExchangeFactories(Collections.singletonList(new Factory.Named<>() {
@Override
public KeyExchange create() {
return kex;
Expand Down Expand Up @@ -112,7 +112,7 @@ private void performAndCheckKeyExchange() throws TransportException {
}

private SSHPacket getKexinitPacket() {
SSHPacket kexinitPacket = new Proposal(config, ImmutableList.of()).getPacket();
SSHPacket kexinitPacket = new Proposal(config, Collections.emptyList()).getPacket();
kexinitPacket.rpos(kexinitPacket.rpos() + 1);
return kexinitPacket;
}
Expand Down

0 comments on commit 23e7c4e

Please sign in to comment.