Factorio-Stdlib/spec/data/recipe_spec.lua at 6c58af5cb09187f73328b5e82b1b3ec546852840 · Afforess/Factorio-Stdlib · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
require('spec/setup/busted')()

local Recipe, Raw, Rawtech
local table = require('__stdlib2__/stdlib/utils/table')

describe('Recipe', function()

    before_each(function()
        require('faketorio/dataloader')
        Recipe = require('__stdlib2__/stdlib/data/recipe')
        Raw = _G["data"].raw["recipe"]
        Rawtech = _G["data"].raw["technology"]["steel-processing"]
    end)

    after_each(function()
        package.remove_stdlib()
    end)

    describe(':get', function()

        it('should get a recipe', function()
            --assert.has_error(function() Recipe() end)
            assert.not_nil(Recipe("stone-furnace"))
            assert.not_nil(Recipe("fake"))
        end)
    end)

    describe(':is_valid', function()

        it('should return the recipe', function()
            assert.truthy(Recipe("stone-furnace"):is_valid())
            assert.truthy(Recipe("stone-furnace"):is_valid("recipe"))
            local recipe = Recipe("stone-furnace")
            assert.truthy(recipe:is_valid())
            assert.truthy(recipe:is_valid("recipe"))
        end)

        it('should return falsy if not a recipe', function()
            assert.not_truthy(Recipe("fake"):is_valid())
            assert.not_truthy(Recipe("fake"):is_valid("recipe"))
            local recipe = Recipe("fake")
            assert.not_truthy(recipe:is_valid())
            assert.not_truthy(recipe:is_valid("recipe"))
        end)
    end)

    describe(':copy', function()

        it('should error without a new name', function()
            assert.has_error(function() Recipe:get("stone-furnace"):copy() end)
        end)

        it('should extend a recipe copy and return it', function()
            assert.is_nil(Raw["fake"])
            local recipe = Recipe("stone-furnace"):copy("fake")
            assert.truthy(Raw["fake"])
            assert.same(Raw["fake"].name, recipe.name)
        end)
    end)

    describe(':make_difficult', function()
        it('should change the recipe to the difficulty version', function()
            local raw = Raw["stone-furnace"]
            assert.is_nil(raw.normal)
            assert.not_nil(raw.ingredients)
            Recipe("stone-furnace"):make_difficult()
            assert.same(1, #raw.normal.ingredients)
            assert.same(1, #raw.normal.results)
            assert.same(1, #raw.expensive.ingredients)
            assert.same(1, #raw.expensive.results)
            assert.is_nil(raw.ingredients)
            assert.is_nil(raw.result, raw.results)
        end)
    end)

    describe(':change_category', function()

        it('should change the category if it exists', function()
            assert.same(nil, _G["data"].raw.recipe["stone-furnace"].category)
            Recipe("stone-furnace"):change_category("hand-held")
            assert.same(nil, _G["data"].raw.recipe["stone-furnace"].category)
            Recipe("stone-furnace"):change_category("advanced-crafting")
            assert.same("advanced-crafting", _G["data"].raw.recipe["stone-furnace"].category)
        end)
    end)

    describe(':pairs()', function()
        it('should classify everything', function()
            for _, class in Recipe:pairs('recipe') do
                assert.same('Recipe', class.__class)
            end
        end)

        -- it('should work without a paramater', function()
        --     for _, class in Recipe:pairs() do
        --         assert.same('Recipe', class.__class)
        --     end
        -- end)
        it('should work on a class', function()
            local r = Recipe('stone-furnace')
            local count = 0
            for _, class in r:pairs() do
                assert.same('Recipe', class.__class)
                count = count + 1
            end
            assert.same(table.size(_G["data"].raw.recipe), count)
        end)
    end)

    describe(':add_unlock', function()
        it('should add the unlock if the tech exists', function()
            Recipe("stone-furnace"):add_unlock("steel-processing")
            log(Rawtech.effects)
            assert.same("stone-furnace", Rawtech.effects[3].recipe)
            assert.not_truthy(_G["data"].raw.recipe["stone-furnace"].enabled)
        end)

        it('should not do anything if the tech doesnt exist', function()
            local s = spy.on(_G, "log")
            Recipe("stone-furnace"):add_unlock("fake")
            assert.spy(s).was_called(1)
        end)
    end)

    describe(':remove_unlock', function()
        it('should remove the unlock from the tech if it exists', function()
            Recipe("steel-plate"):remove_unlock("steel-processing")
            assert.same(1, #Rawtech.effects)
        end)

        -- it('should remove the unlock from all techs', function()
        --     Recipe("steel-plate"):remove_unlock()
        --     assert.same(2, #Rawtech.effects)
        --     assert.same(0, #_G.data.raw["technology"]["steel-processing-2"].effects)
        -- end)

    end)
end)