/*
Copyright 2012 James Iry
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.pogofish.jadt.samples.ast;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.Set;
import org.junit.Test;
import com.pogofish.jadt.samples.ast.Usage;
import com.pogofish.jadt.samples.ast.data.*;
import static com.pogofish.jadt.samples.ast.data.Arg._Arg;
import static com.pogofish.jadt.samples.ast.data.Expression.*;
import static com.pogofish.jadt.samples.ast.data.Function.*;
import static com.pogofish.jadt.samples.ast.data.Statement.*;
import static com.pogofish.jadt.samples.ast.data.Type.*;
import static java.util.Arrays.asList;
import static org.junit.Assert.*;
/**
* Make sure the sample usage does what it says it does
*
* @author jiry
*/
public class UsageTest {
static final Usage usage = new Usage();
@Test
public void testSampleFunction() {
final Function expectedFunction = _Function(_Int(), "addTwo", asList(_Arg(_Int(), "x"), _Arg(_Int(), "y")), asList(
_Return(_Add(_Variable("x"), _Variable("y")))
));
assertEquals(expectedFunction, usage.sampleFunction());
}
@Test
public void testExpressionLiterals() {
final Set<Integer> emptyIntegers = usage.expressionLiterals(_LongLiteral(432L));
assertTrue("Set not empty, got " + emptyIntegers, emptyIntegers.isEmpty());
final Set<Integer> twoIntegers = usage.expressionLiterals(_Add(_Variable("Foo"), _Add(_IntLiteral(2), _IntLiteral(5))));
assertTrue("Wrong number of elements in set", twoIntegers.size() == 2);
assertTrue("Set didn't have 2", twoIntegers.contains(2));
assertTrue("Set didn't have 5", twoIntegers.contains(5));
}
@Test
public void testHasReturn() {
final List<Statement> noReturn = Arrays.asList(_Declaration(_Int(), "Foo", _IntLiteral(2)), _Assignment("foo", _IntLiteral(3)));
assertFalse("Got the wrong answer from a list with no returns", usage.hasReturn(noReturn));
final List<Statement> hasReturn = new ArrayList<Statement>(noReturn);
hasReturn.add(_Return(_LongLiteral(3)));
assertTrue("Got wrong answer from a list with a return", usage.hasReturn(hasReturn));
}
}