def new(self, request):
try:
q = request.query_params
q_d = {k: v[0] if len(v) == 1 else v for k, v in q.lists()}
customer = Customer(**q_d)
customer.save()
return Response(data=CustomerSerializer(customer).data)
except:
print(traceback.format_exc())
return Response(data={"is": "false"})
def create(self, validated_data):
"""
Create company and additional info
:param validated_data:
:return:
"""
validated_data.update({'lead': True})
customer = Customer(**validated_data)
customer.save()
return customer
def save(self, validated_user_data):
user = Customer(
first_name=validated_user_data['first_name'],
last_name=validated_user_data['last_name'],
street_address=validated_user_data['street_address'],
city=validated_user_data['city'],
state_province=validated_user_data['state_province'],
email_address=validated_user_data['email_address'],
)
user.save()
return user
class ModelTestCase(TestCase):
"""This class defines the test suite for the bucketlist model."""
def setUp(self):
"""Define the test client and other test variables."""
self.customer_name = "Write world class code"
self.customer = Customer(name=self.customer_name)
def test_model_can_create_a_bucketlist(self):
"""Test the bucketlist model can create a bucketlist."""
old_count = Customer.objects.count()
self.customer.save()
new_count = Customer.objects.count()
self.assertNotEqual(old_count, new_count)