Mocking decorator on a base class doesn't work for subclasses

[Seluj78]

moto version: 4.2.9

With this example

from moto import mock_dynamodb
import unittest
import os

from botocore.session import Session
from pynamodb.attributes import NumberAttribute
from pynamodb.attributes import UnicodeAttribute
from pynamodb.constants import PAY_PER_REQUEST_BILLING_MODE
from pynamodb.models import Model
import requests


class Logs(Model):
    class Meta:
        table_name = "Logs"
        region = Session().get_config_variable("region")
        billing_mode = PAY_PER_REQUEST_BILLING_MODE

    id = NumberAttribute(hash_key=True)
    content = UnicodeAttribute(null=True)


@mock_dynamodb
class TestCase(unittest.TestCase):
    def setUp(self):
        os.environ["AWS_ACCESS_KEY_ID"] = "testing"
        os.environ["AWS_SECRET_ACCESS_KEY"] = "testing"
        os.environ["AWS_SECURITY_TOKEN"] = "testing"
        os.environ["AWS_SESSION_TOKEN"] = "testing"
        os.environ["AWS_DEFAULT_REGION"] = "us-east-1"

        Logs.create_table()


class TestLogs(TestCase):
    def test_logs(self):
        Logs(id=1, content="test").save()
        requests.post("http://127.0.0.1:7000/test")

You can see that the error raised is

pynamodb.exceptions.PutError: Failed to put item: An error occurred (UnrecognizedClientException) on request (KSN7UV9EQ61HRSN4G52AHA5EBVVV4KQNSO5AEMVJF66Q9ASUAAJG) on table (Logs) when calling the PutItem operation: The security token included in the request is invalid.

If we change the code to

class TestCase(unittest.TestCase):
    def setUp(self):
        os.environ["AWS_ACCESS_KEY_ID"] = "testing"
        os.environ["AWS_SECRET_ACCESS_KEY"] = "testing"
        os.environ["AWS_SECURITY_TOKEN"] = "testing"
        os.environ["AWS_SESSION_TOKEN"] = "testing"
        os.environ["AWS_DEFAULT_REGION"] = "us-east-1"

        with mock_dynamodb():
            Logs.create_table()


@mock_dynamodb
class TestLogs(TestCase):
    def setUp(self):
        super().setUp()

    def test_logs(self):
        Logs(id=1, content="test").save()

then it is working

My problem is that I have a big test suite and I don't want to have @mock_dynamodb above each test suite but rather once in my base class, but it isn't working.

[Seluj78]

Idealy I would love to be able to do

@mock_dynamodb
@mock_ec2
@mock_s3
...

Above my base class and not worry about it anywhere in my test suites

[bblommers]

Hi @Seluj78. This bug happens because the class-decorator only acts on the class as it is compiled. But in this scenario, we only know at runtime that we should be mocking the test_logs-method.

This is essentially the same problem as #4967, even if it's exposed differently. I'm going to close this as a duplicate - please follow that issue for updates.