def create_item(user_id=1): """ Method used to create and return a simple Item object. """ item = Item() item.filename = "sample.mp4" item.owner = User.objects.get(pk=user_id) item.save() return item
def setUp(self): """ Setup the database and index some objects. """ # create a new user user = User.objects.create_user('root', '*****@*****.**', 'root') user.is_superuser = True user.save() # create the search manager self.sm = ESManager() # create and index some items for i in range(100): item = Item() item.type = 'image' item.filename = 'test_' + str(i) item.description = 'test' item.mime_type = 'image/png' item.owner = user item.save() serializer = ItemSerializer(item) self.sm.create({'doc_type' : 'items', 'params' : serializer.data }) sleep(1)
def setUp(self): """ Method used to setup the test database """ # create a new user user = User.objects.create_user('root', '*****@*****.**', 'root') user.is_superuser = True user.save() token = Token.objects.create(user=user) # create Item objects for i in range(10): item = Item() item.type = 'image' item.filename = 'test_' + str(i) item.description = 'test' item.mime_type = 'image/png' item.owner = user item.save() # create Person objects for i in range(10): person = Person() person.category = 'person' person.first_name = 'John' person.last_name = 'Doe' + str(i) person.save() # create Tag objects for i in range(10): tag = Tag() tag.item = Item.objects.get(pk=i+1) tag.entity = Person.objects.get(pk=i+1) tag.type = 'test' tag.save()
def setUp(self): """ Method used to setup the test database """ # create a new user user = User.objects.create_user("root", "*****@*****.**", "root") user.is_superuser = True user.save() token = Token.objects.create(user=user) # create a new Item item = Item() item.filename = "" item.type = "image" item.owner = user item.save() # create a new Person entity = Entity() entity.category = "entity" entity.save() # create a new Tag tag = Tag() tag.item = item tag.entity = entity tag.type = "face" tag.save() # create a new DynamicTag dtag = DynamicTag() dtag.tag = tag dtag.start = 0 dtag.duration = 10000 dtag.save()
def setUp(self): """ Method used to setup the test database """ # create a new user user = User.objects.create_user('root', '*****@*****.**', 'root') user.is_superuser = True user.save() token = Token.objects.create(user=user) # create a new Item item = Item() item.filename = '' item.type = 'image' item.owner = user item.save() # create a new Person entity = Entity() entity.category = 'entity' entity.save() # create a new Tag tag = Tag() tag.item = item tag.entity = entity tag.type = 'face' tag.save()
def test_create(self): """ Method used to index a new object. """ item = Item() item.type = 'image' item.filename = 'test_' item.mime_type = 'image/png' item.owner = User.objects.all()[0] item.save() serializer = ItemSerializer(item) result = self.sm.create({'doc_type' : 'items', 'params' : serializer.data }) self.assertEqual(result['created'], True)
def setUp(self): """ Method used to setup the test database """ # create a new user user = User.objects.create_user('root', '*****@*****.**', 'root') user.is_superuser = True user.save() token = Token.objects.create(user=user) # create a new file item = Item() item.filename = '' item.type = 'image' item.file = File(os.path.join(settings.MEDIA_ROOT, 'tests', 'test_image', 'test.jpg')) item.owner = user item.save()
def test_create(self): """ Index a new Item """ user = User.objects.all()[0] client = APIClient() client.force_authenticate(user=user) item = Item() item.type = 'image' item.filename = 'test_' item.mime_type = 'image/png' item.owner = User.objects.all()[0] item.save() data = {"doc_type" : "items", "params": ItemSerializer(item).data } response = client.post('/api/search/items/create', data) self.assertEqual(response.status_code, status.HTTP_201_CREATED) self.assertEqual(json.loads(response.content)['created'], True)