class ClientSerializerTestCase(TestCase):
    def test_create_client(self):
        data = {"country": "US",
                "name": "Test Company",
                "trade_name": "Test trade name",
                "website": "",
                "referral": "",
                "client_address": {"email": "test_public@email.com", "address_1": "Test address, 11", "address_2": "",
                                   "city": "New York", "state": "New York", "postal_code": "111111111", "country": "US",
                                   "phone": "", "fax": ""},
                "client_contact": {"first_name": "John", "last_name": "Doe", "email": "test_public@email.com",
                                   "phone": "22222222222", "role": ""}
                }
        user = UserFactory(username='user_1')
        serializer = ClientSerializer(data=data)
        serializer.is_valid(raise_exception=True)
        client = serializer.save(user=user)

        self.assertEqual(client.name, 'Test Company')
        self.assertEqual(client.trade_name, 'Test trade name')
        self.assertEqual(client.referral, '')
        self.assertEqual(client.current_onboarding_step, Client.ONBOARDING_TERM_OF_SERVICE)

        address = client.clientaddress_set.last()
        self.assertEqual(address.email, 'test_public@email.com')
        self.assertEqual(address.country, 'US')
        self.assertEqual(address.state, 'New York')
        self.assertEqual(address.city, 'New York')
        self.assertEqual(address.address_1, 'Test address, 11')
        self.assertEqual(address.postal_code, '111111111')

        client_contact = client.clientcontact_set.last()
        self.assertEqual(client_contact.first_name, 'John')
        self.assertEqual(client_contact.last_name, 'Doe')
        self.assertEqual(client_contact.email, 'test_public@email.com')
        self.assertEqual(client_contact.phone, '22222222222')

    def test_update_client(self):
        client = ClientFactory(name='Bill')
        from IPython import embed; embed()
        client_serializer = ClientSerializer(
            instance=client,
            data={'name': 'Bob'},
            partial=True
        )

        self.assertTrue(client_serializer.is_valid())

        client_serializer.is_valid(raise_exception=True)
        client_serializer.save()
        client.refresh_from_db()

        self.assertEqual(client_serializer.data['id'], client.id)
        self.assertEqual(client_serializer.data['name'], 'Bob')
        self.assertNotEqual(client_serializer.data['name'], 'Bill')