123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142 |
- /*
- Copyright (c) 2008 Health Market Science, 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.healthmarketscience.jackcess;
-
- import java.util.ArrayList;
- import java.util.Arrays;
- import java.util.Collections;
- import java.util.Comparator;
- import java.util.List;
-
- import static com.healthmarketscience.jackcess.impl.JetFormatTest.*;
- import com.healthmarketscience.jackcess.impl.RelationshipImpl;
- import junit.framework.TestCase;
- import static com.healthmarketscience.jackcess.TestUtil.*;
-
- /**
- * @author James Ahlborn
- */
- public class RelationshipTest extends TestCase {
-
- private static final Comparator<Relationship> REL_COMP = new Comparator<Relationship>() {
- public int compare(Relationship r1, Relationship r2) {
- return String.CASE_INSENSITIVE_ORDER.compare(r1.getName(), r2.getName());
- }
- };
-
- public RelationshipTest(String name) throws Exception {
- super(name);
- }
-
- public void testTwoTables() throws Exception {
- for (final TestDB testDB : TestDB.getSupportedForBasename(Basename.INDEX, true)) {
- Database db = open(testDB);
- Table t1 = db.getTable("Table1");
- Table t2 = db.getTable("Table2");
- Table t3 = db.getTable("Table3");
-
- List<Relationship> rels = db.getRelationships(t1, t2);
- assertEquals(1, rels.size());
- Relationship rel = rels.get(0);
- assertEquals("Table2Table1", rel.getName());
- assertEquals(t2, rel.getFromTable());
- assertEquals(Arrays.asList(t2.getColumn("id")),
- rel.getFromColumns());
- assertEquals(t1, rel.getToTable());
- assertEquals(Arrays.asList(t1.getColumn("otherfk1")),
- rel.getToColumns());
- assertTrue(rel.hasReferentialIntegrity());
- assertEquals(4096, ((RelationshipImpl)rel).getFlags());
- assertTrue(rel.cascadeDeletes());
- assertSameRelationships(rels, db.getRelationships(t2, t1), true);
-
- rels = db.getRelationships(t2, t3);
- assertTrue(db.getRelationships(t2, t3).isEmpty());
- assertSameRelationships(rels, db.getRelationships(t3, t2), true);
-
- rels = db.getRelationships(t1, t3);
- assertEquals(1, rels.size());
- rel = rels.get(0);
- assertEquals("Table3Table1", rel.getName());
- assertEquals(t3, rel.getFromTable());
- assertEquals(Arrays.asList(t3.getColumn("id")),
- rel.getFromColumns());
- assertEquals(t1, rel.getToTable());
- assertEquals(Arrays.asList(t1.getColumn("otherfk2")),
- rel.getToColumns());
- assertTrue(rel.hasReferentialIntegrity());
- assertEquals(256, ((RelationshipImpl)rel).getFlags());
- assertTrue(rel.cascadeUpdates());
- assertSameRelationships(rels, db.getRelationships(t3, t1), true);
-
- try {
- db.getRelationships(t1, t1);
- fail("IllegalArgumentException should have been thrown");
- } catch(IllegalArgumentException ignored) {
- // success
- }
- }
- }
-
- public void testOneTable() throws Exception {
- for (final TestDB testDB : TestDB.getSupportedForBasename(Basename.INDEX, true)) {
- Database db = open(testDB);
- Table t1 = db.getTable("Table1");
- Table t2 = db.getTable("Table2");
- Table t3 = db.getTable("Table3");
-
- List<Relationship> expected = new ArrayList<Relationship>();
- expected.addAll(db.getRelationships(t1, t2));
- expected.addAll(db.getRelationships(t2, t3));
-
- assertSameRelationships(expected, db.getRelationships(t2), false);
-
- }
- }
-
- public void testNoTables() throws Exception {
- for (final TestDB testDB : TestDB.getSupportedForBasename(Basename.INDEX, true)) {
- Database db = open(testDB);
- Table t1 = db.getTable("Table1");
- Table t2 = db.getTable("Table2");
- Table t3 = db.getTable("Table3");
-
- List<Relationship> expected = new ArrayList<Relationship>();
- expected.addAll(db.getRelationships(t1, t2));
- expected.addAll(db.getRelationships(t2, t3));
- expected.addAll(db.getRelationships(t1, t3));
-
- assertSameRelationships(expected, db.getRelationships(), false);
- }
- }
-
- private static void assertSameRelationships(
- List<Relationship> expected, List<Relationship> found, boolean ordered)
- {
- assertEquals(expected.size(), found.size());
- if(!ordered) {
- Collections.sort(expected, REL_COMP);
- Collections.sort(found, REL_COMP);
- }
- for(int i = 0; i < expected.size(); ++i) {
- Relationship eRel = expected.get(i);
- Relationship fRel = found.get(i);
- assertEquals(eRel.getName(), fRel.getName());
- }
- }
-
- }
|