org.gaixie.jibu.security.service.SettingServiceTest.java Source code

Introduction

Here is the source code for org.gaixie.jibu.security.service.SettingServiceTest.java

Source

/*
 *  Licensed to the Apache Software Foundation (ASF) under one or more
 *  contributor license agreements.  See the NOTICE file distributed with
 *  this work for additional information regarding copyright ownership.
 *  The ASF licenses this file to You under the Apache License, Version 2.0
 *  (the "License"); you may not use this file except in compliance with
 *  the License.  You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 *  Unless required by applicable law or agreed to in writing, software
 *  distributed under the License is distributed on an "AS IS" BASIS,
 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *  See the License for the specific language governing permissions and
 *  limitations under the License.
 */
package org.gaixie.jibu.security.service;

import java.sql.Connection;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.List;

import org.apache.commons.dbutils.DbUtils;
import org.apache.commons.dbutils.QueryRunner;
import org.junit.After;
import org.junit.Assert;
import org.junit.Before;
import org.junit.Test;
import org.gaixie.jibu.CoreTestSupport;
import org.gaixie.jibu.JibuException;
import org.gaixie.jibu.utils.ConnectionUtils;
import org.gaixie.jibu.security.model.Setting;
import org.gaixie.jibu.security.model.User;
import org.gaixie.jibu.security.service.SettingService;
import org.gaixie.jibu.security.service.UserService;

public class SettingServiceTest extends CoreTestSupport {
    private SettingService settingService;
    private UserService userService;

    @Before
    public void setup() throws Exception {
        settingService = getInjector().getInstance(SettingService.class);
        userService = getInjector().getInstance(UserService.class);
        userService.add(new User("Administrator", "admin", "123456", "jibu.gaixie@gmail.com", true));
        Setting setting = new Setting("theme", "blue", 0, true);
        settingService.add(setting);
        setting = new Setting("theme", "gray", 1, true);
        settingService.add(setting);
    }

    @Test
    public void testGet() throws Exception {
        Setting setting = settingService.get("theme", "blue");
        Assert.assertNotNull(setting);
        Setting s = settingService.getDefault("theme");
        Assert.assertTrue(setting.equals(s));
        // test get(Integer) method
        s = settingService.get(setting.getId());
        Assert.assertTrue(setting.equals(s));
    }

    @Test
    public void testUpdateMe() throws Exception {
        User user = userService.get("admin");
        Setting setting = settingService.get("theme", "gray");
        List<Integer> list = new ArrayList<Integer>();
        list.add(setting.getId());
        settingService.updateMe(list, user);
        setting = settingService.getByUsername("theme", "admin");
        Assert.assertTrue("gray".equals(setting.getValue()));
        // ??
        settingService.reset("admin");
        setting = settingService.getByUsername("theme", "admin");
        Assert.assertNull(setting);

    }

    @Test
    public void testUpdateOrDelete() throws Exception {
        Setting setting = settingService.get("theme", "blue");
        setting.setValue("tommy");
        settingService.update(setting);
        setting = settingService.get("theme", "tommy");
        Assert.assertNotNull(setting);
        settingService.delete(setting);
        setting = settingService.get("theme", "tommy");
        Assert.assertNull(setting);
    }

    @Test
    public void testGetByUsername() throws Exception {
        Setting dfsetting = settingService.getDefault("theme");
        User user = userService.get("admin");
        Setting s = settingService.getByUsername("theme", "admin");
        // bind? null
        Assert.assertNull(s);
        Setting setting = settingService.get("theme", "gray");
        settingService.bind(setting.getId(), user.getUsername());
        s = settingService.getByUsername("theme", "admin");
        // bind?? setting
        Assert.assertTrue(setting.equals(s));
        // ?bind??null
        settingService.unbind(setting.getId(), user.getUsername());
        s = settingService.getByUsername("theme", "admin");
        Assert.assertNull(s);
    }

    @Test
    public void testFindByName() throws Exception {
        List<Setting> settings = settingService.findByName("theme");
        Assert.assertTrue(settings.size() == 2);
    }

    @After
    public void tearDown() {
        Connection conn = null;
        try {
            conn = ConnectionUtils.getConnection();
            QueryRunner run = new QueryRunner();
            run.update(conn, "DELETE from user_setting_map");
            run.update(conn, "DELETE from settings");
            run.update(conn, "DELETE from userbase");
            DbUtils.commitAndClose(conn);
        } catch (SQLException e) {
            DbUtils.rollbackAndCloseQuietly(conn);
            System.out.println(e.getMessage());
        }
    }
}