diff --git a/gittensor/classes.py b/gittensor/classes.py index 4856d474..fa3e84f0 100644 --- a/gittensor/classes.py +++ b/gittensor/classes.py @@ -74,9 +74,11 @@ def __post_init__(self): def _calculate_file_extension(self) -> str: basename = self.filename.split('/')[-1] + basename_lower = basename.lower() + if basename_lower.startswith('dockerfile.'): + return 'dockerfile' if '.' in basename: return basename.split('.')[-1].lower() - basename_lower = basename.lower() return basename_lower if basename_lower in EXTENSIONLESS_FILE_EXTENSIONS else '' def is_test_file(self) -> bool: diff --git a/tests/test_classes.py b/tests/test_classes.py index 9f4f47cc..e064a8a4 100644 --- a/tests/test_classes.py +++ b/tests/test_classes.py @@ -78,6 +78,8 @@ def test_is_test_file_preserves_existing_test_conventions(): ('Dockerfile', 'dockerfile'), ('dockerfile', 'dockerfile'), ('ops/Dockerfile', 'dockerfile'), + ('Dockerfile.dev', 'dockerfile'), + ('ops/Dockerfile.prod', 'dockerfile'), ('Makefile', 'makefile'), ('makefile', 'makefile'), ('build.mk', 'mk'),