package clarin.cmdi.componentregistry.impl.database;

import java.util.List;

import org.junit.Before;
import org.junit.Test;
import org.springframework.beans.factory.annotation.Autowired;

import clarin.cmdi.componentregistry.BaseUnitTest;
import clarin.cmdi.componentregistry.model.BaseDescription;
import clarin.cmdi.componentregistry.model.ComponentDescription;
import clarin.cmdi.componentregistry.model.Group;
import clarin.cmdi.componentregistry.model.GroupMembership;
import clarin.cmdi.componentregistry.model.Ownership;
import clarin.cmdi.componentregistry.model.ProfileDescription;
import clarin.cmdi.componentregistry.model.RegistryUser;
import clarin.cmdi.componentregistry.persistence.ComponentDao;
import clarin.cmdi.componentregistry.persistence.jpa.GroupDao;
import clarin.cmdi.componentregistry.persistence.jpa.OwnershipDao;
import clarin.cmdi.componentregistry.persistence.jpa.UserDao;
import clarin.cmdi.componentregistry.rest.DummyPrincipal;
import static org.junit.Assert.*;

public class GroupServiceImplTest extends BaseUnitTest {

    @Autowired
    GroupService groupService;
    @Autowired
    UserDao userDao;

    @Autowired
    GroupDao groupDao;
    
    @Autowired
    OwnershipDao ownershipDao;

    @Autowired
    ComponentDao componentDescriptionDao;

    RegistryUser user;
    RegistryUser user2;
    RegistryUser user3;

    protected BaseDescription makeTestProfile(String sid, boolean isPublic, long ownerId) {
	ProfileDescription profile = new ProfileDescription();
	profile.setDescription("some description");
	profile.setId(ProfileDescription.PROFILE_PREFIX+sid);
	profile.setName("profilename");
	Number id = componentDescriptionDao.insertDescription(profile, "someContent",
		isPublic, ownerId);
	return componentDescriptionDao.getById(id);
    }

    protected BaseDescription makeTestComponent(boolean isPublic,
	    long ownerId) {
	BaseDescription componentDescription = new BaseDescription();
	componentDescription.setDescription("some description");
	componentDescription.setId(ComponentDescription.COMPONENT_PREFIX+"4567");
	componentDescription.setName("componentname");
	Number id = componentDescriptionDao.insertDescription(
		componentDescription, "someContent", isPublic, ownerId);
	return componentDescriptionDao.getById(id);
    }

    @Before
    public void setupThisTest() {
	user = new RegistryUser();
	user.setName("Test User");
	user.setPrincipalName(DummyPrincipal.DUMMY_CREDENTIALS
		.getPrincipalName());
	user = userDao.saveAndFlush(user);
	user = userDao.getByPrincipalName(user.getPrincipalName());

	user2 = new RegistryUser();
	user2.setName("Test User 2");
	user2.setPrincipalName(DummyPrincipal.DUMMY_CREDENTIALS
		.getPrincipalName() + "2");
	user2 = userDao.saveAndFlush(user2);
	user2 = userDao.getByPrincipalName(user2.getPrincipalName());

	user3 = new RegistryUser();
	user3.setName("Test User 3");
	user3.setPrincipalName(DummyPrincipal.DUMMY_CREDENTIALS
		.getPrincipalName() + "3");
	user3 = userDao.saveAndFlush(user3);
	user3 = userDao.getByPrincipalName(user3.getPrincipalName());
    }

    @Test
    public void testCreateNewGroup_getGroupsOwnedByUser() {
	// Expect no groups for the current user
	List<Group> groups = groupService.getGroupsOwnedByUser(user
		.getPrincipalName());
	assertTrue(groups.isEmpty());

	// Make a group
	Group group1 = groupDao.findOne(groupService.createNewGroup("Group 1",
		user.getPrincipalName()));
	// and another one
	Group group2 = groupDao.findOne(groupService.createNewGroup("Group 2",
		user.getPrincipalName()));
	groups = groupService.getGroupsOwnedByUser(user.getPrincipalName());
	assertEquals(2, groups.size());

	assertTrue(groups.contains(group1));
	assertTrue(groups.contains(group2));
    }

