/** * Copyright 2006-2016 the original author or authors. * * 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 mbg.test.ib2j2.conditional; import mbg.test.common.util.TestUtilities; import mbg.test.ib2j2.generated.conditional.dao.AwfulTableDAO; import mbg.test.ib2j2.generated.conditional.dao.PkblobsDAO; import mbg.test.ib2j2.generated.conditional.dao.PkfieldsDAO; import mbg.test.ib2j2.generated.conditional.dao.PkfieldsblobsDAO; import mbg.test.ib2j2.generated.conditional.model.AwfulTable; import mbg.test.ib2j2.generated.conditional.model.Pkblobs; import mbg.test.ib2j2.generated.conditional.model.Pkfields; import mbg.test.ib2j2.generated.conditional.model.PkfieldsKey; import mbg.test.ib2j2.generated.conditional.model.Pkfieldsblobs; import mbg.test.ib2j2.generated.conditional.model.PkfieldsblobsKey; public class UpdateByPrimaryKeyTest extends AbstractConditionalJava2Test { public void testPKFieldsUpdateByPrimaryKey() { PkfieldsDAO dao = getPkfieldsDAO(); try { Pkfields record = new Pkfields(); record.setFirstname("Jeff"); record.setLastname("Smith"); record.setId1(new Integer(1)); record.setId2(new Integer(2)); dao.insert(record); record.setFirstname("Scott"); record.setLastname("Jones"); int rows = dao.updateByPrimaryKey(record); assertEquals(1, rows); PkfieldsKey key = new PkfieldsKey(); key.setId1(new Integer(1)); key.setId2(new Integer(2)); Pkfields record2 = dao.selectByPrimaryKey(key); assertEquals(record.getFirstname(), record2.getFirstname()); assertEquals(record.getLastname(), record2.getLastname()); assertEquals(record.getId1(), record2.getId1()); assertEquals(record.getId2(), record2.getId2()); } catch (Exception e) { fail(e.getMessage()); } } public void testPKFieldsUpdateByPrimaryKeySelective() { PkfieldsDAO dao = getPkfieldsDAO(); try { Pkfields record = new Pkfields(); record.setFirstname("Jeff"); record.setLastname("Smith"); record.setDecimal60field(new Integer(5)); record.setId1(new Integer(1)); record.setId2(new Integer(2)); dao.insert(record); Pkfields newRecord = new Pkfields(); newRecord.setId1(new Integer(1)); newRecord.setId2(new Integer(2)); newRecord.setFirstname("Scott"); newRecord.setDecimal60field(new Integer(4)); int rows = dao.updateByPrimaryKeySelective(newRecord); assertEquals(1, rows); PkfieldsKey key = new PkfieldsKey(); key.setId1(new Integer(1)); key.setId2(new Integer(2)); Pkfields returnedRecord = dao.selectByPrimaryKey(key); assertTrue(TestUtilities.datesAreEqual(record.getDatefield(), returnedRecord .getDatefield())); assertEquals(record.getDecimal100field(), returnedRecord .getDecimal100field()); assertEquals(record.getDecimal155field(), returnedRecord .getDecimal155field()); assertEquals(record.getDecimal30field(), returnedRecord .getDecimal30field()); assertEquals(newRecord.getDecimal60field(), returnedRecord .getDecimal60field()); assertEquals(newRecord.getFirstname(), returnedRecord .getFirstname()); assertEquals(record.getId1(), returnedRecord.getId1()); assertEquals(record.getId2(), returnedRecord.getId2()); assertEquals(record.getLastname(), returnedRecord.getLastname()); assertTrue(TestUtilities.timesAreEqual(record.getTimefield(), returnedRecord .getTimefield())); assertEquals(record.getTimestampfield(), returnedRecord .getTimestampfield()); } catch (Exception e) { fail(e.getMessage()); } } public void testPKBlobsUpdateByPrimaryKeyWithBLOBs() { PkblobsDAO dao = getPkblobsDAO(); try { Pkblobs record = new Pkblobs(); record.setId(new Integer(3)); record.setBlob1(TestUtilities.generateRandomBlob()); record.setBlob2(TestUtilities.generateRandomBlob()); dao.insert(record); record = new Pkblobs(); record.setId(new Integer(3)); record.setBlob1(TestUtilities.generateRandomBlob()); record.setBlob2(TestUtilities.generateRandomBlob()); dao.updateByPrimaryKey(record); Pkblobs newRecord = dao.selectByPrimaryKey(new Integer(3)); assertNotNull(newRecord); assertEquals(record.getId(), newRecord.getId()); assertTrue(TestUtilities.blobsAreEqual(record.getBlob1(), newRecord.getBlob1())); assertTrue(TestUtilities.blobsAreEqual(record.getBlob2(), newRecord.getBlob2())); } catch (Exception e) { fail(e.getMessage()); } } public void testPKBlobsUpdateByPrimaryKeySelective() { PkblobsDAO dao = getPkblobsDAO(); try { Pkblobs record = new Pkblobs(); record.setId(new Integer(3)); record.setBlob1(TestUtilities.generateRandomBlob()); record.setBlob2(TestUtilities.generateRandomBlob()); dao.insert(record); Pkblobs newRecord = new Pkblobs(); newRecord.setId(new Integer(3)); newRecord.setBlob2(TestUtilities.generateRandomBlob()); dao.updateByPrimaryKeySelective(newRecord); Pkblobs returnedRecord = dao.selectByPrimaryKey(new Integer(3)); assertNotNull(returnedRecord); assertEquals(record.getId(), returnedRecord.getId()); assertTrue(TestUtilities.blobsAreEqual(record.getBlob1(), returnedRecord .getBlob1())); assertTrue(TestUtilities.blobsAreEqual(newRecord.getBlob2(), returnedRecord .getBlob2())); } catch (Exception e) { fail(e.getMessage()); } } public void testPKFieldsBlobsUpdateByPrimaryKeyWithBLOBs() { PkfieldsblobsDAO dao = getPkfieldsblobsDAO(); try { Pkfieldsblobs record = new Pkfieldsblobs(); record.setId1(new Integer(3)); record.setId2(new Integer(4)); record.setFirstname("Jeff"); record.setLastname("Smith"); record.setBlob1(TestUtilities.generateRandomBlob()); dao.insert(record); Pkfieldsblobs updateRecord = new Pkfieldsblobs(); updateRecord.setId1(new Integer(3)); updateRecord.setId2(new Integer(4)); updateRecord.setFirstname("Scott"); updateRecord.setLastname("Jones"); updateRecord.setBlob1(TestUtilities.generateRandomBlob()); int rows = dao.updateByPrimaryKeyWithBLOBs(updateRecord); assertEquals(1, rows); PkfieldsblobsKey key = new PkfieldsblobsKey(); key.setId1(new Integer(3)); key.setId2(new Integer(4)); Pkfieldsblobs newRecord = dao.selectByPrimaryKey(key); assertEquals(updateRecord.getFirstname(), newRecord.getFirstname()); assertEquals(updateRecord.getLastname(), newRecord.getLastname()); assertEquals(record.getId1(), newRecord.getId1()); assertEquals(record.getId2(), newRecord.getId2()); assertTrue(TestUtilities.blobsAreEqual(updateRecord.getBlob1(), newRecord .getBlob1())); } catch (Exception e) { fail(e.getMessage()); } } public void testPKFieldsBlobsUpdateByPrimaryKeyWithoutBLOBs() { PkfieldsblobsDAO dao = getPkfieldsblobsDAO(); try { Pkfieldsblobs record = new Pkfieldsblobs(); record.setId1(new Integer(3)); record.setId2(new Integer(4)); record.setFirstname("Jeff"); record.setLastname("Smith"); record.setBlob1(TestUtilities.generateRandomBlob()); dao.insert(record); Pkfieldsblobs updateRecord = new Pkfieldsblobs(); updateRecord.setId1(new Integer(3)); updateRecord.setId2(new Integer(4)); updateRecord.setFirstname("Scott"); updateRecord.setLastname("Jones"); int rows = dao.updateByPrimaryKeyWithoutBLOBs(updateRecord); assertEquals(1, rows); PkfieldsblobsKey key = new PkfieldsblobsKey(); key.setId1(new Integer(3)); key.setId2(new Integer(4)); Pkfieldsblobs newRecord = dao.selectByPrimaryKey(key); assertEquals(updateRecord.getFirstname(), newRecord.getFirstname()); assertEquals(updateRecord.getLastname(), newRecord.getLastname()); assertEquals(record.getId1(), newRecord.getId1()); assertEquals(record.getId2(), newRecord.getId2()); assertTrue(TestUtilities.blobsAreEqual(record.getBlob1(), newRecord.getBlob1())); } catch (Exception e) { fail(e.getMessage()); } } public void testPKFieldsBlobsUpdateByPrimaryKeySelective() { PkfieldsblobsDAO dao = getPkfieldsblobsDAO(); try { Pkfieldsblobs record = new Pkfieldsblobs(); record.setId1(new Integer(3)); record.setId2(new Integer(4)); record.setFirstname("Jeff"); record.setLastname("Smith"); record.setBlob1(TestUtilities.generateRandomBlob()); dao.insert(record); Pkfieldsblobs updateRecord = new Pkfieldsblobs(); updateRecord.setId1(new Integer(3)); updateRecord.setId2(new Integer(4)); updateRecord.setLastname("Jones"); int rows = dao.updateByPrimaryKeySelective(updateRecord); assertEquals(1, rows); PkfieldsblobsKey key = new PkfieldsblobsKey(); key.setId1(new Integer(3)); key.setId2(new Integer(4)); Pkfieldsblobs returnedRecord = dao.selectByPrimaryKey(key); assertEquals(record.getFirstname(), returnedRecord.getFirstname()); assertEquals(updateRecord.getLastname(), returnedRecord .getLastname()); assertEquals(record.getId1(), returnedRecord.getId1()); assertEquals(record.getId2(), returnedRecord.getId2()); assertTrue(TestUtilities.blobsAreEqual(record.getBlob1(), returnedRecord .getBlob1())); } catch (Exception e) { fail(e.getMessage()); } } public void testAwfulTableUpdateByPrimaryKey() { AwfulTableDAO dao = getAwfulTableDAO(); try { AwfulTable record = new AwfulTable(); record.seteMail("fred@fred.com"); record.setEmailaddress("alsofred@fred.com"); record.setFirstFirstName("fred1"); record.setFrom("from field"); record.setId1(new Integer(1)); record.setId2(new Integer(2)); record.setId5(new Integer(5)); record.setId6(new Integer(6)); record.setId7(new Integer(7)); record.setSecondFirstName("fred2"); record.setThirdFirstName("fred3"); Integer generatedCustomerId = dao.insert(record); record.setId1(new Integer(11)); record.setId2(new Integer(22)); int rows = dao.updateByPrimaryKey(record); assertEquals(1, rows); AwfulTable returnedRecord = dao.selectByPrimaryKey(generatedCustomerId); assertEquals(generatedCustomerId, returnedRecord.getCustomerId()); assertEquals(record.geteMail(), returnedRecord.geteMail()); assertEquals(record.getEmailaddress(), returnedRecord .getEmailaddress()); assertEquals(record.getFirstFirstName(), returnedRecord .getFirstFirstName()); assertEquals(record.getFrom(), returnedRecord.getFrom()); assertEquals(record.getId1(), returnedRecord.getId1()); assertEquals(record.getId2(), returnedRecord.getId2()); assertEquals(record.getId5(), returnedRecord.getId5()); assertEquals(record.getId6(), returnedRecord.getId6()); assertEquals(record.getId7(), returnedRecord.getId7()); assertEquals(record.getSecondFirstName(), returnedRecord .getSecondFirstName()); assertEquals(record.getThirdFirstName(), returnedRecord .getThirdFirstName()); } catch (Exception e) { fail(e.getMessage()); } } public void testAwfulTableUpdateByPrimaryKeySelective() { AwfulTableDAO dao = getAwfulTableDAO(); try { AwfulTable record = new AwfulTable(); record.seteMail("fred@fred.com"); record.setEmailaddress("alsofred@fred.com"); record.setFirstFirstName("fred1"); record.setFrom("from field"); record.setId1(new Integer(1)); record.setId2(new Integer(2)); record.setId5(new Integer(5)); record.setId6(new Integer(6)); record.setId7(new Integer(7)); record.setSecondFirstName("fred2"); record.setThirdFirstName("fred3"); Integer generatedCustomerId = dao.insert(record); AwfulTable newRecord = new AwfulTable(); newRecord.setCustomerId(generatedCustomerId); newRecord.setId1(new Integer(11)); newRecord.setId2(new Integer(22)); int rows = dao.updateByPrimaryKeySelective(newRecord); assertEquals(1, rows); AwfulTable returnedRecord = dao.selectByPrimaryKey(generatedCustomerId); assertEquals(generatedCustomerId, returnedRecord.getCustomerId()); assertEquals(record.geteMail(), returnedRecord.geteMail()); assertEquals(record.getEmailaddress(), returnedRecord .getEmailaddress()); assertEquals(record.getFirstFirstName(), returnedRecord .getFirstFirstName()); assertEquals(record.getFrom(), returnedRecord.getFrom()); assertEquals(newRecord.getId1(), returnedRecord.getId1()); assertEquals(newRecord.getId2(), returnedRecord.getId2()); assertEquals(record.getId5(), returnedRecord.getId5()); assertEquals(record.getId6(), returnedRecord.getId6()); assertEquals(record.getId7(), returnedRecord.getId7()); assertEquals(record.getSecondFirstName(), returnedRecord .getSecondFirstName()); assertEquals(record.getThirdFirstName(), returnedRecord .getThirdFirstName()); } catch (Exception e) { fail(e.getMessage()); } } }