package com.sequenceiq.it.cloudbreak.recipes;

import java.util.HashSet;
import java.util.Set;

import org.apache.commons.codec.binary.Base64;
import org.springframework.util.Assert;
import org.testng.annotations.Optional;
import org.testng.annotations.Parameters;
import org.testng.annotations.Test;

import com.sequenceiq.cloudbreak.api.endpoint.v1.RecipeEndpoint;
import com.sequenceiq.cloudbreak.api.model.RecipeRequest;
import com.sequenceiq.cloudbreak.api.model.RecipeType;
import com.sequenceiq.it.IntegrationTestContext;
import com.sequenceiq.it.cloudbreak.AbstractCloudbreakIntegrationTest;
import com.sequenceiq.it.cloudbreak.CloudbreakITContextConstants;

public class RecipeCreationTest extends AbstractCloudbreakIntegrationTest {

    @Test
    @Parameters({ "name", "description", "preScript", "postScript" })
    public void testRecipeCreation(String name, @Optional ("") String description, @Optional("") String preScript,
            @Optional("") String postScript) {
        // GIVEN
        // WHEN
        if (!preScript.isEmpty()) {
            createRecipe(name + "pre", preScript, RecipeType.POST_AMBARI_START, description);
        }
        if (!postScript.isEmpty()) {
            createRecipe(name + "post", postScript, RecipeType.POST_CLUSTER_INSTALL, description);
        }
    }

    private void addRecipeToContext(Long id) {
        IntegrationTestContext itContext = getItContext();
        Set<Long> recipeIds = itContext.getContextParam(CloudbreakITContextConstants.RECIPE_ID, Set.class);
        recipeIds = recipeIds == null ? new HashSet<>() : recipeIds;
        recipeIds.add(id);
        itContext.putContextParam(CloudbreakITContextConstants.RECIPE_ID, recipeIds);
    }

    private void createRecipe(String name, String script, RecipeType recipeType, String description) {
        RecipeRequest recipeRequest = new RecipeRequest();
        recipeRequest.setRecipeType(recipeType);
        recipeRequest.setName(name);
        recipeRequest.setContent(Base64.encodeBase64String(script.getBytes()));
        recipeRequest.setDescription(description);

        RecipeEndpoint recipeEndpoint = getCloudbreakClient().recipeEndpoint();
        Long id = recipeEndpoint.postPrivate(recipeRequest).getId();
        //then
        Assert.notNull(id, "Recipe is not created.");
        addRecipeToContext(id);
    }
}
@@ -140,48 +140,8 @@
 64;%0A
-import org.springframework.util.Assert;%0A
 impo
@@ -253,16 +253,42 @@
 ns.Test;
+%0Aimport org.testng.Assert;
 %0A%0Aimport
@@ -2253,17 +2253,23 @@
  Assert.
-n
+assertN
 otNull(i

Fix assert typo