How to use checksum method in avocado

Best Python code snippet using avocado_python

httpchecksum.py

Source:httpchecksum.py

...166    def read(self, amt=None):167        chunk = super().read(amt=amt)168        self._checksum.update(chunk)169        if amt is None or (not chunk and amt > 0):170            self._validate_checksum()171        return chunk172    def _validate_checksum(self):173        if self._checksum.digest() != base64.b64decode(self._expected):174            error_msg = (175                f"Expected checksum {self._expected} did not match calculated "176                f"checksum: {self._checksum.b64digest()}"177            )178            raise FlexibleChecksumError(error_msg=error_msg)179def resolve_checksum_context(request, operation_model, params):180    resolve_request_checksum_algorithm(request, operation_model, params)181    resolve_response_checksum_algorithms(request, operation_model, params)182def resolve_request_checksum_algorithm(183    request,184    operation_model,185    params,186    supported_algorithms=None,187):188    http_checksum = operation_model.http_checksum189    algorithm_member = http_checksum.get("requestAlgorithmMember")190    if algorithm_member and algorithm_member in params:191        # If the client has opted into using flexible checksums and the192        # request supports it, use that instead of checksum required193        if supported_algorithms is None:194            supported_algorithms = _SUPPORTED_CHECKSUM_ALGORITHMS195        algorithm_name = params[algorithm_member].lower()196        if algorithm_name not in supported_algorithms:197            raise FlexibleChecksumError(198                error_msg="Unsupported checksum algorithm: %s" % algorithm_name199            )200        location_type = "header"201        if operation_model.has_streaming_input:202            # Operations with streaming input must support trailers.203            if request["url"].startswith("https:"):204                # We only support unsigned trailer checksums currently. As this205                # disables payload signing we'll only use trailers over TLS.206                location_type = "trailer"207        algorithm = {208            "algorithm": algorithm_name,209            "in": location_type,210            "name": "x-amz-checksum-%s" % algorithm_name,211        }212        if algorithm["name"] in request["headers"]:213            # If the header is already set by the customer, skip calculation214            return215        checksum_context = request["context"].get("checksum", {})216        checksum_context["request_algorithm"] = algorithm217        request["context"]["checksum"] = checksum_context218    elif operation_model.http_checksum_required or http_checksum.get(219        "requestChecksumRequired"220    ):221        # Otherwise apply the old http checksum behavior via Content-MD5222        checksum_context = request["context"].get("checksum", {})223        checksum_context["request_algorithm"] = "conditional-md5"224        request["context"]["checksum"] = checksum_context225def apply_request_checksum(request):226    checksum_context = request.get("context", {}).get("checksum", {})227    algorithm = checksum_context.get("request_algorithm")228    if not algorithm:229        return230    if algorithm == "conditional-md5":231        # Special case to handle the http checksum required trait232        conditionally_calculate_md5(request)233    elif algorithm["in"] == "header":234        _apply_request_header_checksum(request)235    elif algorithm["in"] == "trailer":236        _apply_request_trailer_checksum(request)237    else:238        raise FlexibleChecksumError(239            error_msg="Unknown checksum variant: %s" % algorithm["in"]240        )241def _apply_request_header_checksum(request):242    checksum_context = request.get("context", {}).get("checksum", {})243    algorithm = checksum_context.get("request_algorithm")244    location_name = algorithm["name"]245    if location_name in request["headers"]:246        # If the header is already set by the customer, skip calculation247        return248    checksum_cls = _CHECKSUM_CLS.get(algorithm["algorithm"])249    digest = checksum_cls().handle(request["body"])250    request["headers"][location_name] = digest251def _apply_request_trailer_checksum(request):252    checksum_context = request.get("context", {}).get("checksum", {})253    algorithm = checksum_context.get("request_algorithm")254    location_name = algorithm["name"]255    checksum_cls = _CHECKSUM_CLS.get(algorithm["algorithm"])256    headers = request["headers"]257    body = request["body"]258    if location_name in headers:259        # If the header is already set by the customer, skip calculation260        return261    headers["Transfer-Encoding"] = "chunked"262    headers["Content-Encoding"] = "aws-chunked"263    headers["X-Amz-Trailer"] = location_name264    content_length = determine_content_length(body)265    if content_length is not None:...

test_computechecksums.py

Source:test_computechecksums.py

