summaryrefslogtreecommitdiff
path: root/test/elixir/test/partition_size_test.exs
blob: 91e79f7b3108ba7bebf5afafee9c143a32117f4c (plain)
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
defmodule PartitionSizeTest do
  use CouchTestCase
  import PartitionHelpers

  @moduledoc """
  Test Partition size functionality
  """
  @tag :with_partitioned_db
  test "get partition size", context do
    db_name = context[:db_name]
    create_partition_docs(db_name)

    info_resp = Couch.get("/#{db_name}")
    %{:body => info} = info_resp
    external_size = info["sizes"]["external"]

    url = "/#{db_name}/_partition/foo"
    resp = Couch.get(url)

    assert resp.status_code == 200
    %{:body => body} = resp
    assert body["doc_count"] == 50
    assert body["partition"] == "foo"
    assert body["sizes"]["external"] == external_size / 2

    url = "/#{db_name}/_partition/bar"
    resp = Couch.get(url)

    assert resp.status_code == 200
    %{:body => body} = resp
    assert body["doc_count"] == 50
    assert body["partition"] == "bar"
    assert body["sizes"]["external"] == external_size / 2
  end

  @tag :with_partitioned_db
  test "get partition size for two partitions on same shard", context do
    db_name = context[:db_name]
    create_partition_docs(db_name, "foo", "bar42")

    info_resp = Couch.get("/#{db_name}")
    %{:body => info} = info_resp
    external_size = info["sizes"]["external"]

    url = "/#{db_name}/_partition/foo"
    resp = Couch.get(url)

    assert resp.status_code == 200
    %{:body => body} = resp
    assert body["doc_count"] == 50
    assert body["partition"] == "foo"
    assert body["sizes"]["external"] == external_size / 2

    url = "/#{db_name}/_partition/bar42"
    resp = Couch.get(url)

    assert resp.status_code == 200
    %{:body => body} = resp
    assert body["doc_count"] == 50
    assert body["partition"] == "bar42"
    assert body["sizes"]["external"] == external_size / 2
  end

  @tag :with_partitioned_db
  test "get partition size with attachment", context do
    db_name = context[:db_name]
    # create_partition_docs(db_name, "foo", "bar42")

    id = "foo:doc-with-attachment"

    doc = %{
      _id: id,
      _attachments: %{
        "foo.txt": %{
          content_type: "text/plain",
          data: Base.encode64("This is a text document to save")
        }
      }
    }

    Couch.put("/#{db_name}/#{id}", body: doc)

    info_resp = Couch.get("/#{db_name}")
    %{:body => info} = info_resp
    external_size = info["sizes"]["external"]

    url = "/#{db_name}/_partition/foo"
    resp = Couch.get(url)

    assert resp.status_code == 200
    %{:body => body} = resp
    assert body["doc_count"] == 1
    assert body["partition"] == "foo"
    assert body["sizes"]["external"] == external_size
  end

  @tag :with_partitioned_db
  test "get multiple partition sizes with attachment", context do
    db_name = context[:db_name]
    create_partition_docs(db_name, "foo", "bar42")

    id = "foo:doc-with-attachment"

    doc = %{
      _id: id,
      _attachments: %{
        "foo.txt": %{
          content_type: "text/plain",
          data: Base.encode64("This is a text document to save")
        }
      }
    }

    Couch.put("/#{db_name}/#{id}", body: doc)

    url = "/#{db_name}/_partition/foo"
    resp = Couch.get(url)

    assert resp.status_code == 200
    %{:body => body} = resp
    assert body["doc_count"] == 51
    assert body["partition"] == "foo"
    # hard coding in values here. Not sure if this is a good idea long term
    assert body["sizes"]["external"] == 4503

    url = "/#{db_name}/_partition/bar42"
    resp = Couch.get(url)

    assert resp.status_code == 200
    %{:body => body} = resp
    assert body["doc_count"] == 50
    assert body["partition"] == "bar42"
    assert body["sizes"]["external"] == 4450
  end
end