from django.test import Client, TestCase


class WeekdayApiTest(TestCase):
    """Tests for Weekday API."""

    def setUp(self):
        self.client = Client()
        self.endpoint = '/api'

    def create_weekday(self, name):
        query = '''
                mutation{
                    createWeekday(input: {name: "%s"}){
                        user{
                            id,
                            originalId,
                            name
                        }
                    }
                }
                ''' % (name)

        return self.client.post(self.endpoint, {'query': query}).json()

    def test_creation_of_weekday_object(self):
        response = self.create_weekday('tuesday')

        expected = {
            'createWeekday': {
                'weekday': {
                    'id': response['data']['createWeekday']['weekday']['id'],
                    'originalId': response['data']['createWeekday']['weekday']['originalId'],
                    'username': 'tuesday'
                }
            }
        }

        self.assertEqual(expected, response['data'])

Set up test for weekday api