class TestRelease(TestCase): def setUp(self): self.release = Release(name='name', sha='sha') @patch_settings('STRETCH_DATA_DIR', '/stretch') def test_data_dir(self): self.assertEquals(self.release.data_dir, '/stretch/releases/sha') @patch('stretch.models.parser.Snapshot') @patch('stretch.models.Release.data_dir', Mock()) @patch('stretch.models.tarfile') @patch('stretch.models.utils') def test_get_snapshot(self, utils, tarfile, Snapshot): utils.temp_dir.return_value = '/temp_dir' tar_file = Mock() tarfile.open.return_value = tar_file self.release.data_dir = '/data' Snapshot.return_value = snapshot = Mock() self.assertEquals(self.release.get_snapshot(), snapshot) tarfile.open.assert_called_with('/data/snapshot.tar.gz') Snapshot.assert_called_with('/temp_dir')
def setUp(self): self.release = Release(name='name', sha='sha')