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)
