Revision 3e5bbd85 snf-tools/synnefo_tools/burnin/images_tests.py

b/snf-tools/synnefo_tools/burnin/images_tests.py
166 166
        """Upload the image to Pithos"""
167 167
        self._set_pithos_account(self._get_uuid())
168 168
        self._create_pithos_container("burnin-images")
169
        file_size = os.path.getsize(self.temp_image_file)
169 170
        with open(self.temp_image_file, "r+b") as fin:
170 171
            self.clients.pithos.upload_object(self.temp_image_name, fin)
171 172

  
173
        # Verify quotas
174
        self._check_quotas(diskspace=file_size)
175

  
172 176
    def test_009_register_image(self):
173 177
        """Register image to Plankton"""
174 178
        location = "pithos://" + self._get_uuid() + \
......
192 196
        # Remove uploaded image
193 197
        self.info("Deleting uploaded image %s", self.temp_image_name)
194 198
        self.clients.pithos.del_object(self.temp_image_name)
199
        # Verify quotas
200
        file_size = os.path.getsize(self.temp_image_file)
201
        self._check_quotas(diskspace=-file_size)
195 202
        self.temp_image_name = None
196 203
        # Remove temp directory
197 204
        self.info("Deleting temp directory %s", self.temp_dir)

Also available in: Unified diff