exp_portal/tests.py

from django.test import TestCase
from django.contrib.auth.hashers import make_password
from django.contrib.auth.models import User
from op_tasks.models import Product, Dataset, OpTask, TaskListItem, UserProfile

# Create your tests here.

class ExperimentTest(TestCase):

	def test_can_add_and_retrieve_database_items(self):
		dataset = Dataset()
		dataset.version = '1'
		dataset.name = 'test'
		dataset.save()

		product = Product()
		product.dataset = dataset
		product.url = 'http://www.cnn.com'
		product.team = 'CNN'
		product.name = 'CNN News'
		product.version = '1'
		product.instructions = 'http://www.cnn.com'
		product.save()

		saved_products = Product.objects.all()
		self.assertEqual(saved_products.count(),1)

	def test_can_count_completed_tasks(self):
		dataset = Dataset(version='1', name='test')
		dataset.save()

		task = OpTask(dataset=dataset, name='test_task', survey_url='test_url')
		task.save()

		user = User(email='john@test.com', password=make_password('paul'))
		user.save()

		userprofile = UserProfile(user=user)
		userprofile.save()

		product = Product(dataset=dataset, url='test_url')
		product.save()

		TaskListItem(
			userprofile=userprofile, 
			op_task=task,
			product=product,
			index=0,
			task_active=True,
			task_complete=False,
			exit_active=False,
			exit_complete=False).save()

		TaskListItem(
			userprofile=userprofile,
			op_task=task,
			product=product,
			index=1,
			task_active=False,
			task_complete=True,
			exit_active=False,
			exit_complete=False).save()

		saved_tasks = TaskListItem.objects.all()
		self.assertEqual(saved_tasks.count(), 2)

		self.assertEqual(saved_tasks.filter(task_active=True).count(), 1)
		self.assertEqual(saved_tasks.filter(task_complete=False).count(), 1)

	def test_can_edit_product_details(self):
		dataset = Dataset(version='1', name='test-dataset')
		dataset.save()

		product = Product(name='test-product', dataset=dataset, url='test-url')
		product.save()

		saved_products = Product.objects.all()
		self.assertEqual(saved_products.count(), 1)

		first_product = saved_products[0]
		self.assertEqual(first_product.name, 'test-product')
		first_product.name = 'new-name'
		first_product.save()

		new_saved_products = Product.objects.all()
		self.assertEqual(new_saved_products.count(), 1)

		new_first_product = new_saved_products[0]
		self.assertEqual(new_first_product.name, 'new-name')

	def test_can_find_product_by_primary_key(self):
		dataset = Dataset(version='1', name='test-dataset')
		dataset.save()

		product = Product(name='test-product', dataset=dataset, url='test-url')
		product.save()

		saved_products = Product.objects.all()
		first_product = saved_products[0]
		# print first_product.id

	# def test_can_find_users_by_experiment(self):
	# 	True