<filename>unbrake-api/calibration/test_models.py<gh_stars>1-10
'''
Tests for models of calibration app
'''

import pytest
from django.test import Client
from calibration.models import CalibrationVibration, CalibrationForce
from calibration.models import CalibrationRelations, CalibrationTemperature
from calibration.models import CalibrationCommand, CalibrationSpeed
from utils.token import create_token  # pylint: disable = unused-import


# First argument are the parameters names
# Second is a tuple of params
# First argument of param is the first parameter name and so on
# id is like the name for the test case
# Is possible to test only one test case with: pytest [file] -k [id]
# pylint: disable = redefined-outer-name
@pytest.mark.django_db
def test_calibration_vibration(create_token):
    '''
        This test save a CalibrationVibration object on db,
        require the saved object by graphql,
        and check if the requirement is equal the saved object
    '''
    token = create_token()
    response = {
        'acquisitionChanel': 6,
        'conversionFactor': 1.00,
        'vibrationOffset': 1.00
    }

    CalibrationVibration(
        acquisition_chanel=6,
        conversion_factor=1.00,
        vibration_offset=1.00,
    ).save()

    CalibrationVibration(
        acquisition_chanel=7,
        conversion_factor=2.00,
        vibration_offset=2.00,
    ).save()

    client = Client()
    result = client.get(
        '/graphql?query={calibrationVibration(id: 1)'
        '{acquisitionChanel, conversionFactor, vibrationOffset}}',
        HTTP_AUTHORIZATION=token)
    assert result.status_code == 200
    single_aux = result.json()['data']['calibrationVibration']
    single_calibration_vibration = single_aux

    client = None
    result = None

    client = Client()
    result = client.get(
        '/graphql?query={allCalibrationVibration'
        '{id, acquisitionChanel, conversionFactor, vibrationOffset}}',
        HTTP_AUTHORIZATION=token)
    assert result.status_code == 200
    multiple_aux = result.json()['data']['allCalibrationVibration']
    multiple_calibration_vibration_1 = multiple_aux[1]

    assert multiple_calibration_vibration_1['id'] == '2'

    assert single_calibration_vibration == response


@pytest.mark.django_db
def test_calibration_force(create_token):
    '''
        This test save a CalibrationForce object on db,
        require the saved object by graphql,
        and check if the requirement is equal the saved object
    '''
    token = create_token()
    response = {
        'acquisitionChanel': 3,
        'conversionFactor': 1.000,
        'forceOffset': 1.000
    }

    CalibrationForce(
        acquisition_chanel=3,
        conversion_factor=1.000,
        force_offset=1.000,
    ).save()

    client = Client()
    result = client.get(
        '/graphql?query={calibrationForce(id: 1)'
        '{acquisitionChanel, conversionFactor, forceOffset}}',
        HTTP_AUTHORIZATION=token)
    assert result.status_code == 200
    single_calibration_force = result.json()['data']['calibrationForce']

    CalibrationForce(
        acquisition_chanel=4,
        conversion_factor=2.000,
        force_offset=2.000,
    ).save()

    result = None
    client = None

    client = Client()
    result = client.get(
        '/graphql?query={allCalibrationForce'
        '{id, acquisitionChanel, conversionFactor, forceOffset}}',
        HTTP_AUTHORIZATION=token)
    assert result.status_code == 200
    multiple_aux = result.json()['data']['allCalibrationForce']
    multiple_calibration_force_1 = multiple_aux[1]

    assert multiple_calibration_force_1['id'] == '2'

    assert single_calibration_force == response