    @Test(expected = ValidationException.class)
    public void testCreateDuplicateGroup() {
	// Make a group
	groupService.createNewGroup("Group X", user.getPrincipalName());
	// Make same group, again
	groupService.createNewGroup("Group X", user.getPrincipalName());
    }

    @Test
    public void testAccessToNonOwnedPrivateProfile() {
	// Make a private profile that belongs to someone else
	BaseDescription profile = makeTestProfile("4567", false, 9999);
	// Expect that user can't access profile
	boolean result = groupService
		.canUserAccessComponentEitherOnHisOwnOrThroughGroupMembership(
			user, profile);
	assertFalse(result);
    }

    @Test
    public void testAccessToNonOwnedPublicProfile() {
	// Make a public profile that belongs to someone else
	BaseDescription profile = makeTestProfile("4567", true, 9999);

	// Expect that user can access the profile
	boolean result = groupService
		.canUserAccessComponentEitherOnHisOwnOrThroughGroupMembership(
			user, profile);
	assertTrue(result);
    }

    @Test
    public void testAccessToOwnedPrivateProfile() {
	// Make a profile that belongs to someone else
	BaseDescription profile = makeTestProfile("4567", false, user.getId()
		.longValue());
	// Add an ownership to that profile to the current user
	Ownership ownership = new Ownership();
	ownership.setComponentId(profile.getId());
	ownership.setUserId(user.getId().longValue());
	groupService.addOwnership(ownership);

	// Expect that user can access the profile
	boolean result = groupService
		.canUserAccessComponentEitherOnHisOwnOrThroughGroupMembership(
			user, profile);
	assertTrue(result);
    }

    @Test
    public void testAccessViaGroupToPrivateProfile() {
	// Make a profile that belongs to someone else
	BaseDescription profile = makeTestProfile("4567", false, 9999);

	// Make a group that belongs to someone else
	Group group = groupDao.findOne(groupService.createNewGroup("Group 1",
		user2.getPrincipalName()));

	groupService.makeMember(user.getPrincipalName(),
		group.getName());
	// Make me a member of that group

	// Add an ownership to that profile to the group
	Ownership ownership = new Ownership();
	ownership.setComponentId(profile.getId());
	ownership.setGroupId(group.getId());
	groupService.addOwnership(ownership);

	// Expect that user can access the profile
	boolean result = groupService
		.canUserAccessComponentEitherOnHisOwnOrThroughGroupMembership(
			user, profile);
	assertTrue(result);
    }

    @Test
    public void testAccessToNonOwnedPrivateComponent() {
	// Make a private profile that belongs to someone else
	BaseDescription componentDescription = makeTestComponent(false,
		9999);

	// Expect that user can't access profile
	boolean result = groupService
		.canUserAccessComponentEitherOnHisOwnOrThroughGroupMembership(
			user, componentDescription);
	assertFalse(result);
    }

    @Test
    public void testAccessToNonOwnedPublicComponent() {
	// Make a public profile that belongs to someone else
	BaseDescription componentDescription = makeTestComponent(true,
		9999);

	// Expect that user can access the profile
	boolean result = groupService
		.canUserAccessComponentEitherOnHisOwnOrThroughGroupMembership(
			user, componentDescription);
	assertTrue(result);
    }

    @Test
    public void testAccessToOwnedPrivateComponent() {
	// Make a profile that belongs to someone else
	BaseDescription componentDescription = makeTestComponent(false,
		9999);

	// Add an ownership to that profile to the current user
	Ownership ownership = new Ownership();
	ownership.setComponentId(componentDescription.getId());
	ownership.setUserId(user.getId().longValue());
	groupService.addOwnership(ownership);

	// Expect that user can access the profile
	boolean result = groupService
		.canUserAccessComponentEitherOnHisOwnOrThroughGroupMembership(
			user, componentDescription);
	assertTrue(result);
    }

