KaneAI - World’s First E2E Software Testing Agent.Book a Demo Join TestMu Conference

How to use mock_cloudwatch_client method in localstack

Best Python code snippet using localstack_python

cloudwatch_utils_test.py

Source:cloudwatch_utils_test.py

...78                }79            )80    return ret81@pytest.fixture(name="mock_cloudwatch_client")82def _mock_cloudwatch_client() -> MagicMock:83    mock_client = MagicMock()84    mock_client.get_metric_data = MagicMock(side_effect=fake_get_metric_data)85    return mock_client86def test_get_metrics(mock_cloudwatch_client: MagicMock) -> None:87    metrics = _get_metrics_from_cloudwatch(88        "db_id",89        mock_cloudwatch_client,90        ["WriteIOPS", "CPUUtilization", "ReplicaLag", "CPUCreditUsage"],91        UTC_NOW,92    )93    expected_data = {94        "WriteIOPS": 1234,95        "CPUUtilization": 5555,96        "CPUCreditUsage": 94041,...

test_cloudwatch_handler.py

Source:test_cloudwatch_handler.py

1import unittest2import os3from botocore.stub import Stubber4from matrix.common.aws.cloudwatch_handler import CloudwatchHandler, MetricName5class TestCloudwatchHandler(unittest.TestCase):6    """7    Environment variables are set in tests/unit/__init__.py8    """9    def setUp(self):10        self.deploment_stage = os.environ["DEPLOYMENT_STAGE"]11        self.handler = CloudwatchHandler()12        self.mock_cloudwatch_client = Stubber(self.handler._client)13    def test_put_metric_data(self):14        metric_data = {'MetricName': MetricName.REQUEST.value, 'Value': 1}15        expected_params = {'MetricData': [metric_data],16                           'Namespace': f"dcp-matrix-service-{self.deploment_stage}"}17        self.mock_cloudwatch_client.add_response('put_metric_data', {}, expected_params)18        self.mock_cloudwatch_client.activate()19        self.handler.put_metric_data(MetricName.REQUEST, 1)20    def test_put_metric_data_with_dimensions(self):21        metric_data = {'MetricName': MetricName.REQUEST.value, 'Value': 1, 'Dimensions': [{'Name': "a", 'Value': "b"}]}22        expected_params = {'MetricData': [metric_data],23                           'Namespace': f"dcp-matrix-service-{self.deploment_stage}"}24        self.mock_cloudwatch_client.add_response('put_metric_data', {}, expected_params)25        self.mock_cloudwatch_client.activate()...

test_app.py

Source:test_app.py

1import sys2import pytest3import json4sys.path.append("..")5from app import app6from flask import Flask, jsonify, json, Response, request, Request7from unittest import mock8from moto import mock_cloudwatch9TEST_EXPECTED_PREDICT_RESPONSE_NO_INPUT = "400 Bad Request"10TEST_IMAGE_URL = 'https://upload.wikimedia.org/wikipedia/commons/9/93/Golden_Retriever_Carlos_%2810581910556%29.jpg'11TEST_EXPECTED_CLASS_NAME = 'Dog'12@pytest.fixture13def client():14    app.app.config['TESTING'] = True15    client = app.app.test_client()16    yield client17@mock.patch('app.app.cloudwatch_client')18def test_app_predict_no_input(mock_cloudwatch_client, client):19    response = client.post('/predict')20    assert TEST_EXPECTED_PREDICT_RESPONSE_NO_INPUT in json.loads(response.data)['error']21@mock.patch('app.app.cloudwatch_client')22def test_app_predict_url_input(mock_cloudwatch_client, client):23    response = client.post('/predict', data={'url': TEST_IMAGE_URL})...

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.