@pytest.mark.django_db
def test_calibration_speed(create_token):
    '''
        This test save a CalibrationSpeed object on db,
        require the saved object by graphql,
        and check if the requirement is equal the saved object
    '''
    token = create_token()
    response = {
        'acquisitionChanel': 5,
        'tireRadius': 0.291550
    }

    CalibrationSpeed(
        acquisition_chanel=5,
        tire_radius=0.291550,
    ).save()

    client = Client()
    result = client.get(
        '/graphql?query={calibrationSpeed(id: 1)'
        '{acquisitionChanel, tireRadius}}', HTTP_AUTHORIZATION=token)
    assert result.status_code == 200
    single_calibration_speed = result.json()['data']['calibrationSpeed']

    CalibrationSpeed(
        acquisition_chanel=6,
        tire_radius=0.291551,
    ).save()

    client = None
    result = None

    client = Client()
    result = client.get(
        '/graphql?query={allCalibrationSpeed'
        '{id, acquisitionChanel, tireRadius}}', HTTP_AUTHORIZATION=token)
    assert result.status_code == 200
    multiple_aux = result.json()['data']['allCalibrationSpeed']
    multiple_calibration_speed_1 = multiple_aux[1]

    assert multiple_calibration_speed_1['id'] == '2'

    assert single_calibration_speed == response


@pytest.mark.django_db
@pytest.mark.parametrize(
    ("acquisition_chanel_0,acquisition_chanel_1,"
     "tire_radius_0, tire_radius_1"),
    (pytest.param(5, 10, 0.291550, 0.583, id='all_speed_test_1'),
     )
)
def test_all_calibration_speed(acquisition_chanel_0, acquisition_chanel_1,
                               tire_radius_0, tire_radius_1, create_token):
    '''
        This test save two CalibrationSpeed objects on db,
        require all the saved objects by graphql,
        and check if the requirement is equal the saveds objects
    '''
    token = create_token()
    response_0 = {
        'acquisitionChanel': acquisition_chanel_0,
        'tireRadius': tire_radius_0
    }

    response_1 = {
        'acquisitionChanel': acquisition_chanel_1,
        'tireRadius': tire_radius_1
    }

    CalibrationSpeed(
        acquisition_chanel=acquisition_chanel_0,
        tire_radius=tire_radius_0,
    ).save()

    CalibrationSpeed(
        acquisition_chanel=acquisition_chanel_1,
        tire_radius=tire_radius_1,
    ).save()

    client = Client()
    result = client.get(
        '/graphql?query={allCalibrationSpeed'
        '{acquisitionChanel, tireRadius}}', HTTP_AUTHORIZATION=token)
    assert result.status_code == 200
    calibration_speed0 = result.json()['data']['allCalibrationSpeed'][0]
    calibration_speed1 = result.json()['data']['allCalibrationSpeed'][1]

    assert calibration_speed0 == response_0

    assert calibration_speed1 == response_1


@pytest.mark.django_db
def test_calibration_relations(create_token):
    '''
        This test save a CalibrationRelations object on db,
        require the saved object by graphql,
        and check if the requirement is equal the saved object
    '''
    token = create_token()
    response = {
        'id': '1',
        'transversalSelectionWidth': 175,
        'heigthWidthRelation': 65,
        'rimDiameter': 14,
        'syncMotorRodation': 1700,
        'sheaveMoveDiameter': 12,
        'sheaveMotorDiameter': 30
    }

    CalibrationRelations(
        transversal_selection_width=175,
        heigth_width_relation=65,
        rim_diameter=14,
        sync_motor_rodation=1700,
        sheave_move_diameter=12,
        sheave_motor_diameter=30,
    ).save()

    CalibrationRelations(
        transversal_selection_width=176,
        heigth_width_relation=66,
        rim_diameter=15,
        sync_motor_rodation=1701,
        sheave_move_diameter=13,
        sheave_motor_diameter=31,
    ).save()

    client = Client()
    result = client.get(
        '/graphql?query={calibrationRelations(id: 1)'
        '{id, transversalSelectionWidth,'
        'heigthWidthRelation, rimDiameter, syncMotorRodation,'
        'sheaveMoveDiameter, sheaveMotorDiameter}}',
        HTTP_AUTHORIZATION=token)

    assert result.status_code == 200

    single_aux = result.json()['data']['calibrationRelations']
    single_calibration_relations = single_aux

    client = None
    result = None

    client = Client()
    result = client.get(
        '/graphql?query={allCalibrationRelations'
        '{id, transversalSelectionWidth,'
        'heigthWidthRelation, rimDiameter, syncMotorRodation,'
        'sheaveMoveDiameter, sheaveMotorDiameter}}',
        HTTP_AUTHORIZATION=token)
    assert result.status_code == 200

    multiple_aux = result.json()['data']['allCalibrationRelations']
    multiple_calibration_relations_1 = multiple_aux[1]

    assert multiple_calibration_relations_1['id'] == '2'

    assert single_calibration_relations == response


