sentinel/tests/sentry/api/serializers/rest_framework/test_base.py

from django.contrib.contenttypes.models import ContentType
from rest_framework import serializers

from sentry.api.serializers.rest_framework.base import (
    CamelSnakeModelSerializer,
    CamelSnakeSerializer,
    camel_to_snake_case,
    convert_dict_key_case,
    snake_to_camel_case,
)
from sentry.testutils.cases import TestCase


class PersonSerializer(CamelSnakeSerializer):
    name = serializers.CharField()
    works_at = serializers.CharField()


class CamelSnakeSerializerTest(TestCase):
    def test_simple(self):
        serializer = PersonSerializer(data={"name": "Rick", "worksAt": "Sentry"})
        assert serializer.is_valid()
        assert serializer.data == {"name": "Rick", "works_at": "Sentry"}

    def test_error(self):
        serializer = PersonSerializer(data={"worksAt": None})
        assert not serializer.is_valid()
        assert serializer.errors == {
            "worksAt": ["This field may not be null."],
            "name": ["This field is required."],
        }


class ContentTypeSerializer(CamelSnakeModelSerializer):
    class Meta:
        model = ContentType
        fields = ["app_label", "model"]


class CamelSnakeModelSerializerTest(TestCase):
    def test_simple(self):
        serializer = ContentTypeSerializer(data={"appLabel": "hello", "model": "Something"})
        assert serializer.is_valid()
        assert serializer.data == {"model": "Something", "app_label": "hello"}

    def test_error(self):
        serializer = ContentTypeSerializer(data={"appLabel": None})
        assert not serializer.is_valid()
        assert serializer.errors == {
            "appLabel": ["This field may not be null."],
            "model": ["This field is required."],
        }


def test_convert_dict_key_case():
    camelData = {
        "appLabel": "hello",
        "model": "Something",
        "nestedList": [
            {"someObject": "someValue", "nestWithinNest": [{"anotherKey": "andAValue"}]}
        ],
    }
    snake_data = convert_dict_key_case(camelData, camel_to_snake_case)
    assert snake_data == {
        "app_label": "hello",
        "model": "Something",
        "nested_list": [
            {"some_object": "someValue", "nest_within_nest": [{"another_key": "andAValue"}]}
        ],
    }

    assert camelData == convert_dict_key_case(snake_data, snake_to_camel_case)