class FileContentFactory(factory.DjangoModelFactory): class Meta: model = models.FileContent django_get_or_create = ("sha1", ) sha1 = utils.sha1(FILE_CONTENTS) contents = utils.compressed_str(FILE_CONTENTS)
def test_file_serializer(self): playbook = factories.PlaybookFactory() serializer = serializers.FileSerializer( data={"path": "/path/playbook.yml", "content": factories.FILE_CONTENTS, "playbook": playbook.id} ) serializer.is_valid() file = serializer.save() file.refresh_from_db() self.assertEqual(file.content.sha1, utils.sha1(factories.FILE_CONTENTS))