@pytest.mark.django_db
def test_calibration_temperature(create_token):
    '''
        This test save a CalibrationTemperature object on db,
        require the saved object by graphql,
        and check if the requirement is equal the saved object
    '''
    token = create_token()
    response = {
        'acquisitionChanel': 1,
        'conversionFactor': 0.200,
        'temperatureOffset': -1.2500
    }

    CalibrationTemperature(
        acquisition_chanel=1,
        conversion_factor=0.200,
        temperature_offset=-1.2500,
    ).save()

    CalibrationTemperature(
        acquisition_chanel=2,
        conversion_factor=0.400,
        temperature_offset=-2.500,
    ).save()

    client = Client()
    result = client.get(
        '/graphql?query={calibrationTemperature(id: 1)'
        '{acquisitionChanel, conversionFactor, temperatureOffset}}',
        HTTP_AUTHORIZATION=token)
    assert result.status_code == 200
    single_aux = result.json()['data']['calibrationTemperature']
    single_calibration_temperature = single_aux

    client = None
    result = None

    client = Client()
    result = client.get(
        '/graphql?query={allCalibrationTemperature'
        '{id, acquisitionChanel, conversionFactor, temperatureOffset}}',
        HTTP_AUTHORIZATION=token)
    assert result.status_code == 200
    multiple_aux = result.json()['data']['allCalibrationTemperature']
    multiple_calibration_temperature_1 = multiple_aux[1]

    assert multiple_calibration_temperature_1['id'] == '2'

    assert single_calibration_temperature == response


@pytest.mark.django_db
def test_calibration_commands(create_token):
    '''
        This test save a CalibrationCommand object on db,
        require the saved object by graphql,
        and check if the requirement is equal the saved object
    '''
    token = create_token()
    response = {
        'commandChanelSpeed': 7,
        'actualSpeed': 0.000,
        'maxSpeed': 100.000,
        'chanelCommandPression': 8,
        'actualPression': 0.000,
        'maxPression': 30.000
    }

    CalibrationCommand(
        command_chanel_speed=7,
        actual_speed=0.000,
        max_speed=100.000,
        chanel_command_pression=8,
        actual_pression=0.000,
        max_pression=30.000,
    ).save()

    CalibrationCommand(
        command_chanel_speed=6,
        actual_speed=1.000,
        max_speed=101.000,
        chanel_command_pression=9,
        actual_pression=1.000,
        max_pression=31.000,
    ).save()

    client = Client()
    result_single = client.get(
        '/graphql?query={calibrationCommand(id: 1)'
        '{commandChanelSpeed,actualSpeed,maxSpeed,'
        'chanelCommandPression,actualPression,maxPression}}',
        HTTP_AUTHORIZATION=token)
    assert result_single.status_code == 200

    single_aux = result_single.json()['data']['calibrationCommand']
    single_calibration_commands = single_aux

    result_multiple = client.get(
        '/graphql?query={allCalibrationCommand'
        '{id,commandChanelSpeed,actualSpeed,maxSpeed,'
        'chanelCommandPression,actualPression,maxPression}}',
        HTTP_AUTHORIZATION=token)
    assert result_multiple.status_code == 200

    multiple_aux = result_multiple.json()['data']['allCalibrationCommand']
    multiple_calibration_commands_1 = multiple_aux[1]

    assert multiple_calibration_commands_1['id'] == '2'

    assert single_calibration_commands == response

<filename>unbrake-api/calibration/test_models.py<gh_stars>1-10
'''
Tests for models of calibration app
'''

import pytest
from django.test import Client
from calibration.models import CalibrationVibration, CalibrationForce
from calibration.models import CalibrationRelations, CalibrationTemperature
from calibration.models import CalibrationCommand, CalibrationSpeed
from utils.token import create_token  # pylint: disable = unused-import


