/*
 * Copyright 2011 Splunk, Inc.
 *
 * Licensed 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 com.splunk;

import java.util.Arrays;

import org.junit.Assert;
import org.junit.Test;

public class LicensePoolTest extends SDKTestCase {
    @Test
    public void testCreateWithNoQuota() throws Exception {
        LicensePoolCollection licensePoolCollection = service.getLicensePools();
        try {
            // The following will fail because there is no quota available
            licensePoolCollection.create("sdk-test", "MAX", "download-trial");
            Assert.fail("Expected pool create to fail");
        }
        catch (HttpException e) {
            Assert.assertEquals(400, e.getStatus());
        }
    }
    
    @Test
    public void testLicensePoolSetters() throws Exception {
        LicensePoolCollection licensePoolCollection = service.getLicensePools();
        
        // Try updating some pools ..
        boolean foundAnUpdatablePool = false;
        for (LicensePool licensePool : licensePoolCollection.values()) {
            if (licensePool.getStackId().equals("download-trial"))
                continue; // Can't edit a pool in stack: download-trial
            if (licensePool.getStackId().equals("forwarder"))
                continue; // Can't edit a pool in stack: forwarder
            if (licensePool.getStackId().equals("free"))
                continue; // Can't edit a pool in stack: free
            foundAnUpdatablePool = true;

            String originalDescription = licensePool.getDescription();
            
            // Probe
            {
                licensePool.setDescription("sdk-test description");
                licensePool.setQuota("1048576");
                licensePool.update();
    
                Assert.assertEquals("sdk-test description", licensePool.getDescription());
                Assert.assertEquals("1048576", licensePool.getQuota());
            }

            licensePool.update(new Args("description", originalDescription));
            licensePool.update(new Args("quota", "MAX"));
        }
        
        if (!foundAnUpdatablePool) {
            System.out.println("WARNING: Didn't find any updatable license pools.");
        }
    }
    
    @Test
    public void testLicensePoolGetters() {
        LicensePoolCollection licensePoolCollection = service.getLicensePools();
        for (LicensePool licensePool : licensePoolCollection.values()) {
            testLicensePoolGetters(licensePool);
        }
    }
    
    private void testLicensePoolGetters(LicensePool licensePool) {
        licensePool.getDescription();
        licensePool.getQuota();
        licensePool.getSlaves();
        licensePool.getSlavesUsageBytes();
        licensePool.getStackId();
        licensePool.getUsedBytes();
    }
}