File: message_test.rb

package info (click to toggle)
rails 2%3A7.2.2.1%2Bdfsg-7
  • links: PTS, VCS
  • area: main
  • in suites: trixie
  • size: 43,352 kB
  • sloc: ruby: 349,799; javascript: 30,703; yacc: 46; sql: 43; sh: 29; makefile: 27
file content (43 lines) | stat: -rw-r--r-- 1,491 bytes parent folder | download | duplicates (2)
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
# frozen_string_literal: true

require "cases/encryption/helper"

class ActiveRecord::Encryption::MessageTest < ActiveRecord::EncryptionTestCase
  test "add_header lets you add headers" do
    message = ActiveRecord::Encryption::Message.new
    message.headers[:header_1] = "value 1"

    assert_equal "value 1", message.headers[:header_1]
  end

  test "add_headers lets you add multiple headers" do
    message = ActiveRecord::Encryption::Message.new
    message.headers.add(header_1: "value 1", header_2: "value 2")
    assert_equal "value 1", message.headers[:header_1]
    assert_equal "value 2", message.headers[:header_2]
  end

  test "headers can't be overridden" do
    message = ActiveRecord::Encryption::Message.new
    message.headers.add(header_1: "value 1")

    assert_raises(ActiveRecord::Encryption::Errors::EncryptedContentIntegrity) do
      message.headers.add(header_1: "value 1")
    end

    assert_raises(ActiveRecord::Encryption::Errors::EncryptedContentIntegrity) do
      message.headers.add(header_1: "value 1")
    end
  end

  test "validates that payloads are either nil or strings" do
    assert_raises ActiveRecord::Encryption::Errors::ForbiddenClass do
      ActiveRecord::Encryption::Message.new(payload: Date.new)
      ActiveRecord::Encryption::Message.new(payload: [])
    end

    ActiveRecord::Encryption::Message.new
    ActiveRecord::Encryption::Message.new(payload: "")
    ActiveRecord::Encryption::Message.new(payload: "Some payload")
  end
end