# First argument are the parameters names
# Second is a tuple of params
# First argument of param is the first parameter name and so on
# id is like the name for the test case
# Is possible to test only one test case with: pytest [file] -k [id]
# pylint: disable = redefined-outer-name
@pytest.mark.django_db
def test_calibration_vibration(create_token):
    '''
        This test save a CalibrationVibration object on db,
        require the saved object by graphql,
        and check if the requirement is equal the saved object
    '''
    token = create_token()
    response = {
        'acquisitionChanel': 6,
        'conversionFactor': 1.00,
        'vibrationOffset': 1.00
    }

    CalibrationVibration(
        acquisition_chanel=6,
        conversion_factor=1.00,
        vibration_offset=1.00,
    ).save()

    CalibrationVibration(
        acquisition_chanel=7,
        conversion_factor=2.00,
        vibration_offset=2.00,
    ).save()

    client = Client()
    result = client.get(
        '/graphql?query={calibrationVibration(id: 1)'
        '{acquisitionChanel, conversionFactor, vibrationOffset}}',
        HTTP_AUTHORIZATION=token)
    assert result.status_code == 200
    single_aux = result.json()['data']['calibrationVibration']
    single_calibration_vibration = single_aux

    client = None
    result = None

    client = Client()
    result = client.get(
        '/graphql?query={allCalibrationVibration'
        '{id, acquisitionChanel, conversionFactor, vibrationOffset}}',
        HTTP_AUTHORIZATION=token)
    assert result.status_code == 200
    multiple_aux = result.json()['data']['allCalibrationVibration']
    multiple_calibration_vibration_1 = multiple_aux[1]

    assert multiple_calibration_vibration_1['id'] == '2'

    assert single_calibration_vibration == response


@pytest.mark.django_db
def test_calibration_force(create_token):
    '''
        This test save a CalibrationForce object on db,
        require the saved object by graphql,
        and check if the requirement is equal the saved object
    '''
    token = create_token()
    response = {
        'acquisitionChanel': 3,
        'conversionFactor': 1.000,
        'forceOffset': 1.000
    }

    CalibrationForce(
        acquisition_chanel=3,
        conversion_factor=1.000,
        force_offset=1.000,
    ).save()

    client = Client()
    result = client.get(
        '/graphql?query={calibrationForce(id: 1)'
        '{acquisitionChanel, conversionFactor, forceOffset}}',
        HTTP_AUTHORIZATION=token)
    assert result.status_code == 200
    single_calibration_force = result.json()['data']['calibrationForce']

    CalibrationForce(
        acquisition_chanel=4,
        conversion_factor=2.000,
        force_offset=2.000,
    ).save()

    result = None
    client = None

    client = Client()
    result = client.get(
        '/graphql?query={allCalibrationForce'
        '{id, acquisitionChanel, conversionFactor, forceOffset}}',
        HTTP_AUTHORIZATION=token)
    assert result.status_code == 200
    multiple_aux = result.json()['data']['allCalibrationForce']
    multiple_calibration_force_1 = multiple_aux[1]

    assert multiple_calibration_force_1['id'] == '2'

    assert single_calibration_force == response


@pytest.mark.django_db
def test_calibration_speed(create_token):
    '''
        This test save a CalibrationSpeed object on db,
        require the saved object by graphql,
        and check if the requirement is equal the saved object
    '''
    token = create_token()
    response = {
        'acquisitionChanel': 5,
        'tireRadius': 0.291550
    }

    CalibrationSpeed(
        acquisition_chanel=5,
        tire_radius=0.291550,
    ).save()

    client = Client()
    result = client.get(
        '/graphql?query={calibrationSpeed(id: 1)'
        '{acquisitionChanel, tireRadius}}', HTTP_AUTHORIZATION=token)
    assert result.status_code == 200
    single_calibration_speed = result.json()['data']['calibrationSpeed']

    CalibrationSpeed(
        acquisition_chanel=6,
        tire_radius=0.291551,
    ).save()

    client = None
    result = None

    client = Client()
    result = client.get(
        '/graphql?query={allCalibrationSpeed'
        '{id, acquisitionChanel, tireRadius}}', HTTP_AUTHORIZATION=token)
    assert result.status_code == 200
    multiple_aux = result.json()['data']['allCalibrationSpeed']
    multiple_calibration_speed_1 = multiple_aux[1]

    assert multiple_calibration_speed_1['id'] == '2'

    assert single_calibration_speed == response