...10    pp = ComputeChecksums()11    pp.comment_index = 012    return pp13class TestComputeChecksums(BaseTestPreprocessor):14    def test_code_cell_no_checksum(self, preprocessor):15        """Test that no checksum is computed for a regular code cell"""16        cell, _ = preprocessor.preprocess_cell(17            create_code_cell(), {}, 0)18        assert "nbgrader" not in cell.metadata or "checksum" not in cell.metadata.nbgrader19    def test_text_cell_no_checksum(self, preprocessor):20        """Test that no checksum is computed for a regular text cell"""21        cell, _ = preprocessor.preprocess_cell(22            create_text_cell(), {}, 0)23        assert "nbgrader" not in cell.metadata or "checksum" not in cell.metadata.nbgrader24    def test_checksum_grade_cell_type(self, preprocessor):25        """Test that the checksum is computed for grade cells of different cell types"""26        cell1 = create_grade_cell("", "code", "foo", 1)27        cell1 = preprocessor.preprocess_cell(cell1, {}, 0)[0]28        cell2 = create_grade_cell("", "markdown", "foo", 1)29        cell2 = preprocessor.preprocess_cell(cell2, {}, 0)[0]30        assert cell1.metadata.nbgrader["checksum"] == compute_checksum(cell1)31        assert cell2.metadata.nbgrader["checksum"] == compute_checksum(cell2)32        assert cell1.metadata.nbgrader["checksum"] != cell2.metadata.nbgrader["checksum"]33    def test_checksum_solution_cell_type(self, preprocessor):34        """Test that the checksum is computed for solution cells of different cell types"""35        cell1 = create_solution_cell("", "code", "foo")36        cell1 = preprocessor.preprocess_cell(cell1, {}, 0)[0]37        cell2 = create_solution_cell("", "markdown", "foo")38        cell2 = preprocessor.preprocess_cell(cell2, {}, 0)[0]39        assert cell1.metadata.nbgrader["checksum"] == compute_checksum(cell1)40        assert cell2.metadata.nbgrader["checksum"] == compute_checksum(cell2)41        assert cell1.metadata.nbgrader["checksum"] != cell2.metadata.nbgrader["checksum"]42    def test_checksum_locked_cell_type(self, preprocessor):43        """Test that the checksum is computed for locked cells"""44        cell1 = create_locked_cell("", "code", "foo")45        cell1 = preprocessor.preprocess_cell(cell1, {}, 0)[0]46        cell2 = create_locked_cell("", "markdown", "foo")47        cell2 = preprocessor.preprocess_cell(cell2, {}, 0)[0]48        assert cell1.metadata.nbgrader["checksum"] == compute_checksum(cell1)49        assert cell2.metadata.nbgrader["checksum"] == compute_checksum(cell2)50        assert cell1.metadata.nbgrader["checksum"] != cell2.metadata.nbgrader["checksum"]51    def test_checksum_points(self, preprocessor):52        """Test that the checksum is computed for grade cells with different points"""53        cell1 = create_grade_cell("", "code", "foo", 1)54        cell1 = preprocessor.preprocess_cell(cell1, {}, 0)[0]55        cell2 = create_grade_cell("", "code", "foo", 2)56        cell2 = preprocessor.preprocess_cell(cell2, {}, 0)[0]57        assert cell1.metadata.nbgrader["checksum"] == compute_checksum(cell1)58        assert cell2.metadata.nbgrader["checksum"] == compute_checksum(cell2)59        assert cell1.metadata.nbgrader["checksum"] != cell2.metadata.nbgrader["checksum"]60    def test_checksum_grade_id(self, preprocessor):61        """Test that the checksum is computed for grade cells with different ids"""62        cell1 = create_grade_cell("", "code", "foo", 1)63        cell1 = preprocessor.preprocess_cell(cell1, {}, 0)[0]64        cell2 = create_grade_cell("", "code", "bar", 1)65        cell2 = preprocessor.preprocess_cell(cell2, {}, 0)[0]66        assert cell1.metadata.nbgrader["checksum"] == compute_checksum(cell1)67        assert cell2.metadata.nbgrader["checksum"] == compute_checksum(cell2)68        assert cell1.metadata.nbgrader["checksum"] != cell2.metadata.nbgrader["checksum"]69    def test_checksum_grade_source(self, preprocessor):70        """Test that the checksum is computed for grade cells with different sources"""71        cell1 = create_grade_cell("a", "code", "foo", 1)72        cell1 = preprocessor.preprocess_cell(cell1, {}, 0)[0]73        cell2 = create_grade_cell("b", "code", "foo", 1)74        cell2 = preprocessor.preprocess_cell(cell2, {}, 0)[0]75        assert cell1.metadata.nbgrader["checksum"] == compute_checksum(cell1)76        assert cell2.metadata.nbgrader["checksum"] == compute_checksum(cell2)77        assert cell1.metadata.nbgrader["checksum"] != cell2.metadata.nbgrader["checksum"]78    def test_checksum_solution_source(self, preprocessor):79        """Test that the checksum is computed for solution cells with different sources"""80        cell1 = create_solution_cell("a", "code", "foo")81        cell1 = preprocessor.preprocess_cell(cell1, {}, 0)[0]82        cell2 = create_solution_cell("b", "code", "foo")83        cell2 = preprocessor.preprocess_cell(cell2, {}, 0)[0]84        assert cell1.metadata.nbgrader["checksum"] == compute_checksum(cell1)85        assert cell2.metadata.nbgrader["checksum"] == compute_checksum(cell2)86        assert cell1.metadata.nbgrader["checksum"] != cell2.metadata.nbgrader["checksum"]87    def test_checksum_grade_and_solution(self, preprocessor):88        """Test that a checksum is created for grade cells that are also solution cells"""89        cell1 = create_grade_cell("", "markdown", "foo", 1)90        cell1 = preprocessor.preprocess_cell(cell1, {}, 0)[0]91        cell2 = create_grade_cell("", "markdown", "foo", 1)92        cell2.metadata.nbgrader["solution"] = True93        cell2 = preprocessor.preprocess_cell(cell2, {}, 0)[0]94        assert cell1.metadata.nbgrader["checksum"] == compute_checksum(cell1)95        assert cell2.metadata.nbgrader["checksum"] == compute_checksum(cell2)...

