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 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 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)