@pytest.mark.django_db
@pytest.mark.parametrize(
    ("acquisition_chanel_0,acquisition_chanel_1,"
     "tire_radius_0, tire_radius_1"),
    (pytest.param(5, 10, 0.291550, 0.583, id='all_speed_test_1'),
     )
)
def test_all_calibration_speed(acquisition_chanel_0, acquisition_chanel_1,
                               tire_radius_0, tire_radius_1, create_token):
    '''
        This test save two CalibrationSpeed objects on db,
        require all the saved objects by graphql,
        and check if the requirement is equal the saveds objects
    '''
    token = create_token()
    response_0 = {
        'acquisitionChanel': acquisition_chanel_0,
        'tireRadius': tire_radius_0
    }

    response_1 = {
        'acquisitionChanel': acquisition_chanel_1,
        'tireRadius': tire_radius_1
    }

    CalibrationSpeed(
        acquisition_chanel=acquisition_chanel_0,
        tire_radius=tire_radius_0,
    ).save()

    CalibrationSpeed(
        acquisition_chanel=acquisition_chanel_1,
        tire_radius=tire_radius_1,
    ).save()

    client = Client()
    result = client.get(
        '/graphql?query={allCalibrationSpeed'
        '{acquisitionChanel, tireRadius}}', HTTP_AUTHORIZATION=token)
    assert result.status_code == 200
    calibration_speed0 = result.json()['data']['allCalibrationSpeed'][0]
    calibration_speed1 = result.json()['data']['allCalibrationSpeed'][1]

    assert calibration_speed0 == response_0

    assert calibration_speed1 == response_1


@pytest.mark.django_db
def test_calibration_relations(create_token):
    '''
        This test save a CalibrationRelations object on db,
        require the saved object by graphql,
        and check if the requirement is equal the saved object
    '''
    token = create_token()
    response = {
        'id': '1',
        'transversalSelectionWidth': 175,
        'heigthWidthRelation': 65,
        'rimDiameter': 14,
        'syncMotorRodation': 1700,
        'sheaveMoveDiameter': 12,
        'sheaveMotorDiameter': 30
    }

    CalibrationRelations(
        transversal_selection_width=175,
        heigth_width_relation=65,
        rim_diameter=14,
        sync_motor_rodation=1700,
        sheave_move_diameter=12,
        sheave_motor_diameter=30,
    ).save()

    CalibrationRelations(
        transversal_selection_width=176,
        heigth_width_relation=66,
        rim_diameter=15,
        sync_motor_rodation=1701,
        sheave_move_diameter=13,
        sheave_motor_diameter=31,
    ).save()

    client = Client()
    result = client.get(
        '/graphql?query={calibrationRelations(id: 1)'
        '{id, transversalSelectionWidth,'
        'heigthWidthRelation, rimDiameter, syncMotorRodation,'
        'sheaveMoveDiameter, sheaveMotorDiameter}}',
        HTTP_AUTHORIZATION=token)

    assert result.status_code == 200

    single_aux = result.json()['data']['calibrationRelations']
    single_calibration_relations = single_aux

    client = None
    result = None

    client = Client()
    result = client.get(
        '/graphql?query={allCalibrationRelations'
        '{id, transversalSelectionWidth,'
        'heigthWidthRelation, rimDiameter, syncMotorRodation,'
        'sheaveMoveDiameter, sheaveMotorDiameter}}',
        HTTP_AUTHORIZATION=token)
    assert result.status_code == 200

    multiple_aux = result.json()['data']['allCalibrationRelations']
    multiple_calibration_relations_1 = multiple_aux[1]

    assert multiple_calibration_relations_1['id'] == '2'

    assert single_calibration_relations == response


