import pytest

from lab_control.sds1000xe import SDS1000XE
from lab_control.test.mocksocket import MockSocket

testIP = "127.0.0.1"
testPort = 4000
defaultPort = 5025

@pytest.fixture
def mockSocket():
    return MockSocket()

@pytest.fixture
def uut(mockSocket):
    return SDS1000XE(testIP, port=testPort, socket=mockSocket)

def test_defaults():
    mockSocket = MockSocket()
    assert not mockSocket.connected
    assert mockSocket.ip == None
    assert mockSocket.port == None

    uut = SDS1000XE(testIP, socket=mockSocket)
    assert mockSocket.connected
    assert mockSocket.ip == testIP
    assert mockSocket.port == defaultPort

def test_differentPort(uut, mockSocket):
    assert mockSocket.connected
    assert mockSocket.ip == testIP
    assert mockSocket.port == testPort