test: add botkiller detection tests

This commit is contained in:
xenticore
2025-04-07 15:40:44 -04:00
parent 8fa0eb55e1
commit 9a8ce24313

View File

@@ -1,5 +1,5 @@
import { describe, expect, test, mock } from "bun:test";
import { ItemSchema, ItemSlot, getItemIndexByName, getTradableStatusByDefindex, getTradableStatusByName, linkFestiveVariants, prepareSchema } from '../src/content/schemaService'
import { ItemSchema, ItemSlot, getItemIndexByName, getTradableStatusByDefindex, getTradableStatusByName, linkBotkillerVariants, linkFestiveVariants, prepareSchema } from '../src/content/schemaService'
// Mock the storage and log functions
mock.module('../src/content/storage', () => ({
@@ -20,7 +20,8 @@ const mockSchema: ItemSchema = {
tradable: false,
hasAustraliumVariant: false,
canKillstreakify: true,
festiveVariant: null
festiveVariant: null,
botkillerVariants: null
},
'208': {
name: 'Flame Thrower',
@@ -28,7 +29,8 @@ const mockSchema: ItemSchema = {
tradable: true,
hasAustraliumVariant: true,
canKillstreakify: true,
festiveVariant: 659
festiveVariant: 659,
botkillerVariants: [798, 807]
},
'659': {
name: 'Festive Flame Thrower',
@@ -36,7 +38,26 @@ const mockSchema: ItemSchema = {
tradable: true,
hasAustraliumVariant: false,
canKillstreakify: true,
festiveVariant: null
festiveVariant: null,
botkillerVariants: null
},
'798': {
name: 'Silver Botkiller Flame Thrower Mk.I',
slot: ItemSlot.Primary,
tradable: true,
hasAustraliumVariant: false,
canKillstreakify: true,
festiveVariant: null,
botkillerVariants: null
},
'807': {
name: 'Gold Botkiller Flame Thrower Mk.I',
slot: ItemSlot.Primary,
tradable: true,
hasAustraliumVariant: false,
canKillstreakify: true,
festiveVariant: null,
botkillerVariants: null
},
'5021': {
name: 'Mann Co. Supply Crate Key',
@@ -44,7 +65,8 @@ const mockSchema: ItemSchema = {
tradable: true,
hasAustraliumVariant: false,
canKillstreakify: false,
festiveVariant: null
festiveVariant: null,
botkillerVariants: null
},
'15141': {
name: 'Flame Thrower',
@@ -52,7 +74,8 @@ const mockSchema: ItemSchema = {
tradable: true,
hasAustraliumVariant: false,
canKillstreakify: true,
festiveVariant: null
festiveVariant: null,
botkillerVariants: null
},
'69420': {
name: 'Non-Tradable Item',
@@ -60,7 +83,8 @@ const mockSchema: ItemSchema = {
tradable: false,
hasAustraliumVariant: false,
canKillstreakify: false,
festiveVariant: null
festiveVariant: null,
botkillerVariants: null
}
}
@@ -140,7 +164,8 @@ describe('Schema Service', () => {
tradable: false,
hasAustraliumVariant: false,
canKillstreakify: false,
festiveVariant: null
festiveVariant: null,
botkillerVariants: null
},
'208': { // Original Flame Thrower
name: 'Flame Thrower',
@@ -148,7 +173,8 @@ describe('Schema Service', () => {
tradable: true,
hasAustraliumVariant: true,
canKillstreakify: true,
festiveVariant: null
festiveVariant: null,
botkillerVariants: null
},
'659': { // Festive Flame Thrower (should be detected)
name: 'Festive Flame Thrower',
@@ -156,7 +182,8 @@ describe('Schema Service', () => {
tradable: true,
hasAustraliumVariant: false,
canKillstreakify: true,
festiveVariant: null
festiveVariant: null,
botkillerVariants: null
},
'15141': { // Decorated (should be ignored)
name: 'Flame Thrower',
@@ -164,7 +191,8 @@ describe('Schema Service', () => {
tradable: true,
hasAustraliumVariant: false,
canKillstreakify: true,
festiveVariant: null
festiveVariant: null,
botkillerVariants: null
}
};
const mockResponseItems = [
@@ -188,7 +216,8 @@ describe('Schema Service', () => {
tradable: true,
hasAustraliumVariant: false,
canKillstreakify: false,
festiveVariant: null
festiveVariant: null,
botkillerVariants: null
}
};
@@ -201,4 +230,111 @@ describe('Schema Service', () => {
expect(testSchema['163'].festiveVariant).toBeNull();
});
});
describe('linkBotkillerVariants', () => {
test('should link botkiller variants to their original counterparts', () => {
const testSchema = {
'21': { // Original Rocket Launcher
name: 'Flame Thrower',
slot: ItemSlot.Primary,
tradable: false,
hasAustraliumVariant: false,
canKillstreakify: false,
festiveVariant: null,
botkillerVariants: null
},
'208': { // Original Flame Thrower
name: 'Flame Thrower',
slot: ItemSlot.Primary,
tradable: true,
hasAustraliumVariant: true,
canKillstreakify: true,
festiveVariant: null,
botkillerVariants: null
},
'798': {
name: 'Silver Botkiller Flame Thrower Mk. I',
slot: ItemSlot.Primary,
tradable: true,
hasAustraliumVariant: false,
canKillstreakify: true,
festiveVariant: null,
botkillerVariants: null
},
'807': {
name: 'Gold Botkiller Flame Thrower Mk. I',
slot: ItemSlot.Primary,
tradable: true,
hasAustraliumVariant: false,
canKillstreakify: true,
festiveVariant: null,
botkillerVariants: null
},
'15141': { // Decorated (should be ignored)
name: 'Flame Thrower',
slot: ItemSlot.Primary,
tradable: true,
hasAustraliumVariant: false,
canKillstreakify: true,
festiveVariant: null,
botkillerVariants: null
}
};
const mockResponseItems = [
{ item_class: 'tf_weapon_flamethrower', defindex: 21, item_name: 'Flame Thrower', item_type_name: 'Flame Thrower'
}, // Incorrect; stock
{
item_class: 'tf_weapon_flamethrower',
defindex: 208,
item_name: 'Flame Thrower',
item_type_name: 'Flame Thrower'
},
{
item_class: 'tf_weapon_flamethrower',
defindex: 798,
item_name: 'Silver Botkiller Flame Thrower Mk.I',
item_type_name: 'Flame Thrower'
},
{
item_class: 'tf_weapon_flamethrower',
defindex: 807,
item_name: 'Gold Botkiller Flame Thrower Mk.I',
item_type_name: 'Flame Thrower'
},
{ item_class: 'tf_weapon_flamethrower', defindex: 15141, item_name: 'Flame Thrower', item_type_name: 'Flame Thrower' }, // Incorrect; decorated
];
linkBotkillerVariants(mockResponseItems, testSchema);
expect(testSchema['21'].botkillerVariants).toBeNull()
expect(testSchema['208'].botkillerVariants).toEqual([798, 807]);
expect(testSchema['798'].botkillerVariants).toBeNull();
expect(testSchema['807'].botkillerVariants).toBeNull();
expect(testSchema['15141'].botkillerVariants).toBeNull()
});
test('should not link if no botkiller variants exist', () => {
const testSchema = {
'163': { // Crit-a-Cola
name: 'Crit-a-Cola',
slot: ItemSlot.Secondary,
tradable: true,
hasAustraliumVariant: false,
festiveVariant: null,
canKillstreakify: false,
botkillerVariants: null,
}
};
const mockResponseItems = [
{
item_class: 'tf_weapon_lunchbox_drink',
defindex: 163,
item_name: 'Crit-a-Cola',
item_type_name: 'Lunch Box',
}
];
linkBotkillerVariants(mockResponseItems, testSchema);
expect(testSchema['163'].botkillerVariants).toBeNull();
});
})
})