@pytest.mark.django_db
def test_calibration_temperature(create_token):
    '''
        This test save a CalibrationTemperature object on db,
        require the saved object by graphql,
        and check if the requirement is equal the saved object
    '''
    token = create_token()
    response = {
        'acquisitionChanel': 1,
        'conversionFactor': 0.200,
        'temperatureOffset': -1.2500
    }

    CalibrationTemperature(
        acquisition_chanel=1,
        conversion_factor=0.200,
        temperature_offset=-1.2500,
    ).save()

    CalibrationTemperature(
        acquisition_chanel=2,
        conversion_factor=0.400,
        temperature_offset=-2.500,
    ).save()

    client = Client()
    result = client.get(
        '/graphql?query={calibrationTemperature(id: 1)'
        '{acquisitionChanel, conversionFactor, temperatureOffset}}',
        HTTP_AUTHORIZATION=token)
    assert result.status_code == 200
    single_aux = result.json()['data']['calibrationTemperature']
    single_calibration_temperature = single_aux

    client = None
    result = None

    client = Client()
    result = client.get(
        '/graphql?query={allCalibrationTemperature'
        '{id, acquisitionChanel, conversionFactor, temperatureOffset}}',
        HTTP_AUTHORIZATION=token)
    assert result.status_code == 200
    multiple_aux = result.json()['data']['allCalibrationTemperature']
    multiple_calibration_temperature_1 = multiple_aux[1]

    assert multiple_calibration_temperature_1['id'] == '2'

    assert single_calibration_temperature == response


@pytest.mark.django_db
def test_calibration_commands(create_token):
    '''
        This test save a CalibrationCommand object on db,
        require the saved object by graphql,
        and check if the requirement is equal the saved object
    '''
    token = create_token()
    response = {
        'commandChanelSpeed': 7,
        'actualSpeed': 0.000,
        'maxSpeed': 100.000,
        'chanelCommandPression': 8,
        'actualPression': 0.000,
        'maxPression': 30.000
    }

    CalibrationCommand(
        command_chanel_speed=7,
        actual_speed=0.000,
        max_speed=100.000,
        chanel_command_pression=8,
        actual_pression=0.000,
        max_pression=30.000,
    ).save()

    CalibrationCommand(
        command_chanel_speed=6,
        actual_speed=1.000,
        max_speed=101.000,
        chanel_command_pression=9,
        actual_pression=1.000,
        max_pression=31.000,
    ).save()

    client = Client()
    result_single = client.get(
        '/graphql?query={calibrationCommand(id: 1)'
        '{commandChanelSpeed,actualSpeed,maxSpeed,'
        'chanelCommandPression,actualPression,maxPression}}',
        HTTP_AUTHORIZATION=token)
    assert result_single.status_code == 200

    single_aux = result_single.json()['data']['calibrationCommand']
    single_calibration_commands = single_aux

    result_multiple = client.get(
        '/graphql?query={allCalibrationCommand'
        '{id,commandChanelSpeed,actualSpeed,maxSpeed,'
        'chanelCommandPression,actualPression,maxPression}}',
        HTTP_AUTHORIZATION=token)
    assert result_multiple.status_code == 200

    multiple_aux = result_multiple.json()['data']['allCalibrationCommand']
    multiple_calibration_commands_1 = multiple_aux[1]

    assert multiple_calibration_commands_1['id'] == '2'

    assert single_calibration_commands == response

Tests for models of calibration app # pylint: disable = unused-import # First argument are the parameters names # Second is a tuple of params # First argument of param is the first parameter name and so on # id is like the name for the test case # Is possible to test only one test case with: pytest [file] -k [id] # pylint: disable = redefined-outer-name This test save a CalibrationVibration object on db,         require the saved object by graphql,         and check if the requirement is equal the saved object This test save a CalibrationForce object on db,         require the saved object by graphql,         and check if the requirement is equal the saved object This test save a CalibrationSpeed object on db,         require the saved object by graphql,         and check if the requirement is equal the saved object This test save two CalibrationSpeed objects on db,         require all the saved objects by graphql,         and check if the requirement is equal the saveds objects This test save a CalibrationRelations object on db,         require the saved object by graphql,         and check if the requirement is equal the saved object This test save a CalibrationTemperature object on db,         require the saved object by graphql,         and check if the requirement is equal the saved object This test save a CalibrationCommand object on db,         require the saved object by graphql,         and check if the requirement is equal the saved object