import unittest
from probe_identity import evaluate


class IdentityTests(unittest.TestCase):
    def test_old_identity_and_root_never_pass(self):
        for uid in (0, 1000):
            self.assertFalse(evaluate(uid, 'a'*64, uid, 'a'*64)['identity_matches'])

    def test_wrong_host_or_user_fail(self):
        self.assertFalse(evaluate(1200, 'a'*64, 1201, 'a'*64)['identity_matches'])
        self.assertFalse(evaluate(1200, 'a'*64, 1200, 'b'*64)['identity_matches'])

    def test_matching_child_is_never_assignment_admission(self):
        result = evaluate(1200, 'a'*64, 1200, 'a'*64)
        self.assertTrue(result['identity_matches'])
        self.assertFalse(result['assignment_admitted'])


if __name__ == '__main__':
    unittest.main()