    @Test
    public void testAccessViaGroupToPrivateComponent() {
	// Make a profile that belongs to someone else
	BaseDescription componentDescription = makeTestComponent(false,
		user3.getId().longValue());

	// Make a group that belongs to someone else
	Group group = groupDao.findOne(groupService.createNewGroup("Group 1",
		user2.getPrincipalName()));

	// Make me a member of that group
	groupService.makeMember(user.getPrincipalName(),
		group.getName());

	// Add an ownership to that component to the group
	Ownership ownership = new Ownership();
	ownership.setComponentId(componentDescription.getId());
	ownership.setGroupId(group.getId());
	groupService.addOwnership(ownership);

	// Expect that user can access the profile
	boolean result = groupService
		.canUserAccessComponentEitherOnHisOwnOrThroughGroupMembership(
			user, componentDescription);
	assertTrue(result);
    }

    @Test
    public void testTransferComponentOwnershipFromUserToGroup() {

	// Make a group
	Group group = groupDao.findOne(groupService.createNewGroup("Group 1",
		user.getPrincipalName()));

	// Make user and user2 members of the same group
	groupService.makeMember(user.getPrincipalName(), group.getName());
	
	assertTrue(groupService.getGroupsOfWhichUserIsAMember(user.getPrincipalName()).contains(group));
	groupService.makeMember(user2.getPrincipalName(), group.getName());
	assertTrue(groupService.getGroupsOfWhichUserIsAMember(user2.getPrincipalName()).contains(group));

	// Make a component that belongs to user
	BaseDescription componentDescription = makeTestComponent(false,
		user.getId().longValue());

	// Just for the fun of it: check that user can access the component
	// while user2 can't
	boolean result = groupService
		.canUserAccessComponentEitherOnHisOwnOrThroughGroupMembership(
			user, componentDescription);
	assertTrue(result);
	result = groupService
		.canUserAccessComponentEitherOnHisOwnOrThroughGroupMembership(
			user2, componentDescription);
	assertFalse(result);

	// user transfers ownership of the component to his group
	groupService.transferItemOwnershipFromUserToGroup(
		user.getPrincipalName(), group.getName(), componentDescription.getId());
	
	List<Ownership> ownerships = (List<Ownership>)ownershipDao.findAll();

	// Check that user and user2 have access to the component...
	result = groupService
		.canUserAccessComponentEitherOnHisOwnOrThroughGroupMembership(
			user, componentDescription);
	assertTrue(result);
	result = groupService
		.canUserAccessComponentEitherOnHisOwnOrThroughGroupMembership(
			user2, componentDescription);
	assertTrue(result);

	// ... but someone unreleated not
	result = groupService
		.canUserAccessComponentEitherOnHisOwnOrThroughGroupMembership(
			user3, componentDescription);
	assertFalse(result);
    }

    @Test
    public void testtransferComponentOwnershipFromUserToGroup() {

	// Make a group
	Group group = groupDao.findOne(groupService.createNewGroup("Group 1",
		user.getPrincipalName()));

	// Make user and user2 members of the same group
	groupService.makeMember(user.getPrincipalName(), group.getName());
	
	assertTrue(groupService.getGroupsOfWhichUserIsAMember(user.getPrincipalName()).contains(group));
	groupService.makeMember(user2.getPrincipalName(), group.getName());
	assertTrue(groupService.getGroupsOfWhichUserIsAMember(user2.getPrincipalName()).contains(group));

	// Make a profile that belongs to user
	BaseDescription profileDescription= makeTestProfile("4567", false,
		user.getId().longValue());

	// Just for the fun of it: check that user can access the profile
	// while user2 can't
	boolean result = groupService
		.canUserAccessComponentEitherOnHisOwnOrThroughGroupMembership(
			user, profileDescription);
	assertTrue(result);
	result = groupService
		.canUserAccessComponentEitherOnHisOwnOrThroughGroupMembership(
			user2, profileDescription);
	assertFalse(result);

	// user transfers ownership of the component to his group
	groupService.transferItemOwnershipFromUserToGroup(
		user.getPrincipalName(), group.getName(),
		profileDescription.getId());
	
	List<Ownership> ownerships = (List<Ownership>)ownershipDao.findAll();

	// Check that user and user2 have access to the component...
	result = groupService
		.canUserAccessComponentEitherOnHisOwnOrThroughGroupMembership(
			user, profileDescription);
	assertTrue(result);
	result = groupService
		.canUserAccessComponentEitherOnHisOwnOrThroughGroupMembership(
			user2, profileDescription);
	assertTrue(result);

	// ... but someone unreleated not
	result = groupService
		.canUserAccessComponentEitherOnHisOwnOrThroughGroupMembership(
			user3, profileDescription);
	assertFalse(result);
    }

