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)