@alexkratky/rpi-info/tests/RaspberryPiInfo.test.ts

Detect Raspberry Pi model

import fs from 'fs';
import { RaspberryPiInfo, RaspberryPiInfoResult } from '../index'; // Adjust the path as necessary

jest.mock('fs');

describe('RaspberryPiInfo', () => {
    beforeAll(() => {
        // Mock the content of /proc/cpuinfo
        const cpuInfoContent = `
processor   : 0
BogoMIPS    : 108.00
Features    : fp asimd evtstrm aes pmull sha1 sha2 crc32 atomics fphp asimdhp cpuid asimdrdm lrcpc dcpop asimddp
CPU implementer : 0x41
CPU architecture: 8
CPU variant : 0x4
CPU part    : 0xd0b
CPU revision: 1

processor   : 1
BogoMIPS    : 108.00
Features    : fp asimd evtstrm aes pmull sha1 sha2 crc32 atomics fphp asimdhp cpuid asimdrdm lrcpc dcpop asimddp
CPU implementer : 0x41
CPU architecture: 8
CPU variant : 0x4
CPU part    : 0xd0b
CPU revision: 1

processor   : 2
BogoMIPS    : 108.00
Features    : fp asimd evtstrm aes pmull sha1 sha2 crc32 atomics fphp asimdhp cpuid asimdrdm lrcpc dcpop asimddp
CPU implementer : 0x41
CPU architecture: 8
CPU variant : 0x4
CPU part    : 0xd0b
CPU revision: 1

processor   : 3
BogoMIPS    : 108.00
Features    : fp asimd evtstrm aes pmull sha1 sha2 crc32 atomics fphp asimdhp cpuid asimdrdm lrcpc dcpop asimddp
CPU implementer : 0x41
CPU architecture: 8
CPU variant : 0x4
CPU part    : 0xd0b
CPU revision: 1

Revision    : c04170
Serial      : d18eff6ba14f4538
Model       : Raspberry Pi 5 Model B Rev 1.0
`;

        // Mock the readFileSync method to return the cpuInfoContent
        (fs.readFileSync as jest.Mock).mockReturnValue(cpuInfoContent);
    });

    it('should detect Raspberry Pi model based on /proc/cpuinfo', () => {
        const piInfo = new RaspberryPiInfo();
        const result: RaspberryPiInfoResult = piInfo.detect();

        expect(result).toEqual({
            isRaspberry: true,
            model: '5',
            fullModelName: 'Raspberry Pi 5',
            fullModelNameWithRam: 'Raspberry Pi 5 - 4GB',
            ram: '4GB',
            manufacturer: 'Sony UK',
            revisionCode: 'c04170'
        });
    });

    it('should detect Raspberry Pi model based on /proc/cpuinfo with custom revision', () => {
        const piInfo = new RaspberryPiInfo({'c04170': 
            { model: '0', ram: '128GB', manufacturer: 'Custom' }
        });
        const result: RaspberryPiInfoResult = piInfo.detect();

        expect(result).toEqual({
            isRaspberry: true,
            model: '0',
            fullModelName: 'Raspberry Pi 0',
            fullModelNameWithRam: 'Raspberry Pi 0 - 128GB',
            ram: '128GB',
            manufacturer: 'Custom',
            revisionCode: 'c04170'
        });
    });

    it('should detect Raspberry Pi model based on /proc/cpuinfo with custom rpi name', () => {
        const piInfo = new RaspberryPiInfo({}, 'Custom RPI');
        const result: RaspberryPiInfoResult = piInfo.detect();

        expect(result).toEqual({
            isRaspberry: true,
            model: '5',
            fullModelName: 'Custom RPI 5',
            fullModelNameWithRam: 'Custom RPI 5 - 4GB',
            ram: '4GB',
            manufacturer: 'Sony UK',
            revisionCode: 'c04170'
        });
    });

    it('should return default values when /proc/cpuinfo does not contain Revision', () => {
        // Mock a different content without the Revision line
        const noRevisionContent = `
processor   : 0
BogoMIPS    : 108.00
Features    : fp asimd evtstrm aes pmull sha1 sha2 crc32 atomics fphp asimdhp cpuid asimdrdm lrcpc dcpop asimddp
CPU implementer : 0x41
CPU architecture: 8
CPU variant : 0x4
CPU part    : 0xd0b
CPU revision: 1
`;

        (fs.readFileSync as jest.Mock).mockReturnValue(noRevisionContent);

        const piInfo = new RaspberryPiInfo();
        const result: RaspberryPiInfoResult = piInfo.detect();

        expect(result).toEqual({
            isRaspberry: false,
            model: null,
            fullModelName: null,
            fullModelNameWithRam: null,
            ram: null,
            manufacturer: null,
            revisionCode: null
        });
    });

    it('should return default values when revision code is not found in revisions', () => {
        // Mock a different content with an unknown Revision code
        const unknownRevisionContent = `
processor   : 0
BogoMIPS    : 108.00
Features    : fp asimd evtstrm aes pmull sha1 sha2 crc32 atomics fphp asimdhp cpuid asimdrdm lrcpc dcpop asimddp
CPU implementer : 0x41
CPU architecture: 8
CPU variant : 0x4
CPU part    : 0xd0b
CPU revision: 1

Revision    : unknown
Serial      : d18eff6ba14f4538
Model       : Raspberry Pi 5 Model B Rev 1.0
`;

        (fs.readFileSync as jest.Mock).mockReturnValue(unknownRevisionContent);

        const piInfo = new RaspberryPiInfo();
        const result: RaspberryPiInfoResult = piInfo.detect();

        expect(result).toEqual({
            isRaspberry: false,
            model: null,
            fullModelName: null,
            fullModelNameWithRam: null,
            ram: null,
            manufacturer: null,
            revisionCode: null
        });
    });
});