コード例 #1
0
    def test_add_value(self):
        with pytest.raises(TypeError) as err:
            containers = Containers()
            containers.add_value("rubbish")
        assert "must be of type" in str(err.value)  # nosec

        with pytest.raises(TypeError) as err:
            containers = Containers()
            containers.set_num_shards("rubbish")
        # pylint: disable=line-too-long
        assert "num_shards parameter must be of type int" in str(
            err.value)  # nosec
        # pylint: enable=line-too-long

        containers = Containers()
        containers.add_value(self.vos_container1)
        containers.add_value(self.vos_container2)

        containers.set_num_shards(500)

        raw_container1 = self.test_data.create_default_container(csum_gran=300)
        raw_container2 = self.test_data.create_default_container(csum_gran=400)
        want = {
            "num_shards": 500,
            "containers": [raw_container1, raw_container2]
        }
        assert want == containers.dump()  # nosec
コード例 #2
0
ファイル: util.py プロジェクト: xyuan/daos
    def _get_yaml_from_dfs(self, fse, use_average=False):
        dfs_sb = get_dfs_sb_obj()

        if use_average:
            dfs = fse.get_dfs_average()
        else:
            dfs = fse.get_dfs()

        container = dfs.get_container()
        container.add_value(dfs_sb)
        containers = Containers()
        containers.add_value(container)
        containers.set_num_shards(self._args.num_shards)

        return containers.dump()