    @Test
    public void testTransferItemOwnershipFromUserToGroup() {
	//TODO: improve test by mixing in components and exclusing components/profiles
	// Make a group
	Group group = groupDao.findOne(groupService.createNewGroup("Group 1",
		user.getPrincipalName()));

	// Make user and user2 members of the same group
	groupService.makeMember(user.getPrincipalName(), group.getName());
	
	assertTrue(groupService.getGroupsOfWhichUserIsAMember(user.getPrincipalName()).contains(group));
	groupService.makeMember(user2.getPrincipalName(), group.getName());
	assertTrue(groupService.getGroupsOfWhichUserIsAMember(user2.getPrincipalName()).contains(group));

	// Make a profile that belongs to user
	BaseDescription profileDescription= makeTestProfile("4567", false,
		user.getId().longValue());

	// Just for the fun of it: check that user can access the profile
	// while user2 can't
	boolean result = groupService
		.canUserAccessComponentEitherOnHisOwnOrThroughGroupMembership(
			user, profileDescription);
	assertTrue(result);
	result = groupService
		.canUserAccessComponentEitherOnHisOwnOrThroughGroupMembership(
			user2, profileDescription);
	assertFalse(result);

	// user transfers ownership of the component to his group
	groupService.transferItemOwnershipFromUserToGroup(
		user.getPrincipalName(), group.getName(),
		profileDescription.getId());
	
	List<Ownership> ownerships = (List<Ownership>)ownershipDao.findAll();

	// Check that user and user2 have access to the component...
	result = groupService
		.canUserAccessComponentEitherOnHisOwnOrThroughGroupMembership(
			user, profileDescription);
	assertTrue(result);
	result = groupService
		.canUserAccessComponentEitherOnHisOwnOrThroughGroupMembership(
			user2, profileDescription);
	assertTrue(result);

	// ... but someone unreleated not
	result = groupService
		.canUserAccessComponentEitherOnHisOwnOrThroughGroupMembership(
			user3, profileDescription);
	assertFalse(result);
    }

    @Test
    public void testGetGroupsTheItemIsAMemberOf() {
	// Make a group
	Group group1 = groupDao.findOne(groupService.createNewGroup("Group 1",
		user.getPrincipalName()));

	// And another one which the item is not assigned to. will use to test exclusion
	Group group2 = groupDao.findOne(groupService.createNewGroup("Group 2",
		user.getPrincipalName()));

	// Make user members of the same group
	groupService.makeMember(user.getPrincipalName(), group1.getName());
	// Make user members of the same group
	groupService.makeMember(user.getPrincipalName(), group2.getName());

	// Make a profile that belongs to user
	BaseDescription profileDescription= makeTestProfile("4567", false,
		user.getId().longValue());

	// Make a profile that belongs to user, but that one will belong to group2
	BaseDescription profileDescription2= makeTestProfile("666", false,
		user.getId().longValue());

	// user transfers ownership of the profile to his group
	groupService.transferItemOwnershipFromUserToGroup(
		user.getPrincipalName(), group1.getName(),
		profileDescription.getId());

	// user transfers ownership of the profile to his group
	groupService.transferItemOwnershipFromUserToGroup(
		user.getPrincipalName(), group2.getName(),
		profileDescription2.getId());

	List<Group> groups = groupService.getGroupsTheItemIsAMemberOf(profileDescription.getId());
	assertEquals(1, groups.size());
	Group g = groups.get(0);
	assertEquals(group1.getId(), g.getId());
    }
}