← Back to branch summary

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

« back to all changes in this revision

Viewing changes to src/maasserver/tests/test_forms_downloadprogress.py

merged upstream

Show diffs side-by-side

added added

removed removed

Lines of Context:
src/maasserver/tests/test_forms_downloadprogress.py
23
23
class TestDownloadProgressForm(MAASServerTestCase):
24
24
 
25
25
    def test_updates_instance(self):
26
 
        progress = factory.make_download_progress_incomplete(size=None)
 
26
        progress = factory.make_DownloadProgress_incomplete(size=None)
27
27
        new_bytes_downloaded = progress.bytes_downloaded + 1
28
28
        size = progress.bytes_downloaded + 2
29
29
        error = factory.make_string()
50
50
        self.assertFalse(form.is_valid())
51
51
 
52
52
    def test_get_download_returns_ongoing_download(self):
53
 
        progress = factory.make_download_progress_incomplete()
 
53
        progress = factory.make_DownloadProgress_incomplete()
54
54
 
55
55
        self.assertEqual(
56
56
            progress,
59
59
                progress.bytes_downloaded + 1))
60
60
 
61
61
    def test_get_download_recognises_start_of_new_download(self):
62
 
        nodegroup = factory.make_node_group()
 
62
        nodegroup = factory.make_NodeGroup()
63
63
        filename = factory.make_string()
64
64
        progress = DownloadProgressForm.get_download(nodegroup, filename, None)
65
65
        self.assertIsNotNone(progress)
70
70
    def test_get_download_returns_none_for_unknown_ongoing_download(self):
71
71
        self.assertIsNone(
72
72
            DownloadProgressForm.get_download(
73
 
                factory.make_node_group(), factory.make_string(), 1))
 
73
                factory.make_NodeGroup(), factory.make_string(), 1))