def test_errors(self, cmdgen, err_indication, err_status): cmdgen.CommandGenerator.return_value.getCmd.return_value = ( err_indication, err_status, None, [[OID_TONER_CUR, 500]], ) with pytest.raises(IOError): _snmp('logjam', OID_TONER_CUR)
def test_snmp(self, cmdgen): cmdgen.CommandGenerator.return_value.getCmd.return_value = ( None, None, None, [[OID_TONER_CUR, 500]], ) assert _snmp('logjam', OID_TONER_CUR) == 500 cmdgen.CommandGenerator.return_value.getCmd.assert_called_with( cmdgen.CommunityData('my-agent', 'public', 0), cmdgen.UdpTransportTarget(('logjam', 161)), OID_TONER_CUR, )