diff --git a/public/biopet-framework/src/test/scala/nl/lumc/sasc/biopet/tools/CheckAllelesVcfInBamTest.scala b/public/biopet-framework/src/test/scala/nl/lumc/sasc/biopet/tools/CheckAllelesVcfInBamTest.scala index 6a99b92d3fd3b9ba16f9c09e67094cff1e40bedf..ade8c28fa9e1de2f2de945ca58dd7b4d94253a9e 100644 --- a/public/biopet-framework/src/test/scala/nl/lumc/sasc/biopet/tools/CheckAllelesVcfInBamTest.scala +++ b/public/biopet-framework/src/test/scala/nl/lumc/sasc/biopet/tools/CheckAllelesVcfInBamTest.scala @@ -15,8 +15,11 @@ */ package nl.lumc.sasc.biopet.tools +import java.io.File import java.nio.file.Paths +import htsjdk.samtools.{SamReaderFactory, SamReader} +import htsjdk.variant.vcf.VCFFileReader import org.scalatest.Matchers import org.scalatest.mock.MockitoSugar import org.scalatest.testng.TestNGSuite @@ -38,6 +41,7 @@ class CheckAllelesVcfInBamTest extends TestNGSuite with MockitoSugar with Matche val vcf = resourcePath("/chrQ.vcf") val bam = resourcePath("/single01.bam") + val vcf2 = new File(resourcePath("/chrQ2.vcf.gz")) val rand = new Random() @Test def testOutputTypeVcf() = { @@ -58,4 +62,19 @@ class CheckAllelesVcfInBamTest extends TestNGSuite with MockitoSugar with Matche main(arguments) } + @Test + def testCheckAllelesNone() = { + val variant = new File(vcf) + val samRecord = SamReaderFactory.makeDefault().open(new File(bam)).iterator().next() + val varRecord = new VCFFileReader(variant, false).iterator().next() + checkAlleles(samRecord, varRecord) shouldBe None + } + + @Test + def testCheckAlleles() = { + val samRecord = SamReaderFactory.makeDefault().open(new File(bam)).iterator().next() + val varRecord = new VCFFileReader(vcf2).iterator().next() + checkAlleles(samRecord, varRecord) shouldBe Some("T") + } + }