Skip to content
Snippets Groups Projects
Select Git revision
  • develop default
  • master protected
  • 0.9
  • 0.8
  • 0.6
  • 0.5
  • 0.4
  • 0.3
  • 0.2
  • 0.1
10 results
Blame Permalink
test_forms.py 3.85 KiB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95 






"""Tests for the file_validator forms."""
import csv
import io
from unittest.mock import MagicMock

from django.core.files.uploadedfile import SimpleUploadedFile

from sage_validation.file_validator.forms import CSVUploadForm


def create_modified_csv(sample_file: SimpleUploadedFile, modifications: dict[str, str]) -> SimpleUploadedFile:
    """
    Modify specific fields in the first row of a CSV file and return a new SimpleUploadedFile.

    Args:
        sample_file (SimpleUploadedFile): The original CSV file.
        modifications (dict): Dictionary of column names to modified values.

    Returns:
        SimpleUploadedFile: The modified CSV file.
    """
    csv_content = sample_file.read().decode("utf-8").splitlines()
    reader = csv.DictReader(csv_content)
    rows = list(reader)

    for key, value in modifications.items():
        rows[0][key] = value

    output = io.StringIO()
    writer = csv.DictWriter(output, fieldnames=reader.fieldnames or [])
    writer.writeheader()
    writer.writerows(rows)

    return SimpleUploadedFile("test_modified.csv", output.getvalue().encode("utf-8"), content_type="text/csv")


def test_valid_csv_upload(sample_input_file: SimpleUploadedFile, mock_meo_database: MagicMock) -> None:
    """Test CSV upload with valid data."""
    form = CSVUploadForm(files={"file": sample_input_file})
    assert form.is_valid(), f"Form errors: {form.errors}"


def test_invalid_file_extension() -> None:
    """Test form rejects non-CSV files."""
    bad_file = SimpleUploadedFile("test.txt", b"Some text content", content_type="text/plain")
    form = CSVUploadForm(files={"file": bad_file})
    assert not form.is_valid()
    assert "File must be in CSV format." in form.errors["file"]


def test_missing_required_columns() -> None:
    """Test form rejects CSV missing required headers."""
    invalid_csv = SimpleUploadedFile("test.csv", b"AccountNumber,CBAccountNumber\n12345,54321", content_type="text/csv")
    form = CSVUploadForm(files={"file": invalid_csv})
    assert not form.is_valid()
    assert "Missing required columns" in str(form.errors)


def test_source_and_trader_type_validation(sample_input_file: SimpleUploadedFile, mock_meo_database: MagicMock) -> None:
    """Test validation for Source and SYSTraderTranType columns."""
    modified_file = create_modified_csv(sample_input_file, {"Source": "90", "SYSTraderTranType": "5"})
    form = CSVUploadForm(files={"file": modified_file})
    assert not form.is_valid()
    assert "Row 1: 'Source' must be 80" in form.errors["file"][0]
    assert "Row 1: 'SYSTraderTranType' must be 4" in form.errors["file"][1]


def test_validate_nominal_analysis_account(sample_input_file: SimpleUploadedFile, mock_meo_database: MagicMock) -> None:
    """Test validation for nominal analysis account."""
    modified_file = create_modified_csv(sample_input_file,
                                        {"NominalAnalysisNominalAnalysisNarrative/1": "Invalid Name"})
    form = CSVUploadForm(files={"file": modified_file})
    assert not form.is_valid()
    assert form.errors["file"][0] == (
        "Row 1: 'AccountNumber' must match 'Sample Narrative' in 'NominalAnalysisNominalAnalysisNarrative/1'"
        ", but found 'Invalid Name'.")


def test_validate_nc_cc_dep_combination_against_meo_sage_account(
        sample_input_file: SimpleUploadedFile, mock_meo_database: MagicMock
) -> None:
    """Test validation for nominal analysis fields against MEO valid Sage accounts."""
    modified_file = create_modified_csv(
        sample_input_file,
        {
            "NominalAnalysisNominalCostCentre/1": "Invalid_CC",
            "NominalAnalysisNominalAccountNumber/1": "Invalid_Account"
        })

    form = CSVUploadForm(files={"file": modified_file})
    assert not form.is_valid()
    assert ("Row 1: 'NominalAnalysisNominalCostCentre/1' (Invalid_CC) is not a valid cost centre."
            in str(form.errors["file"][0]))