ChecksumEncoder.py

Source:ChecksumEncoder.py

1from Classes.CPPDefs import CPPDefs2from Classes.Debugger import Debugger3from Classes.Logic.LogicLong import LogicLong4class ChecksumEncoder:5    def __init__(self):6        self.checksum = 07        self.checksum2 = 08        self.checksumEnabled = True9    def destruct(self):10        self.checksum = 011        self.checksum2 = 012        self.checksumEnabled = True13    def enableCheckSum(self, state):14        if not self.checksumEnabled or state:15            if not self.checksumEnabled and state:16                self.checksum = self.checksum217            self.checksumEnabled = state18        else:19            self.checksum2 = self.checksum20            self.checksumEnabled = False21    def equals(self, checksum_instance):22        if not checksum_instance:23            return False24        if not checksum_instance.checksumEnabled:25            checksum = checksum_instance.checksum26        else:27            checksum2 = checksum_instance.checksum228        if not self.checksumEnabled:29            checksum = self.checksum30        else:31            checksum2 = self.checksum232        return checksum == checksum233    def getCheckSum(self):34        if not self.checksumEnabled:35            checksum = self.checksum236        else:37            checksum = self.checksum38        return checksum39    @staticmethod40    def hashCode():41        Debugger.error("ChecksumEncoder hashCode not designed")42        return 4243    @staticmethod44    def isByteStream():45        return False46    def isCheckSumEnabled(self):47        return self.checksumEnabled48    @staticmethod49    def isCheckSumOnlyMode():50        return True51    def resetCheckSum(self):52        self.checksum = 053    def writeBoolean(self, value):54        if value: integer = 1355        else: integer = 756        self.checksum = integer + CPPDefs.__ROR4__(self.checksum, 31)57    def writeByte(self, value):58        self.checksum = CPPDefs.__ROR4__(self.checksum, 31) + value + 1159    def writeBytes(self, value, length):60        if value: integer = length + 3861        else: integer = 3762        self.checksum = CPPDefs.__ROR4__(self.checksum, 31)63    def writeInt8(self, value):64        if value + 0x80 >= 0x100:65            Debugger.error("")66        self.checksum = CPPDefs.__ROR4__(self.checksum, 31) + value + 1167    def writeInt16(self, value):68        if value + 0x8000 >= 0x10000:69            Debugger.error("")70        self.checksum = CPPDefs.__ROR4__(self.checksum, 31) + value + 1971    def writeInt24(self, value):72        if value + 0x800000 >= 0x1000000:73            Debugger.error("")74        self.checksum = (value & 0xFFFFFF) + CPPDefs.__ROR4__(self.checksum, 31) + value + 2175    def writeInt(self, value):76        self.checksum = CPPDefs.__ROR4__(self.checksum, 31) + value + 977    @staticmethod78    def writeLong(bytestream, logicLong):79        logicLong.encode(bytestream)80    def writeLongLong(self, logicLong):81        self.checksum = logicLong.getLowerInt() + CPPDefs.__ROR4__(logicLong.getHigherInt() + CPPDefs.__ROR4__(self.checksum, 31) + 67, 31) + 9182    def writeShort(self, value):83        self.checksum = CPPDefs.__ROR4__(self.checksum, 31) + value + 1984    def writeString(self, value):85        checksum = CPPDefs.__ROR4__(self.checksum, 31)86        if value:87            self.checksum = checksum + len(value) + 2888        else:89            self.checksum = checksum + 2790    def writeStringReference(self, value):91        self.checksum = len(value) + CPPDefs.__ROR4__(self.checksum, 31) + 3892    def writeVInt(self, value):93        self.checksum = value + CPPDefs.__ROR4__(self.checksum, 31) + 3394    def writeVLong(self, high, low):...

Automation Testing Tutorials

Learn to execute automation testing from scratch with LambdaTest Learning Hub. Right from setting up the prerequisites to run your first automation test, to following best practices and diving deeper into advanced test scenarios. LambdaTest Learning Hubs compile a list of step-by-step guides to help you be proficient with different test automation frameworks i.e. Selenium, Cypress, TestNG etc.