/*
* LinShare is an open source filesharing software, part of the LinPKI software
* suite, developed by Linagora.
*
* Copyright (C) 2015 LINAGORA
*
* This program is free software: you can redistribute it and/or modify it under
* the terms of the GNU Affero General Public License as published by the Free
* Software Foundation, either version 3 of the License, or (at your option) any
* later version, provided you comply with the Additional Terms applicable for
* LinShare software by Linagora pursuant to Section 7 of the GNU Affero General
* Public License, subsections (b), (c), and (e), pursuant to which you must
* notably (i) retain the display of the “LinShare™” trademark/logo at the top
* of the interface window, the display of the “You are using the Open Source
* and free version of LinShare™, powered by Linagora © 2009–2015. Contribute to
* Linshare R&D by subscribing to an Enterprise offer!” infobox and in the
* e-mails sent with the Program, (ii) retain all hypertext links between
* LinShare and linshare.org, between linagora.com and Linagora, and (iii)
* refrain from infringing Linagora intellectual property rights over its
* trademarks and commercial brands. Other Additional Terms apply, see
* <http://www.linagora.com/licenses/> for more details.
*
* This program is distributed in the hope that it will be useful, but WITHOUT
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
* FOR A PARTICULAR PURPOSE. See the GNU Affero General Public License for more
* details.
*
* You should have received a copy of the GNU Affero General Public License and
* its applicable Additional Terms for LinShare along with this program. If not,
* see <http://www.gnu.org/licenses/> for the GNU Affero General Public License
* version 3 and <http://www.linagora.com/licenses/> for the Additional Terms
* applicable to LinShare software.
*/
package org.linagora.linshare.repository.hibernate;
import static org.junit.Assert.*;
import java.util.Date;
import java.util.GregorianCalendar;
import java.util.List;
import org.junit.Before;
import org.junit.Test;
import org.linagora.linshare.core.domain.constants.ContainerQuotaType;
import org.linagora.linshare.core.domain.entities.AbstractDomain;
import org.linagora.linshare.core.domain.entities.Account;
import org.linagora.linshare.core.domain.entities.DomainQuota;
import org.linagora.linshare.core.domain.entities.ContainerQuota;
import org.linagora.linshare.core.domain.entities.Quota;
import org.linagora.linshare.core.domain.entities.User;
import org.linagora.linshare.core.repository.AccountQuotaRepository;
import org.linagora.linshare.core.repository.AccountRepository;
import org.linagora.linshare.core.repository.DomainQuotaRepository;
import org.linagora.linshare.core.repository.ContainerQuotaRepository;
import org.linagora.linshare.core.repository.UserRepository;
import org.linagora.linshare.service.LoadingServiceTestDatas;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.test.context.ContextConfiguration;
import org.springframework.test.context.junit4.AbstractTransactionalJUnit4SpringContextTests;
@ContextConfiguration(locations = {
"classpath:springContext-test.xml",
"classpath:springContext-datasource.xml",
"classpath:springContext-repository.xml"})
public class QuotaRepositoryImplTest
extends AbstractTransactionalJUnit4SpringContextTests {
@Autowired
@Qualifier("accountRepository")
private AccountRepository<Account> accountRepository;
@Autowired
private AccountQuotaRepository accountQuotaRepository;
@Autowired
private DomainQuotaRepository domainQuotaRepository;
@Autowired
private ContainerQuotaRepository ensembleQuotaRepository;
@Autowired
@Qualifier("userRepository")
private UserRepository<User> userRepository;
LoadingServiceTestDatas dates;
private User jane;
@Before
public void setUp(){
this.executeSqlScript("import-tests-stat.sql", false);
this.executeSqlScript("import-tests-operationHistory.sql", false);
this.executeSqlScript("import-tests-quota.sql", false);
dates = new LoadingServiceTestDatas(userRepository);
dates.loadUsers();
jane = dates.getUser2();
}
@Test
public void test() {
AbstractDomain domain2 = jane.getDomain();
AbstractDomain domain1 = domain2.getParentDomain();
Account account1 = jane;
Quota result = accountQuotaRepository.find(account1);
assertNotNull(result);
DomainQuota domain2Quota = domainQuotaRepository.find(domain2);
ContainerQuota entity = new ContainerQuota(domain2, domain1, domain2Quota, 100, 90, 10, 40, 20, ContainerQuotaType.WORK_GROUP);
ensembleQuotaRepository.create(entity);
Quota result3 = domainQuotaRepository.find(domain2);
assertEquals(new Long(1096), result3.getCurrentValue());
assertEquals(new Long(1900), result3.getQuota());
result3 = domainQuotaRepository.find(domain1);
assertEquals(new Long(1096), result3.getCurrentValue());
assertEquals(new Long(100), result3.getLastValue());
assertEquals(new Long(2300), result3.getQuota());
assertEquals(new Long(2000), result3.getQuotaWarning());
List<String> listIdentifier = accountQuotaRepository.findDomainUuidByBatchModificationDate(yesterday());
assertEquals(1, listIdentifier.size());
logger.debug(" domain identifier : "+listIdentifier.get(0));
}
private Date yesterday() {
GregorianCalendar dateCalender = new GregorianCalendar();
dateCalender.add(GregorianCalendar.DATE, -1);
dateCalender.set(GregorianCalendar.HOUR_OF_DAY, 23);
dateCalender.set(GregorianCalendar.MINUTE, 59);
dateCalender.set(GregorianCalendar.SECOND, 59);
return dateCalender.getTime();
}
}