# Suites of tests should have classes named Test<SuiteName>.
# For something to be run before each test, define a function named setUp.
# For something to be run after each test, define a function named tearDown.
# Individual units should have the name <unit>_test.

from bandage.tests import *

from bandage import model

class TestUser(TestController):
    def setUp(self):
        self.user = model.User()
        model.meta.Session.add(self.user)
        model.meta.Session.commit()

    def existance_test(self):
        assert self.user.id != None

    def get_by_id_test(self):
        found_user = model.User.get_by_id(self.user.id)
        assert found_user == self.user

    def tearDown(self):
        model.meta.Session.delete(self.user)
        model.meta.Session.commit()

class TestSubmission(TestController):
    def setUp(self):
        self.submission = model.Submission()
        self.submission.name = "test"
        model.meta.Session.add(self.submission)
        model.meta.Session.commit()

    def existance_test(self):
        assert self.submission._id != None

    def get_by_id_test(self):
        found_submission = model.Submission.get_by_id(self.submission._id)
        assert found_submission == self.submission

    def duplicate_test(self):
	self.submission2 = model.Submission()
        self.submission2.name = "test"
        model.meta.Session.add(self.submission2)
        model.meta.Session.commit()
        assert self.submission._id != self.submission2._id
        assert self.submission.name != self.submission2.name
	model.meta.Session.delete(self.submission2)
	model.meta.Session.commit()

    def tearDown(self):
        model.meta.Session.delete(self.submission)
        model.meta.Session.commit()