File: TestDigitalFactoryFileModel.py

package info (click to toggle)
cura 5.0.0-6
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 122,888 kB
  • sloc: python: 44,572; sh: 81; xml: 32; makefile: 16
file content (73 lines) | stat: -rw-r--r-- 2,807 bytes parent folder | download | duplicates (2)
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
from pathlib import Path

from src.DigitalFactoryFileModel import DigitalFactoryFileModel
from src.DigitalFactoryFileResponse import DigitalFactoryFileResponse


file_1 = DigitalFactoryFileResponse(client_id = "client_id_1",
                                    content_type = "zomg",
                                    file_name = "file_1.3mf",
                                    file_id = "file_id_1",
                                    library_project_id = "project_id_1",
                                    status = "test",
                                    user_id = "user_id_1",
                                    username = "username_1",
                                    uploaded_at = "2021-04-07T10:33:25.000Z")

file_2 = DigitalFactoryFileResponse(client_id ="client_id_2",
                                    content_type = "zomg",
                                    file_name = "file_2.3mf",
                                    file_id = "file_id_2",
                                    library_project_id = "project_id_2",
                                    status = "test",
                                    user_id = "user_id_2",
                                    username = "username_2",
                                    uploaded_at = "2021-02-06T09:33:22.000Z")

file_wtf = DigitalFactoryFileResponse(client_id ="client_id_1",
                                      content_type = "zomg",
                                      file_name = "file_3.wtf",
                                      file_id = "file_id_3",
                                      library_project_id = "project_id_1",
                                      status = "test",
                                      user_id = "user_id_1",
                                      username = "username_1",
                                      uploaded_at = "2021-04-06T12:33:25.000Z")


def test_setFiles():
    model = DigitalFactoryFileModel()

    assert model.count == 0

    model.setFiles([file_1, file_2])
    assert model.count == 2

    assert model.getItem(0)["fileName"] == "file_1.3mf"
    assert model.getItem(1)["fileName"] == "file_2.3mf"


def test_clearProjects():
    model = DigitalFactoryFileModel()
    model.setFiles([file_1, file_2])
    model.clearFiles()
    assert model.count == 0


def test_setProjectMultipleTimes():
    model = DigitalFactoryFileModel()
    model.setFiles([file_1, file_2])
    model.setFiles([file_2])
    assert model.count == 1
    assert model.getItem(0)["fileName"] == "file_2.3mf"


def test_setFilter():
    model = DigitalFactoryFileModel()

    model.setFiles([file_1, file_2, file_wtf])
    model.setFilters({"file_name": lambda x: Path(x).suffix[1:].lower() in ["3mf"]})
    assert model.count == 2

    model.clearFilters()
    assert model.count == 3