def should_persist_document_on_creation(self):
self.assertEqual(IntegerFieldDocument.objects.count(), 0)
fuel = MongoFuel(IntegerFieldDocument)
document = fuel._create()
self.assertEqual(IntegerFieldDocument.objects.count(), 1)
def should_not_save_document_if_user_does_not_want_to(self):
self.assertEqual(IntegerFieldDocument.objects.count(), 0)
fuel = MongoFuel(IntegerFieldDocument)
document = fuel._create(persists=False)
self.assertEqual(IntegerFieldDocument.objects.count(), 0)
def must_create_multiple_values_for_list_field_with_embedded_document_field(self):
fuel = MongoFuel(EmbeddedDocumentListFieldDocument)
document = fuel._create()
self.assertTrue(document.emb_list_field)
self.assertIsInstance(document.emb_list_field, list)
for emb_value in document.emb_list_field:
self.assertIsInstance(emb_value, UsersEmbeddedDocument)
def must_create_multiple_values_for_list_field_with_basic_fields(self):
fuel = MongoFuel(BasicListFieldDocument)
document = fuel._create()
self.assertTrue(document.int_list_field)
self.assertIsInstance(document.int_list_field, list)
for int_value in document.int_list_field:
self.assertIsInstance(int_value, int)
def must_return_correct_instance(self):
fuel = MongoFuel(UsersEmbeddedDocument)
embedded_document = fuel._create()
self.assertIsInstance(embedded_document, UsersEmbeddedDocument)
self.assertIsInstance(embedded_document.name, str)
self.assertIsInstance(embedded_document.age, int)
def must_create_multiple_values_for_list_field_with_reference_field(self):
fuel = MongoFuel(ReferenceListFieldDocument)
document = fuel._create()
self.assertTrue(document.ref_list_field)
self.assertIsInstance(document.ref_list_field, list)
for ref_value in document.ref_list_field:
self.assertIsInstance(ref_value, IntegerFieldDocument)
def should_create_more_than_one_document_and_dont_persist(self):
self.assertEqual(IntegerFieldDocument.objects.count(), 0)
documents = MongoFuel.create_many(IntegerFieldDocument, persists=False)
self.assertEqual(IntegerFieldDocument.objects.count(), 0)
self.assertGreaterEqual(len(documents), 2)
def should_not_create_fields_that_are_not_required(self):
doc = MongoFuel.create_one(NotRequiredFieldsDocument)
for field_name, field_obj in doc._fields.items():
if isinstance(field_obj, ObjectIdField):
continue
value = getattr(doc, field_name)
self.assertFalse(value)
def should_use_choices_values(self):
doc = MongoFuel.create_one(FieldsWithChoicesDocument)
for field_name, field_obj in doc._fields.items():
if isinstance(field_obj, ObjectIdField):
continue
value = getattr(doc, field_name)
self.assertIn(value, field_obj.choices)
def should_not_overrides_default_values(self):
doc = MongoFuel.create_one(DefaultFieldsDocument)
for field_name, field_obj in doc._fields.items():
if isinstance(field_obj, ObjectIdField):
continue
value = getattr(doc, field_name)
default = field_obj.default
self.assertEqual(value, default)
def test_classmethod_to_create_documents(self):
self.assertEqual(IntegerFieldDocument.objects.count(), 0)
document = MongoFuel.create_one(IntegerFieldDocument)
self.assertEqual(IntegerFieldDocument.objects.count(), 1)
def should_work_for_datetime_field(self):
fuel = MongoFuel(DatetimeFieldDocument)
document = fuel._create()
self.assertIsInstance(document.datetime_field, datetime)
def should_work_for_reference_field(self):
fuel = MongoFuel(ReferenceFieldDocument)
document = fuel._create()
self.assertIsInstance(document.reference, IntegerFieldDocument)
def should_work_for_embedded_document_field(self):
fuel = MongoFuel(UsersEmbeddedFieldDocument)
document = fuel._create()
self.assertIsInstance(document.user, UsersEmbeddedDocument)
def should_work_for_email_field(self):
fuel = MongoFuel(EmailFieldDocument)
document = fuel._create()
self.assertIsInstance(document.email_field, str)
def should_work_for_decimal_field(self):
fuel = MongoFuel(DecimalFieldDocument)
document = fuel._create()
self.assertIsInstance(document.decimal_field, Decimal)
def should_work_for_float_field(self):
fuel = MongoFuel(FloatFieldDocument)
document = fuel._create()
self.assertIsInstance(document.float_field, float)
def should_work_for_boolean_field(self):
fuel = MongoFuel(StringFieldDocument)
document = fuel._create()
self.assertIsInstance(document.str_field, str)
self.assertTrue(document.str_field)
def should_create_more_than_one_document(self):
self.assertEqual(IntegerFieldDocument.objects.count(), 0)
documents = MongoFuel.create_many(IntegerFieldDocument)
self.assertGreaterEqual(IntegerFieldDocument.objects.count(), 2)
def should_create_a_quantity_of_documents_that_user_want(self):
self.assertEqual(IntegerFieldDocument.objects.count(), 0)
documents = MongoFuel.create_many(IntegerFieldDocument, instances=3)
self.assertEqual(IntegerFieldDocument.objects.count(), 3)
def should_work_for_boolean_field(self):
fuel = MongoFuel(BooleanFieldDocument)
document = fuel._create()
self.assertIn(document.bool_field, [True, False])
def should_not_override_attrs_setted_by_the_user(self):
fuel = MongoFuel(IntegerFieldDocument)
document = fuel._create(int_field=3)
self.assertEqual(document.int_field, 3)
def should_work_for_int_field(self):
fuel = MongoFuel(IntegerFieldDocument)
document = fuel._create()
self.assertIsInstance(document.int_field, int)
