← Back to branch summary

~rvb/maas/transaction-1.7-bug-1409852

« back to all changes in this revision

Viewing changes to src/maasserver/models/tests/test_filestorage.py

merged upstream

Show diffs side-by-side

added added

removed removed

Lines of Context:
src/maasserver/models/tests/test_filestorage.py
42
42
    def test_save_file_creates_storage(self):
43
43
        filename = factory.make_string()
44
44
        content = self.make_data()
45
 
        user = factory.make_user()
 
45
        user = factory.make_User()
46
46
        storage = FileStorage.objects.save_file(
47
47
            filename, BytesIO(content), user)
48
48
        self.assertEqual(
52
52
    def test_storage_can_be_retrieved(self):
53
53
        filename = factory.make_string()
54
54
        content = self.make_data()
55
 
        factory.make_file_storage(filename=filename, content=content)
 
55
        factory.make_FileStorage(filename=filename, content=content)
56
56
        storage = FileStorage.objects.get(filename=filename)
57
57
        self.assertEqual(
58
58
            (filename, content),
59
59
            (storage.filename, storage.content))
60
60
 
61
61
    def test_stores_binary_data(self):
62
 
        storage = factory.make_file_storage(content=sample_binary_data)
 
62
        storage = factory.make_FileStorage(content=sample_binary_data)
63
63
        self.assertEqual(sample_binary_data, storage.content)
64
64
 
65
65
    def test_overwrites_file(self):
67
67
        # reference to the old data gets overwritten with one to the new
68
68
        # data.
69
69
        filename = factory.make_name('filename')
70
 
        old_storage = factory.make_file_storage(
 
70
        old_storage = factory.make_FileStorage(
71
71
            filename=filename, content=self.make_data('old data'))
72
72
        new_data = self.make_data('new-data')
73
 
        new_storage = factory.make_file_storage(
 
73
        new_storage = factory.make_FileStorage(
74
74
            filename=filename, content=new_data)
75
75
        self.assertEqual(old_storage.filename, new_storage.filename)
76
76
        self.assertEqual(
78
78
 
79
79
    def test_key_gets_generated(self):
80
80
        # The generated system_id looks good.
81
 
        storage = factory.make_file_storage()
 
81
        storage = factory.make_FileStorage()
82
82
        self.assertEqual(len(storage.key), 36)
83
83
 
84
84
    def test_key_includes_random_part(self):
85
 
        storage1 = factory.make_file_storage()
86
 
        storage2 = factory.make_file_storage()
 
85
        storage1 = factory.make_FileStorage()
 
86
        storage2 = factory.make_FileStorage()
87
87
        self.assertNotEqual(storage1.key, storage2.key)