def test_charm_base_inheritance(self): """ get_sha256() should be implemented in the base class, and should use compute_sha256 to calculate the digest. """ sha256 = self.get_charm_sha256() bundle = CharmBundle(self.filename) self.assertEquals(bundle.get_sha256(), sha256)
def as_bundle(self): if self._temp_bundle is None: prefix = "%s-%d.charm." % (self.metadata.name, self.get_revision()) temp_file = tempfile.NamedTemporaryFile(prefix=prefix) self.make_archive(temp_file.name) self._temp_bundle = CharmBundle(temp_file.name) # Attach the life time of temp_file to self: self._temp_bundle_file = temp_file return self._temp_bundle
def test_competing_revisions(self): zf = zipfile.ZipFile(self.filename, "a") zf.writestr("revision", "999") data = yaml.load(zf.read("metadata.yaml")) data["revision"] = 303 zf.writestr("metadata.yaml", yaml.dump(data)) zf.close() charm = CharmBundle(self.filename) self.assertEquals(charm.get_revision(), 999)
def test_add_charm_and_publish(self): open_file_count = _count_open_files() yield self.publisher.add_charm(self.charm_id, self.charm) result = yield self.publisher.publish() self.assertEquals(_count_open_files(), open_file_count) children = yield self.client.get_children("/charms") self.assertEqual(children, [self.charm_key]) fh = yield self.storage.get(self.charm_storage_key) bundle = CharmBundle(fh) self.assertEqual(self.charm.get_sha256(), bundle.get_sha256()) self.assertEqual( result[0].bundle_url, "file://%s/%s" % (self.storage_dir, self.charm_storage_key))
def test_revision_in_metadata(self): filename = self.makeFile() zf_dst = zipfile.ZipFile(filename, "w") zf_src = zipfile.ZipFile(self.filename, "r") for name in zf_src.namelist(): if name == "revision": continue content = zf_src.read(name) if name == "metadata.yaml": data = yaml.load(content) data["revision"] = 303 content = yaml.dump(data) zf_dst.writestr(name, content) zf_src.close() zf_dst.close() charm = CharmBundle(filename) self.assertEquals(charm.get_revision(), 303)
def test_charm_download(self): """ Downloading a charm should store the charm locally. """ yield self.agent.startService() yield self.agent.download_charm(self.charm_state) checksum = self.charm.get_sha256() charm_id = local_charm_id(self.charm) charm_key = under.quote("%s:%s" % (charm_id, checksum)) charm_path = os.path.join(self.agent.charms_directory, charm_key) self.assertTrue(os.path.exists(charm_path)) bundle = CharmBundle(charm_path) self.assertEquals(bundle.get_revision(), self.charm.get_revision()) self.assertEquals(bundle.get_sha256(), checksum) self.assertIn("Downloading charm %s" % charm_id, self.output.getvalue())
def test_charm_download_file(self): """Downloading a charm should store the charm locally. """ charm, charm_state = yield self.publish_charm() charm_directory = self.makeDir() # Download the charm yield download_charm(self.client, charm_state.id, charm_directory) # Verify the downloaded copy checksum = charm.get_sha256() charm_id = local_charm_id(charm) charm_key = under.quote("%s:%s" % (charm_id, checksum)) charm_path = os.path.join(charm_directory, charm_key) self.assertTrue(os.path.exists(charm_path)) bundle = CharmBundle(charm_path) self.assertEquals(bundle.get_revision(), charm.get_revision()) self.assertEqual(checksum, bundle.get_sha256())
def download_charm(client, charm_id, charms_directory): charm_state_manager = CharmStateManager(client) charm_state = yield charm_state_manager.get_charm_state(charm_id) # Calculate local charm path checksum = yield charm_state.get_sha256() charm_key = under.quote("%s:%s" % (charm_state.id, checksum)) local_charm_path = os.path.join(charms_directory, charm_key) # Retrieve charm from provider storage link if charm_state.bundle_url.startswith("file://"): file_path = charm_state.bundle_url[len("file://"):] if not os.path.exists(file_path): raise FileNotFound(charm_state.bundle_url) shutil.copyfileobj(open(file_path), open(local_charm_path, "w")) else: try: yield downloadPage(charm_state.bundle_url, local_charm_path) except Error: raise FileNotFound(charm_state.bundle_url) returnValue(CharmBundle(local_charm_path))
def test_cannot_set_revision(self): charm = CharmBundle(self.filename) self.assertRaises(NotImplementedError, charm.set_revision, 123)
def test_initialization(self): bundle = CharmBundle(self.filename) self.assertEquals(bundle.path, self.filename)
def test_file_handle_as_path(self): sha256 = self.get_charm_sha256() fh = open(self.filename) bundle = CharmBundle(fh) self.assertEquals(bundle.get_sha256(), sha256)
def test_compute_sha256(self): sha256 = self.get_charm_sha256() bundle = CharmBundle(self.filename) self.assertEquals(bundle.compute_sha256(), sha256)
def test_as_bundle(self): bundle = CharmBundle(self.filename) self.assertEquals(bundle.as_bundle(), bundle)
def test_info(self): bundle = CharmBundle(self.filename) self.assertTrue(bundle.metadata is not None) self.assertTrue(isinstance(bundle.metadata, MetaData)) self.assertEquals(bundle.metadata.name, "dummy")
def test_bundled_config(self): """Make sure that config is accessible from a bundle.""" from juju.charm.tests.test_config import sample_yaml_data bundle = CharmBundle(self.filename) self.assertEquals(bundle.config.get_serialization_data(), sample_yaml_data)