#!/usr/bin/python
import unittest as u
import re, fnmatch, os, sys
sampleMappingFile = '../examples/samples.mapping.txt'
cSamplesDir = '../examples/'
javaSamplesDir = '../examples/java/'
cppSamplesDir = '../examples/c++/'
class SampleNames(u.TestCase):
def test_existing_samples(self):
missing_c_files = []
missing_cpp_files = []
missing_java_files = []
with open (sampleMappingFile, "r") as f:
for line in f:
sampleNames = line.split();
cSampleName = sampleNames[0]
javaSampleName = sampleNames[1]
#check for C files
if cSampleName.endswith('.c'):
ok = False
for file in os.listdir(cSamplesDir):
if file == cSampleName:
ok = True
break
if not ok:
missing_c_files.append(cSampleName)
#check for Cpp files
if cSampleName.endswith('.cpp'):
ok = False
for file in os.listdir(cppSamplesDir):
if file == cSampleName:
ok = True
break
if not ok:
missing_cpp_files.append(cSampleName)
#check for java files
javaSampleName = javaSampleName.lstrip("java/")
if javaSampleName.endswith('.java'):
ok = False
for file in os.listdir(javaSamplesDir):
if file == javaSampleName:
ok = True
break
if not ok:
missing_java_files.append(javaSampleName)
self.assertEqual( len(missing_java_files) + len(missing_c_files) + len(missing_cpp_files), 0,
"\nThe following files are missing from samples:\n" + \
"\n".join(missing_c_files) + "\n" + \
"\n".join(missing_cpp_files) + "\n" + \
"\n".join(missing_java_files))
if __name__ == '__main__':
u.main()