DooFuS/test.py at master · pattonater/DooFuS · GitHub

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
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
## Testing program for DooFuS
##
## TEST MODULE WRITERS, PLEASE READ:
##      - A test returns 1 if successful, and 0 otherwise.
##      - Use traceback.print_tb to print the full trace of the exception.
##      - Catch all errors in your module so other tests will be performed!
##      - Add your module to the main method or it won't be called.
##
## To use: python3 test.py <test1> <test2> ... <testn>

import sys
import traceback

def _test_dfs():
    import modules.dfs.dfs as dfs

    prefix = "DFS test: ".ljust(15)

    try:
        # Test DFS instantiation
        file_system = dfs.DFS("test_dfs.json")

        # Add and remove and add file
        ls = file_system.list_files()

        file_system.add_file("newfile", "userA")
        if len(ls) == len(file_system.list_files()):
           print(prefix + "ERROR: list_files() did not reflect change in file list.")
           return 0

        file_system.delete_file("newfile")
        if len(file_system.list_files()) != 0:
            print(prefix + "ERROR: delete_file() did not remove file form file list.")
            return 0

        file_system.add_file("newfile", "userA")

        # Try to remove non-existent file
        try:
            file_system.delete_file("ff")
            print(prefix + "ERROR: was able to \"remove\" nonexistent file.")
            return 0
        except dfs.DFSRemoveFileError:
            pass

        # Try to add pre-existing file
        try:
            file_system.add_file("newfile", "userA")
            print(prefix + "ERROR: was able to add the same file twice.")
            return 0
        except dfs.DFSAddFileError:
            pass

    # All unintentional errors caught here
    except Exception as e:
        print(prefix + str(e))
        traceback.print_tb(e.__traceback__)
        return 0

    print(prefix + "SUCCESS")
    return 1

def _test_Msg_abstraction():
    prefix = "Msg abstraction test: ".ljust(15)
    try:
        from modules.network.msg import Msg
        if not Msg.VERIFY == "V":
            return 0

        return 1

    except Exception as e:
        print(prefix + str(e))
        traceback.print_tb(e.__traceback__)
        return 0

def _test_dfs_manager():
    prefix = "DFSManager: ".ljust(15)
    try:
        import modules.dfs.dfsmanager as manager
        import modules.dfs.dfs as dfs

        m = manager.DFSManager(None, "tester", "testmanagerdfs.json")

        m.acknowledge_replica("test.txt", "tester", "127.0.0.1")

        try:
            m.upload_file("test.txt")
            print(prefix + "ERROR: Was able to upload pre-existing file.")
            return 0
        except dfs.DFSAddFileError:
            pass
    except Exception as e:
        print(prefix + str(e))
        traceback.print_tb(e.__traceback__)
        return 0

    print(prefix + "SUCCESS")
    return 1


if __name__ == "__main__":
    outcome = 0

    for test in sys.argv[1:]:

        if test == "dfs":
            outcome += _test_dfs()

        if test == "Msg":
            outcome += _test_Msg_abstraction()

        if test == "dfsm":
            outcome += _test_dfs_manager()

        ## ADDITIONAL MODULES:
        #elif test == "othertestmodule":
        #	outcome += _other_test_module()

    result = "Success" if outcome == len(sys.argv[1:]) else "Failure"
    print(result + ": passed " + str(outcome) + " of " + str(len(sys.argv[1:])) + " testing modules.")